More command line stuff

scribeengine/config/cmd.py

@@ -24,6 +24,17 @@ The :mod:`~scribeengine.config` module contains helper classes for config handli
 """
 from argparse import ArgumentParser
 from configparser import ConfigParser
+from random import SystemRandom
+from string import ascii_letters, digits, punctuation
+
+ALLOWED_CHARS = ascii_letters + digits + punctuation
+
+
+def _generate_secret(length=32):
+    """
+    Generate a secret
+    """
+    return ''.join(SystemRandom().choice(ALLOWED_CHARS) for _ in range(length))
 
 
 def parse_args():
@@ -75,21 +86,79 @@ def prompt(question, options=None, default=None, type_=str):
     return answer
 
 
-def ask_db_questions():
+def build_db_url(db_type=None, db_name=None, db_user=None, db_password=None, db_host=None, db_port=None):
-    db_type = prompt('Database type', ['sqlite', 'mysql', 'postgres'], 'sqlite').lower()
+    """
-    if db_type == 'sqlite':
+    Build an SQLAlchemy connection string using the values given, using defaults for omitted values
-        db_name = prompt('Database filename',  default='scribeengine.sqlite')
+
-        return '{db_type}:///{db_name}'.format(db_type=db_type, db_name=db_name)
+    :param db_type: The type of database, "sqlite", "mysql", "postgres"
-    db_name = prompt('Database name', default='scribeengine')
+    :param db_name: The name of the database
-    db_user = prompt('Database user', default='scribeengine')
+    :param db_user: The user to connect to the database
-    db_password = prompt('Database password', default='mysupersecretpassword')
+    :param db_password: The password used to authenticate the user
-    db_host = prompt('Database host', default='localhost')
+    :param db_host: The hostname or IP address of the database server
-    if db_type in ['mysql', 'mariadb']:
+    :param db_port: The port of the database server
-        db_port = prompt('Database port', default=3306, type_=int)
+;
-    elif db_type == 'postgres':
+    :return: An SQLAlchemy connection string
-        db_port = prompt('Database port', default=5432, type_=int)
+    """
-    else:
+    if not db_type:
-        db_port = prompt('Database port', default=0, type_=int)
+        db_type = 'sqlite'
+    if not db_name:
+        if db_type == 'sqlite':
+            return 'sqlite:///scribeengine.sqlite'
+        db_name = 'scribeengine'
+    if not db_user:
+        db_user = 'scribeegine'
+    if not db_password:
+        db_password = _generate_secret(40)
+    if not db_host:
+        db_host = 'localhost'
+    if not db_port:
+        if db_type == 'mysql':
+            db_port = 3306
+        elif db_type == 'postgres':
+            db_port = 5432
+        else:
+            db_port = 0
+    return '{db_type}://{db_user}:{db_password}@{db_host}:{db_port}/{db_name}'.format(db_type=db_type,
+                                                                                      db_user=db_user,
+                                                                                      db_password=db_password,
+                                                                                      db_host=db_host,
+                                                                                      db_port=db_port,
+                                                                                      db_name=db_name)
+
+
+def ask_db_questions(db_type=None, db_name=None, db_user=None, db_password=None, db_host=None, db_port=None):
+    """
+    Ask questions about the database in order to build a SQLAlchemy connection string
+
+    :param db_type: The type of database, "sqlite", "mysql", "postgres"
+    :param db_name: The name of the database
+    :param db_user: The user to connect to the database
+    :param db_password: The password used to authenticate the user
+    :param db_host: The hostname or IP address of the database server
+    :param db_port: The port of the database server
+
+    :return: An SQLAlchemy connection string
+    """
+    if not db_type:
+        db_type = prompt('Database type', ['sqlite', 'mysql', 'postgres'], 'sqlite').lower()
+    if not db_name:
+        if db_type == 'sqlite':
+            db_name = prompt('Database filename',  default='scribeengine.sqlite')
+            return '{db_type}:///{db_name}'.format(db_type=db_type, db_name=db_name)
+        db_name = prompt('Database name', default='scribeengine')
+    if not db_user:
+        db_user = prompt('Database user', default='scribeengine')
+    if not db_password:
+        db_password = prompt('Database password', default=_generate_secret(40))
+    if not db_host:
+        db_host = prompt('Database host', default='localhost')
+    if not db_port:
+        if db_type in ['mysql', 'mariadb']:
+            db_port = prompt('Database port', default=3306, type_=int)
+        elif db_type == 'postgres':
+            db_port = prompt('Database port', default=5432, type_=int)
+        else:
+            db_port = prompt('Database port', default=0, type_=int)
     return '{db_type}://{db_user}:{db_password}@{db_host}:{db_port}/{db_name}'.format(db_type=db_type,
                                                                                       db_user=db_user,
                                                                                       db_password=db_password,
@@ -103,9 +172,25 @@ def run_cmd():
     Ask the user a set of questions, and then write them to a config file.
     """
     args = parse_args()
+    if args.non_interactive:
+        db_url = build_db_url(args.db_type, args.db_name, args.db_user, args.db_password, args.db_host, args.db_port)
+    else:
+        db_url = ask_db_questions(args.db_type, args.db_name, args.db_user, args.db_password, args.db_host,
+                                  args.db_port)
     config = {
+        'flask': {
+            'secret_key': _generate_secret()
+        },
         'sqlalchemy': {
-            'database_uri': ask_db_questions()
+            'database_uri': db_url,
+            'track_modifications': False
+        },
+        'mail': {
+
+        },
+        'paths': {
+            'theme_path',
+            'uploads_path'
         }
     }
     write_config_file(config, args.config or 'config.ini')