import configparser def create_config(config): config['Discord'] = {'Token': '', 'Status': '', 'Prefix': '?', 'Bot_Mod': '' } config['SQL'] = {'Dialect+Driver': 'mysql+mysqldb', 'Username': '', 'Password': '', 'Host': '', 'Port': '', 'Database': '' } config['Minecraft'] = {'Dynmap_Url': '', 'World_Name': '' } config['Logging'] = {'Count': '', 'Rotation_Duration': '' } with open('GeoffreyConfig.ini', 'w') as configfile: config.write(configfile) def read_config(): config = configparser.ConfigParser() config.read('GeoffreyConfig.ini') if len(config.sections()) == 0: create_config(config) print("GeoffreyConfig.ini generated.") quit(0) return config class Config: def __init__(self): try: self.config = read_config() self.engine_args = self.read_engine_arg() self.token = self.config['Discord']['Token'] self.world_name = self.config['Minecraft']['World_Name'] self.status = self.config['Discord']['Status'] self.prefix = self.config['Discord']['Prefix'] self.dynmap_url = self.config['Minecraft']['Dynmap_Url'] self.bot_mod = self.config['Discord']['Bot_Mod'] self.count = int(self.config['Logging']['Count']) self.rotation_duration = int(self.config['Logging']['Rotation_Duration']) except Exception as e: print("Invalid config file, missing {}.".format(e)) quit(1) def read_engine_arg(self): driver = self.config['SQL']['Dialect+Driver'] username = self.config['SQL']['Username'] password = self.config['SQL']['Password'] host = self.config['SQL']['Host'] port = self.config['SQL']['Port'] database_name = self.config['SQL']['Database'] engine_args = '{}://{}:{}@{}:{}/{}?charset=utf8mb4&use_unicode=1' return engine_args.format(driver, username, password, host, port, database_name) bot_config = Config()