PointsBot/pointsbot/config.py

import os
import os.path
from collections import namedtuple
from copy import deepcopy

import toml

from .level import Level

### Globals ###

DATADIR = os.path.join(os.path.expanduser('~'), '.pointsbot')
CONFIGPATH = os.path.join(DATADIR, 'pointsbot.toml')

# Path to the sample config file
SAMPLEPATH = os.path.abspath(os.path.join(os.path.dirname(__file__),
                                          '..',
                                          'pointsbot.sample.toml'))

### Classes ###


class Config:

    # Default config vals
    DEFAULT_DBNAME = 'pointsbot.db'

    def __init__(self, filepath, subreddit, client_id, client_secret, username,
                 password, levels, database_path=None):
        self._filepath = filepath
        self._dirname = os.path.dirname(filepath)

        if not database_path:
            database_path = os.path.join(self._dirname, self.DEFAULT_DBNAME)
        elif os.path.isdir(database_path):
            database_path = os.path.join(database_path, self.DEFAULT_DBNAME)
        self.database_path = database_path

        self.subreddit = subreddit

        self.client_id = client_id
        self.client_secret = client_secret
        self.username = username
        self.password = password

        self.levels = levels

    @classmethod
    def from_toml(cls, filepath):
        obj = toml.load(filepath)

        # Create list of level objects, in ascending order by point value
        levels = []
        for lvl in obj['levels']:
            flair_template_id = lvl.get('flair_template_id', None)
            if flair_template_id == '':
                flair_template_id = None
            levels.append(Level(lvl['name'], lvl['points'], flair_template_id))
        levels.sort(key=lambda l: l.points)

        dbpath = obj['filepaths']['database']
        if dbpath:
            dbpath = os.path.abspath(os.path.expandvars(os.path.expanduser(dbpath)))

        return cls(
            filepath,
            obj['core']['subreddit'],
            obj['credentials']['client_id'],
            obj['credentials']['client_secret'],
            obj['credentials']['username'],
            obj['credentials']['password'],
            levels,
            database_path=dbpath,
        )

    def save(self):
        obj = deepcopy(vars(self))
        orig_levels = obj['levels']
        obj['levels'] = []
        for level in orig_levels:
            obj['levels'].append({
                'name': level.name,
                'points': level.points,
                'flair_template_id': level.flair_template_id,
            })

        with open(self._filepath, 'w') as f:
            toml.dump(obj, f)


### Functions ###


def load(filepath=CONFIGPATH):
    # Prompt user for config values if file doesn't exist
    if not os.path.exists(filepath):
        datadir = os.path.dirname(filepath)
        if not os.path.exists(datadir):
            os.makedirs(datadir)
        interactive_config(filepath)

    return Config.from_toml(filepath)


### Interactive Config Editing ###


def interactive_config(dest):
    configvals = {
        'core': {},
        'filepaths': {},
        'credentials': {},
        'levels': [],
    }

    print('#' * 80 + '\nCONFIGURING THE BOT\n' + '#' * 80)
    print('\nType a value for each field, then press enter.')
    print('\nIf the field is specified as optional, leave blank to skip.\n')

    configvals['core']['subreddit'] = input('subreddit? ')
    print()
    configvals['filepaths']['database'] = input('database filename? (optional) ')
    print()
    configvals['credentials']['client_id'] = input('client_id? ')
    configvals['credentials']['client_secret'] = input('client_secret? ')
    configvals['credentials']['username'] = input('username? ')
    configvals['credentials']['password'] = input('password? ')

    add_another_level = True
    while add_another_level:
        level = {}
        level['name'] = input('\nLevel name? ')
        level['points'] = int(input('Level points? '))
        level['flair_template_id'] = input('Flair template ID? (optional) ')
        configvals['levels'].append(level)

        response = input('\nAdd another level? (y/n) ')
        add_another_level = response.lower().startswith('y')

    with open(dest, 'w') as f:
        toml.dump(configvals, f)
    print('#' * 80 + f'\nConfig settings saved to {dest}\n' + '#' * 80)