56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
import json
|
|
import logging
|
|
|
|
__all__ = ["Config"]
|
|
|
|
from typing import List, Dict
|
|
|
|
import discord
|
|
|
|
from tuxbot.core.data_manager import data_path
|
|
|
|
log = logging.getLogger("tuxbot.config")
|
|
|
|
|
|
class Config:
|
|
def __init__(
|
|
self,
|
|
cog_instance: str = None
|
|
):
|
|
self._cog_instance = cog_instance
|
|
|
|
def __getitem__(self, item) -> Dict:
|
|
path = data_path(self._cog_instance)
|
|
|
|
if item != 'core':
|
|
path = path / 'cogs' / item
|
|
else:
|
|
path /= 'core'
|
|
|
|
settings_file = path / 'settings.json'
|
|
|
|
if not settings_file.exists():
|
|
raise FileNotFoundError(f"Unable to find settings file "
|
|
f"'{settings_file}'")
|
|
else:
|
|
with settings_file.open('r') as f:
|
|
return json.load(f)
|
|
|
|
def __call__(self, item):
|
|
return self.__getitem__(item)
|
|
|
|
def owner_ids(self) -> List[int]:
|
|
return self.__getitem__('core').get('owner_ids')
|
|
|
|
def token(self) -> str:
|
|
return self.__getitem__('core').get('token')
|
|
|
|
def get_prefixes(self, guild: discord.Guild) -> List[str]:
|
|
core = self.__getitem__('core')
|
|
prefixes = core\
|
|
.get('guild', {}) \
|
|
.get(guild.id, {}) \
|
|
.get('prefixes', [])
|
|
|
|
return prefixes
|