import os import configparser class config: """ config.ini object config -- list with ini data default -- if true, we have generated a default config.ini """ config = configparser.ConfigParser() fileName = "" # config filename default = True # Check if config.ini exists and load/generate it def __init__(self, file): """ Initialize a config object file -- filename """ self.fileName = file if os.path.isfile(self.fileName): # config.ini found, load it self.config.read(self.fileName) self.default = False else: # config.ini not found, generate a default one self.generateDefaultConfig() self.default = True # Check if config.ini has all needed the keys def checkConfig(self): """ Check if this config has the required keys return -- True if valid, False if not """ try: # Try to get all the required keys self.config.get("db","host") self.config.get("db","username") self.config.get("db","password") self.config.get("db","database") self.config.get("db","workers") self.config.get("server","port") self.config.get("server","gzip") self.config.get("server","gziplevel") self.config.get("server","cikey") self.config.get("server","cloudflare") self.config.get("debug","enable") self.config.get("debug","packets") self.config.get("debug","time") self.config.get("sentry","enable") self.config.get("sentry","banchodns") self.config.get("sentry","ircdns") self.config.get("discord","enable") self.config.get("discord","boturl") self.config.get("discord","devgroup") self.config.get("irc","enable") self.config.get("irc","port") self.config.get("localize","enable") self.config.get("localize","ipapiurl") return True except: return False # Generate a default config.ini def generateDefaultConfig(self): """Open and set default keys for that config file""" # Open config.ini in write mode f = open(self.fileName, "w") # Set keys to config object self.config.add_section("db") self.config.set("db", "host", "localhost") self.config.set("db", "username", "root") self.config.set("db", "password", "") self.config.set("db", "database", "ripple") self.config.set("db", "workers", "4") self.config.add_section("server") self.config.set("server", "port", "5001") self.config.set("server", "gzip", "1") self.config.set("server", "gziplevel", "6") self.config.set("server", "cikey", "changeme") self.config.set("server", "cloudflare", "0") self.config.add_section("debug") self.config.set("debug", "enable", "0") self.config.set("debug", "packets", "0") self.config.set("debug", "time", "0") self.config.add_section("sentry") self.config.set("sentry", "enable", "0") self.config.set("sentry", "banchodns", "") self.config.set("sentry", "ircdns", "") self.config.add_section("discord") self.config.set("discord", "enable", "0") self.config.set("discord", "boturl", "") self.config.set("discord", "devgroup", "") self.config.add_section("irc") self.config.set("irc", "enable", "1") self.config.set("irc", "port", "6667") self.config.add_section("localize") self.config.set("localize", "enable", "1") self.config.set("localize", "ipapiurl", "http://ip.zxq.co") # Write ini to file and close self.config.write(f) f.close()