roxbot/config/server_config.py

import json

class ServerConfig():
	def __init__(self):
		self.servers_template = {
			"example": {
				"greets": {
					"enabled": 0,
					"welcome-channel": "",
					"member-role": "",
					"custom-message": "",
					"default-message": "Be sure to read the rules."
				},
				"goodbyes": {
					"enabled": 0,
					"goodbye-channel": "",
				},
				"selfAssign": {
					"enabled": 0,
					"roles": []
				},
				"twitch": {
					"enabled": 0,
					"twitch-channel": "",
					"whitelist": {
						"enabled": 0,
						"list": []
					},
				},
				"mute": {
					"role": "",
					"admin-role": []
				},
				"nsfw": {
					"enabled": 0
				}
			}
		}
		self.servers = self.load_config()
		# TODO: Move this to Checks
		self.no_perms_reponse = ":no_entry_sign: You do not have permission to use this command."
		self.delete_after = 20

	def load_config(self):
		with open('config/servers.json', 'r') as config_file:
			return json.load(config_file)

	def update_config(self, config):
		with open('config/servers.json', 'w') as conf_file:
			json.dump(config, conf_file)

	def error_check(self, servers):
		for server in servers:
			if server.id not in self.servers:
				self.servers[server.id] = self.servers_template["example"]
				self.update_config(self.servers)
				print(
					"WARNING: The config file for {} was not found. A template has been loaded and saved. All cogs are turned off by default.".format(
						server.name.upper()))
			else:
				for cog_setting in self.servers_template["example"]:
					for setting in self.servers_template["example"][cog_setting]:
						if setting not in self.servers[server.id][cog_setting]:
							self.servers[server.id][cog_setting][setting] = self.servers_template["example"][
								cog_setting][setting]
							self.update_config(self.servers)
							print(
								"WARNING: The config file for {} was missing the {} setting in the {} cog. This has been fixed with the template version. It is disabled by default.".format(
									server.name.upper(), setting.upper(), cog_setting.upper()))
					if cog_setting not in self.servers[server.id]:
						self.servers[server.id][cog_setting] = self.servers_template["example"][
							cog_setting]
						self.update_config(self.servers)
						print(
							"WARNING: The config file for {} was missing the {} cog. This has been fixed with the template version. It is disabled by default.".format(
								server.name.upper(), cog_setting.upper()))
		print("")