diff --git a/BotResources.py b/BotResources.py index ea8e993..027cfc6 100644 --- a/BotResources.py +++ b/BotResources.py @@ -53,20 +53,24 @@ def read_config_file(): config = configparser.ConfigParser() config.read('./config.ini') - config_return = { - 'Bot Token': config['Bot_Info']['Token'], - 'Device ID': int(config['Device']['ID']), - 'Device Name': str(config['Device']['Name']), - 'Mention Group': str(config['Bot_Info']['Mention_Group']), - 'Channel ID': int(config['Bot_Info']['Channel_ID']), - 'Handler': str(config['Config']['Handler']) - } + try: + config_return = { + 'Bot Token': config['Bot_Info']['Token'], + 'Device ID': int(config['Device']['ID']), + 'Device Name': str(config['Device']['Name']), + 'Mention Group': str(config['Bot_Info']['Mention_Group']), + 'Channel ID': int(config['Bot_Info']['Channel_ID']), + 'Handler': str(config['Config']['Handler']) + } - LOGGER.debug("Found config options:") - for key in config_return.keys(): - LOGGER.debug(f"\t{key} : {config_return[key]}") + LOGGER.debug("Found config options:") + for key in config_return.keys(): + LOGGER.debug(f"\t{key} : {config_return[key]}") - return config_return + return config_return + except Exception as err: + LOGGER.warning(err) + return None def write_config_file(**kwargs): diff --git a/main.py b/main.py index 88aa9af..00825ed 100644 --- a/main.py +++ b/main.py @@ -39,6 +39,10 @@ def main(**passed_config): config = BotResources.read_config_file() + if not config: + LOGGER.warning("No config file exists, please enter this information now") + BotResources.write_config_file(init=True) + # Overwrite config options if they were passed if len(passed_config.keys()) == 0: for sub in config: