ShiteMusicBot/cogs/utils/settings.py

import os
import json
import codecs
import locale

class Settings:
    def __init__(self, **default_settings):
        self.DATA_PATH = 'data/settings/'
        self.SETTINGS_PATH = self.DATA_PATH + 'settings.json'
        self.default_prefix = default_settings["default_prefix"]
        self.default_admin = default_settings["default_admin"]
        self.default_mod = default_settings["default_mod"]
        self.default_locale = default_settings["default_locale"]
        self.default_lang_path = default_settings["default_lang_path"] or "./localization"
        if not self.default_locale:
            locale, codepage = locale.getlocale()
            default_locale, default_codepage = locale.getdefaultlocale()
            self.default_locale = locale or default_locale

        if not os.path.exists(self.DATA_PATH):
            os.makedirs(self.DATA_PATH)

        if not os.path.isfile(self.SETTINGS_PATH):
            with codecs.open(self.SETTINGS_PATH, "w+", encoding='utf8') as f:
                json.dump({"locale": {}, "prefixes": {},
                           "roles": {}}, f, indent=4)

        with codecs.open(self.SETTINGS_PATH, "r", encoding='utf8') as f:
            self.settings = json.load(f)

    def get_prefix(self, guild_id):
        guild_id = str(guild_id)
        if guild_id in self.settings["prefixes"].keys():
            return self.settings["prefixes"][guild_id]
        return self.default_prefix

    def set_prefix(self, guild_id, prefixes):
        if prefixes is None:
            self.settings["prefixes"].pop(str(guild_id), None)
        else:
            self.settings["prefixes"][str(guild_id)] = prefixes
        with codecs.open(self.SETTINGS_PATH, "w", encoding='utf8') as f:
            json.dump(self.settings, f, indent=4)

    def get_mod_role(self, guild_id):
        guild_id = str(guild_id)
        if guild_id in self.settings["roles"].keys():
            if 'mod' in self.settings["roles"][guild_id].keys():
                return self.settings["roles"][guild_id]['mod']
        return self.default_mod

    def get_admin_role(self, guild_id):
        guild_id = str(guild_id)
        if guild_id in self.settings["roles"].keys():
            if 'admin' in self.settings["roles"][guild_id].keys():
                return self.settings["roles"][guild_id]['admin']
        return self.default_admin

    def set_mod_admin_role(self, guild_id, admin, mod):
        guild_id = str(guild_id)
        if admin is None or mod is None:
            self.settings["role"].pop(guild_id, None)
        else:
            self.settings["role"][guild_id]["mod"] = mod
            self.settings["role"][guild_id]["admin"] = admin

        with codecs.open(self.SETTINGS_PATH, "w", encoding='utf8') as f:
            json.dump(self.settings, f, indent=4)

    def set_locale(self, guild_id, locale):
        guild_id = str(guild_id)
        self.settings["locale"][guild_id] = locale
        with codecs.open(self.SETTINGS_PATH, "w", encoding='utf8') as f:
            json.dump(self.settings, f, indent=4)

    def get_locale(self, guild_id):
        guild_id = str(guild_id)
        return self.settings.get("locale", {}).get(guild_id, self.default_locale).lower()

    def get_language_path(self):
        return self.settings.get("language_path", self.default_lang_path)