app.py

import threading

from dotenv import load_dotenv
from app import create_app, db
import argparse
from flask import render_template, request, Response
import json
import os


from app.features.rule.rule_format.utils_format.utils_import_update import delete_existing_repo_folder
from app.core.utils.init_db import create_admin, create_default_user, insert_default_formats, show_admin_first_connection




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")
args = parser.parse_args()



os.environ.setdefault('FLASKENV', 'development')

load_dotenv()

app = create_app()

@app.errorhandler(404)
def error_page_not_found(e):
    if request.path.startswith('/api/'):
        return Response(json.dumps({"status": "error", "reason": "404 Not Found"}, indent=2, sort_keys=True), mimetype='application/json'), 404
    return render_template('404.html'), 404
    

if args.init_db:
    with app.app_context():
        db.create_all()
        admin, raw_password = create_admin()
        editor = create_default_user()
        insert_default_formats()
        show_admin_first_connection(admin , raw_password)

elif args.recreate_db:
    with app.app_context():
        db.drop_all()
        db.create_all()
        delete_existing_repo_folder("Rules_Github")
        admin , raw_password = create_admin()
        insert_default_formats()
        show_admin_first_connection(admin , raw_password)

        editor = create_default_user()
elif args.delete_db:
    with app.app_context():
        db.drop_all()
        print("DB delete with success")
else:
    app.run(host=app.config.get("FLASK_URL"), port=app.config.get("FLASK_PORT"))