mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-12-08 10:22:31 -05:00
@@ -4,11 +4,17 @@ import os
|
||||
|
||||
default_path = "data/red/settings.json"
|
||||
|
||||
|
||||
class Settings:
|
||||
def __init__(self,path=default_path):
|
||||
|
||||
def __init__(self, path=default_path):
|
||||
self.path = path
|
||||
self.check_folders()
|
||||
self.default_settings = {"EMAIL" : "EmailHere", "PASSWORD" : "", "OWNER" : "id_here", "PREFIXES" : [], "default":{"ADMIN_ROLE" : "Transistor", "MOD_ROLE" : "Process"}, "LOGIN_TYPE" : "email"}
|
||||
self.default_settings = {
|
||||
"EMAIL": "EmailHere", "PASSWORD": "", "OWNER": "id_here",
|
||||
"PREFIXES": [],
|
||||
"default": {"ADMIN_ROLE": "Transistor", "MOD_ROLE": "Process"},
|
||||
"LOGIN_TYPE": "email"}
|
||||
if not dataIO.is_valid_json(self.path):
|
||||
self.bot_settings = self.default_settings
|
||||
self.save_settings()
|
||||
@@ -18,7 +24,8 @@ class Settings:
|
||||
for key in self.default_settings.keys():
|
||||
if key not in current.keys():
|
||||
current[key] = self.default_settings[key]
|
||||
print("Adding " + str(key) + " field to red settings.json")
|
||||
print("Adding " + str(key) +
|
||||
" field to red settings.json")
|
||||
dataIO.save_json(self.path, current)
|
||||
self.bot_settings = dataIO.load_json(self.path)
|
||||
if "default" not in self.bot_settings:
|
||||
@@ -32,14 +39,14 @@ class Settings:
|
||||
os.makedirs(folder)
|
||||
|
||||
def save_settings(self):
|
||||
dataIO.save_json(self.path,self.bot_settings)
|
||||
dataIO.save_json(self.path, self.bot_settings)
|
||||
|
||||
def update_old_settings(self):
|
||||
mod = self.bot_settings["MOD_ROLE"]
|
||||
admin = self.bot_settings["ADMIN_ROLE"]
|
||||
del self.bot_settings["MOD_ROLE"]
|
||||
del self.bot_settings["ADMIN_ROLE"]
|
||||
self.bot_settings["default"] = {"MOD_ROLE":mod,"ADMIN_ROLE":admin}
|
||||
self.bot_settings["default"] = {"MOD_ROLE": mod, "ADMIN_ROLE": admin}
|
||||
self.save_settings()
|
||||
|
||||
@property
|
||||
@@ -47,7 +54,7 @@ class Settings:
|
||||
return self.bot_settings["OWNER"]
|
||||
|
||||
@owner.setter
|
||||
def owner(self,value):
|
||||
def owner(self, value):
|
||||
self.bot_settings["OWNER"] = value
|
||||
self.save_settings()
|
||||
|
||||
@@ -56,7 +63,7 @@ class Settings:
|
||||
return self.bot_settings["EMAIL"]
|
||||
|
||||
@email.setter
|
||||
def email(self,value):
|
||||
def email(self, value):
|
||||
self.bot_settings["EMAIL"] = value
|
||||
self.save_settings()
|
||||
|
||||
@@ -65,7 +72,7 @@ class Settings:
|
||||
return self.bot_settings["PASSWORD"]
|
||||
|
||||
@password.setter
|
||||
def password(self,value):
|
||||
def password(self, value):
|
||||
self.bot_settings["PASSWORD"] = value
|
||||
self.save_settings()
|
||||
|
||||
@@ -74,8 +81,8 @@ class Settings:
|
||||
return self.bot_settings["PREFIXES"]
|
||||
|
||||
@prefixes.setter
|
||||
def prefixes(self,value):
|
||||
assert isinstance(value,list)
|
||||
def prefixes(self, value):
|
||||
assert isinstance(value, list)
|
||||
self.bot_settings["PREFIXES"] = value
|
||||
self.save_settings()
|
||||
|
||||
@@ -83,10 +90,10 @@ class Settings:
|
||||
def default_admin(self):
|
||||
if "default" not in self.bot_settings:
|
||||
self.update_old_settings()
|
||||
return self.bot_settings["default"].get("ADMIN_ROLE","")
|
||||
return self.bot_settings["default"].get("ADMIN_ROLE", "")
|
||||
|
||||
@default_admin.setter
|
||||
def default_admin(self,value):
|
||||
def default_admin(self, value):
|
||||
if "default" not in self.bot_settings:
|
||||
self.update_old_settings()
|
||||
self.bot_settings["default"]["ADMIN_ROLE"] = value
|
||||
@@ -96,10 +103,10 @@ class Settings:
|
||||
def default_mod(self):
|
||||
if "default" not in self.bot_settings:
|
||||
self.update_old_settings()
|
||||
return self.bot_settings["default"].get("MOD_ROLE","")
|
||||
return self.bot_settings["default"].get("MOD_ROLE", "")
|
||||
|
||||
@default_mod.setter
|
||||
def default_mod(self,value):
|
||||
def default_mod(self, value):
|
||||
if "default" not in self.bot_settings:
|
||||
self.update_old_settings()
|
||||
self.bot_settings["default"]["MOD_ROLE"] = value
|
||||
@@ -108,60 +115,62 @@ class Settings:
|
||||
@property
|
||||
def servers(self):
|
||||
ret = {}
|
||||
server_ids = list(filter(lambda x: str(x).isdigit(),self.bot_settings))
|
||||
server_ids = list(
|
||||
filter(lambda x: str(x).isdigit(), self.bot_settings))
|
||||
for server in server_ids:
|
||||
ret.update({server:self.bot_settings[server]})
|
||||
ret.update({server: self.bot_settings[server]})
|
||||
return ret
|
||||
|
||||
@property
|
||||
def login_type(self):
|
||||
return self.bot_settings["LOGIN_TYPE"]
|
||||
return self.bot_settings["LOGIN_TYPE"]
|
||||
|
||||
@login_type.setter
|
||||
def login_type(self,value):
|
||||
self.bot_settings["LOGIN_TYPE"] = value
|
||||
self.save_settings()
|
||||
def login_type(self, value):
|
||||
self.bot_settings["LOGIN_TYPE"] = value
|
||||
self.save_settings()
|
||||
|
||||
def get_server(self,server):
|
||||
def get_server(self, server):
|
||||
if server is None:
|
||||
return self.bot_settings["default"].copy()
|
||||
assert isinstance(server,discord.Server)
|
||||
return self.bot_settings.get(server.id,self.bot_settings["default"]).copy()
|
||||
assert isinstance(server, discord.Server)
|
||||
return self.bot_settings.get(server.id,
|
||||
self.bot_settings["default"]).copy()
|
||||
|
||||
def get_server_admin(self,server):
|
||||
def get_server_admin(self, server):
|
||||
if server is None:
|
||||
return self.default_admin
|
||||
assert isinstance(server,discord.Server)
|
||||
assert isinstance(server, discord.Server)
|
||||
if server.id not in self.bot_settings:
|
||||
return self.default_admin
|
||||
return self.bot_settings[server.id].get("ADMIN_ROLE","")
|
||||
return self.bot_settings[server.id].get("ADMIN_ROLE", "")
|
||||
|
||||
def set_server_admin(self,server,value):
|
||||
def set_server_admin(self, server, value):
|
||||
if server is None:
|
||||
return
|
||||
assert isinstance(server,discord.Server)
|
||||
assert isinstance(server, discord.Server)
|
||||
if server.id not in self.bot_settings:
|
||||
self.add_server(server.id)
|
||||
self.bot_settings[server.id]["ADMIN_ROLE"] = value
|
||||
self.save_settings()
|
||||
|
||||
def get_server_mod(self,server):
|
||||
def get_server_mod(self, server):
|
||||
if server is None:
|
||||
return self.default_mod
|
||||
assert isinstance(server,discord.Server)
|
||||
assert isinstance(server, discord.Server)
|
||||
if server.id not in self.bot_settings:
|
||||
return self.default_mod
|
||||
return self.bot_settings[server.id].get("MOD_ROLE","")
|
||||
return self.bot_settings[server.id].get("MOD_ROLE", "")
|
||||
|
||||
def set_server_mod(self,server,value):
|
||||
def set_server_mod(self, server, value):
|
||||
if server is None:
|
||||
return
|
||||
assert isinstance(server,discord.Server)
|
||||
assert isinstance(server, discord.Server)
|
||||
if server.id not in self.bot_settings:
|
||||
self.add_server(server.id)
|
||||
self.bot_settings[server.id]["MOD_ROLE"] = value
|
||||
self.save_settings()
|
||||
|
||||
def add_server(self,sid):
|
||||
def add_server(self, sid):
|
||||
self.bot_settings[sid] = self.bot_settings["default"].copy()
|
||||
self.save_settings()
|
||||
|
||||
Reference in New Issue
Block a user