136 lines
3.1 KiB
Python
136 lines
3.1 KiB
Python
import json
|
|
import logging
|
|
from typing import List, Dict, Union, Any
|
|
|
|
import discord
|
|
|
|
from tuxbot.core.data_manager import data_path
|
|
|
|
__all__ = ["Config"]
|
|
|
|
log = logging.getLogger("tuxbot.core.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 owners_id(self) -> List[int]:
|
|
"""Simply return the owners id saved in config file.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
Owners id.
|
|
"""
|
|
return self.__getitem__('core').get('owners_id')
|
|
|
|
def token(self) -> str:
|
|
"""Simply return the bot token saved in config file.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
Bot token.
|
|
"""
|
|
return self.__getitem__('core').get('token')
|
|
|
|
def get_prefixes(self, guild: discord.Guild) -> List[str]:
|
|
"""Get custom prefixes for one guild.
|
|
|
|
Parameters
|
|
----------
|
|
guild:discord.Guild
|
|
The required guild prefixes.
|
|
|
|
Returns
|
|
-------
|
|
List[str]
|
|
List of all prefixes.
|
|
"""
|
|
core = self.__getitem__('core')
|
|
prefixes = core \
|
|
.get('guild', {}) \
|
|
.get(guild.id, {}) \
|
|
.get('prefixes', [])
|
|
|
|
return prefixes
|
|
|
|
def get_blacklist(self, key: str) -> List[Union[str, int]]:
|
|
"""Return list off all blacklisted values
|
|
|
|
Parameters
|
|
----------
|
|
key:str
|
|
Which type of blacklist to choice (guilds ? channels ?,...).
|
|
|
|
Returns
|
|
-------
|
|
List[Union[str, int]]
|
|
List containing blacklisted values.
|
|
"""
|
|
core = self.__getitem__('core')
|
|
blacklist = core \
|
|
.get('blacklist', {}) \
|
|
.get(key, [])
|
|
|
|
return blacklist
|
|
|
|
def update(self, cog_name: str, item: str, value: Any) -> dict:
|
|
"""Update values in config file.
|
|
|
|
Parameters
|
|
----------
|
|
cog_name:str
|
|
Name of cog who's corresponding to the config file.
|
|
item:str
|
|
Key to update.
|
|
value:Any
|
|
New values to apply.
|
|
|
|
Returns
|
|
-------
|
|
dict:
|
|
Updated values.
|
|
|
|
"""
|
|
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
|