steamgifts-bot/src/run.py

import configparser
from configparser import ConfigParser
from random import randint

import log

logger = log.get_logger(__name__)


class MyException(Exception):
    pass


def value_range(min, max):
    return [str(x) for x in [*range(min, max + 1)]]


class MyConfig(ConfigParser):
    required_values = {
        'DEFAULT': {
            'gift_types': ('All', 'Wishlist', 'Recommended', 'Copies', 'DLC', 'New'),
            'pinned': ('true', 'false'),
            'minimum_points': '%s' % (value_range(0, 400)),
            'max_entries': '%s' % (value_range(0, 100000)),
            'max_time_left': '%s' % (value_range(0, 21600)),
            'minimum_game_points': '%s' % (value_range(0, 50))
        }
    }
    default_values = {
        'DEFAULT':  {
            'cookie': '',
            'gift_types': 'All',
            'pinned': 'true',
            'minimum_points': f"{randint(20, 100)}",
            'max_entries': f"{randint(1000, 2500)}",
            'max_time_left': f"{randint(180,500)}",
            'minimum_game_points': '1',
            'blacklist_keywords': 'hentai,adult'
        }
    }

    def __init__(self, config_file):
        super(MyConfig, self).__init__()
        self.read(config_file)
        modified = self.create_defaults()
        if modified:
            with open(config_file, 'w+') as file:
                self.write(file)
        self.validate_config()

    def create_defaults(self):
        modified = False
        for section, keys in self.default_values.items():
            if section not in self:
                self.add_section(section)
                modified = True
            for key, value in keys.items():
                if key not in self[section]:
                    self.set(section, key, value)
                    modified = True
        return modified

    def validate_config(self):
        for section, keys in self.required_values.items():
            if section not in self:
                raise MyException(
                    'Missing section %s in the config file' % section)

            for key, values in keys.items():
                if key not in self[section] or self[section][key] == '':
                    raise MyException((
                                              'Missing value for %s under section %s in ' +
                                              'the config file') % (key, section))

                if values:
                    if self[section][key] not in values:
                        raise MyException((
                                                  'Invalid value for %s under section %s in ' +
                                                  'the config file') % (key, section))


def run():
    from main import SteamGifts as SG

    file_name = '../config/config.ini'
    config = None
    try:
        config = MyConfig(file_name)
    except IOError:
        txt = f"{file_name} doesn't exist. Rename {file_name}.example to {file_name} and fill out."
        logger.warning(txt)
        exit(-1)
    except MyException as e:
        logger.error(e)
        exit(-1)

    config.read(file_name)
    cookie = config['DEFAULT'].get('cookie')
    pinned_games = config['DEFAULT'].getboolean('pinned')
    gift_types = config['DEFAULT'].get('gift_types')
    minimum_points = config['DEFAULT'].getint('minimum_points')
    max_entries = config['DEFAULT'].getint('max_entries')
    max_time_left = config['DEFAULT'].getint('max_time_left')
    minimum_game_points = config['DEFAULT'].getint('minimum_game_points')
    blacklist = config['DEFAULT'].get('blacklist_keywords')

    s = SG(cookie, gift_types, pinned_games, minimum_points, max_entries, max_time_left, minimum_game_points, blacklist)
    s.start()


if __name__ == '__main__':
    run()