Added interactive config creation

pointsbot/bot.py

@@ -19,7 +19,8 @@ TEST_COMMENTS = False
 
 
 def run():
-    cfg = config.Config.load()
+    #cfg = config.Config.load()
+    cfg = config.load()
     levels = cfg.levels
 
     reddit = praw.Reddit(client_id=cfg.client_id,

pointsbot/config.py

@@ -1,6 +1,7 @@
 import os
 import os.path
-# from os.path import abspath, dirname, expanduser, join
+from collections import namedtuple
+from copy import deepcopy
 
 import toml
 
@@ -8,24 +9,34 @@ from .level import Level
 
 ### Globals ###
 
-# ROOTPATH = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+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:
 
-    DATADIR = os.path.join(os.path.expanduser('~'), '.pointsbot')
-    PATH = os.path.join(DATADIR, 'pointsbot.toml')
-
     # Default config vals
-    DEFAULT_DBPATH = os.path.join(DATADIR, 'pointsbot.db')
+    # DEFAULT_DBPATH = os.path.join(DATADIR, 'pointsbot.db')
+    DEFAULT_DBNAME = 'pointsbot.db'
 
-    def __init__(self, subreddit, client_id, client_secret, username, password,
-                 levels, database_path=DEFAULT_DBPATH):
-        self.subreddit = subreddit
+    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)
         self.database_path = database_path
 
+        self.subreddit = subreddit
+
         self.client_id = client_id
         self.client_secret = client_secret
         self.username = username
@@ -34,32 +45,103 @@ class Config:
         self.levels = levels
 
     @classmethod
-    def load(cls, filepath=PATH):
+    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['flair_template_id']
-            if flair_template_id == "":
+            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)
 
-        database_path = os.path.join(cls.DATADIR, obj['filepaths']['database'])
+        # database_path = os.path.join(DATADIR, obj['filepaths']['database'])
 
         return cls(
+            filepath,
             obj['core']['subreddit'],
             obj['credentials']['client_id'],
             obj['credentials']['client_secret'],
             obj['credentials']['username'],
             obj['credentials']['password'],
             levels,
-            database_path=database_path,
+            database_path=obj['filepaths']['database'],
+            # database_path=database_path,
         )
 
-    @classmethod
-    def dump(cls, obj, filepath=PATH):
-        pass
+    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)
+
+        # with open(SAMPLEPATH) as fin:
+            # with open(filepath, 'w') as fout:
+                # fout.write(fin.read())
+
+        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'] = 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(f'\nConfig settings saved to {dest}')