tuxbot-bot/tuxbot/core/config.py

82 lines
2 KiB
Python
Raw Normal View History

2020-06-04 19:16:51 +02:00
import json
import logging
2020-06-04 19:16:51 +02:00
__all__ = ["Config"]
2020-06-05 00:29:14 +02:00
from typing import List, Dict, Union
2020-06-04 19:16:51 +02:00
import discord
2020-06-04 19:16:51 +02:00
from tuxbot.core.data_manager import data_path
2020-06-04 19:16:51 +02:00
log = logging.getLogger("tuxbot.config")
2020-06-04 19:16:51 +02:00
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)
2020-06-05 00:29:14 +02:00
def owners_id(self) -> List[int]:
return self.__getitem__('core').get('owners_id')
2020-06-04 19:16:51 +02:00
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
2020-06-05 00:29:14 +02:00
def get_blacklist(self, key: str) -> List[Union[str, int]]:
core = self.__getitem__('core')
blacklist = core \
.get('blacklist', {}) \
.get(key, [])
return blacklist
def update(self, cog_name, item, value) -> dict:
datas = self.__getitem__(cog_name)
path = data_path(self._cog_instance)
datas[item] = value
if cog_name != 'core':
path = path / 'cogs' / cog_name
else:
path /= 'core'
settings_file = path / 'settings.json'
with settings_file.open('w') as f:
json.dump(datas, f, indent=4)
return datas