[Core] Settings as Bot class attribute (#486)

And pep8
This commit is contained in:
Will
2016-11-22 17:10:42 -05:00
committed by Twentysix
parent aef46f3820
commit 84ce5c3a8b
2 changed files with 82 additions and 58 deletions

View File

@@ -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()