add cli

Author: tylerdave

implementation/betterapis/__init__.py

@@ -1,12 +1,14 @@
 import connexion
-import logging
 from connexion.resolver import RestyResolver
 from flask_sqlalchemy import SQLAlchemy
 
-logging.basicConfig(level=logging.DEBUG)
+# Uncomment to see import-time debug logging:
+#import logging
+#logging.basicConfig(level=logging.DEBUG)
 
 api = connexion.FlaskApp(__name__, specification_dir='specs/')
-api.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/test.db'
+api.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/betterapis.db'
+api.app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 db = SQLAlchemy(api.app)
 api.add_api('betterapis.yaml', resolver=RestyResolver('betterapis.controllers'))
 

implementation/betterapis/cli.py

@@ -0,0 +1,36 @@
+import click
+import logging
+from . import api, db
+
+
+@click.group()
+@click.option('-l', '--log-level', default='DEBUG', type=click.Choice(
+    ['DEBUG', 'INFO', 'WARN', 'ERROR']))
+def betterapis_cli(log_level):
+    level = getattr(logging, log_level)
+    logging.basicConfig(level=level)
+
+@betterapis_cli.command()
+@click.option('-p', '--port', default=8080, type=click.IntRange(1,65535))
+@click.option('--debug/--no-debug', default=True)
+def run(port, debug):
+    api.run(port=port, debug=debug)
+    click.echo('Goodbye!')
+
+@betterapis_cli.command(name='reset-db')
+def reset_db():
+    click.confirm('This will ERASE the application databsase. Are you sure?',
+            abort=True)
+    click.echo('Dropping all tables...')
+    db.drop_all()
+    click.echo('Creating all tables...')
+    db.create_all()
+    click.echo('Database reset!')
+
+@betterapis_cli.command(name='create-tables')
+def create_tables():
+    click.echo('Creating all tables...')
+    db.create_all()
+    click.echo('Tables created.')
+
+

implementation/betterapis/controllers/__init__.py

@@ -0,0 +1,11 @@
+from flask import jsonify, url_for
+
+from betterapis import api
+
+
+def get():
+    # Get all of the mapped 'search' endpoints and list them as links
+    links = {rule.rule.lstrip('/'): url_for(rule.endpoint, _external=True) \
+                for rule in api.app.url_map.iter_rules() \
+                if rule.endpoint.endswith('_search')}
+    return jsonify(links=links)

implementation/betterapis/specs/betterapis.yaml

@@ -22,6 +22,13 @@ produces:
   - application/json
 
 paths:
+  /:
+    get:
+      summary: Get a list of endpoints
+      description: Lists all API endpoints with links
+      responses:
+        200:
+          description: Returns a list of endpoints.
   /talks:
     get:
       summary: Get a list of talks

implementation/setup.py

@@ -10,4 +10,9 @@
             'connexion',
             'flask_sqlalchemy'
             ],
+        entry_points={
+        'console_scripts':[
+            'betterapis=betterapis.cli:betterapis_cli',
+            ],
+        }
         )