from app import create_app, db import argparse from flask import render_template import os from app.utils.init_modules import create_modules_db parser = argparse.ArgumentParser() parser.add_argument("-i", "--init_db", help="Initialise the db if it not exist", action="store_true") parser.add_argument("-r", "--recreate_db", help="Delete and initialise the db", action="store_true") parser.add_argument("-d", "--delete_db", help="Delete the db", action="store_true") parser.add_argument("-m", "--create_module", help="Create modules in db", action="store_true") args = parser.parse_args() os.environ.setdefault('FLASKENV', 'development') app = create_app() @app.errorhandler(404) def error_page_not_found(e): return render_template('404.html'), 404 if args.init_db: with app.app_context(): db.create_all() elif args.recreate_db: with app.app_context(): db.drop_all() db.create_all() elif args.delete_db: with app.app_context(): db.drop_all() elif args.create_module: with app.app_context(): create_modules_db() else: app.run(host=app.config.get("FLASK_URL"), port=app.config.get("FLASK_PORT"))