pds-2021-g2-nest/code/backend/nest_backend/__main__.py

"""
This is the runner for the server.
"""
import os
import werkzeug.middleware.proxy_fix
from .routes import *
from .database import Base, tables
import psycopg2
from .gestione import *
from flask_cors import CORS
from flask_jwt_extended import *
from .app import app
from .tweet_explorer.__main__ import search_repo_conditions, start_exploring
from .api_spec import spec
from .swagger import swagger_ui_blueprint, SWAGGER_URL

Base.init_app(app=app)
jwt = JWTManager(app)
cors = CORS(app)
app.config['CORS_HEADERS'] = 'Content-Type'

reverse_proxy_app = werkzeug.middleware.proxy_fix.ProxyFix(app=app, x_for=1, x_proto=0, x_host=1, x_port=0, x_prefix=0)
# Routes setup

app.add_url_rule("/doa", view_func=page_doa, methods=["GET", "POST"])
app.add_url_rule("/api/v1/login", view_func=page_login, methods=["POST"])
app.add_url_rule("/api/v1/users", view_func=page_users, methods=["GET", "POST"])
app.add_url_rule("/api/v1/users/<string:email>", view_func=page_user, methods=["GET", "PATCH", "DELETE"])
app.add_url_rule("/api/v1/repositories/", view_func=page_repositories, methods=["GET", "POST"])
app.add_url_rule("/api/v1/repositories/<int:rid>", view_func=page_repository, methods=["GET", "PATCH", "DELETE", "PUT"])
app.add_url_rule("/api/v1/repositories/<int:rid>/conditions", view_func=page_repository_conditions,
                 methods=["GET", "POST"])
app.add_url_rule("/api/v1/conditions/<int:cid>", view_func=page_condition, methods=["GET", "PATCH", "DELETE"])

app.register_error_handler(Exception, error_handler)
app.register_blueprint(swagger_ui_blueprint, url_prefix=SWAGGER_URL)


with app.test_request_context():
    print(" * Getting docs ready...")
    for fn_name in app.view_functions:
        if fn_name == 'static':
            continue
        view_fn = app.view_functions[fn_name]
        spec.path(view=view_fn)
    print(" * Docs have been compiled on http://127.0.0.1:5000/docs")


@app.route("/docs/swagger.json")
def create_swagger_doc():
    return jsonify(spec.to_dict())


if __name__ == "__main__":
    with app.app_context():
        Base.create_all(app=app)
        if not User.query.filter_by(isAdmin=True).all():
            Base.session.add(
                User(email="admin@admin.com", password=gen_password("password"), username="admin", isAdmin=True))
            Base.session.commit()
        if not Repository.query.filter_by(id=1).first():
            Base.session.add(Repository(id=1, name="Draghi", owner_id="admin@admin.com"))
            Base.session.commit()
        if not Condition.query.filter_by(id=1).first():
            Base.session.add(Condition(id=1, type=Enums.ConditionType.hashtag, content="draghi"))
            Base.session.commit()
        if not Condition.query.filter_by(id=2).first():
            Base.session.add(Condition(id=2, type=Enums.ConditionType.location, content="Modena"))
            Base.session.commit()
        if not Condition.query.filter_by(id=3).first():
            Base.session.add(Condition(id=3, type=Enums.ConditionType.time, content="2021-04-27"))
            Base.session.commit()
        if not Uses.query.filter_by(rid=1, cid=1).first():
            Base.session.add(Uses(rid=1, cid=1))
            Base.session.commit()
        if not Uses.query.filter_by(rid=1, cid=2).first():
            Base.session.add(Uses(rid=1, cid=2))
            Base.session.commit()
        if not Uses.query.filter_by(rid=1, cid=3).first():
            Base.session.add(Uses(rid=1, cid=3))
            Base.session.commit()
        debug = True
        if os.getenv("DISABLE_DEBUG"):
            debug = False

        #print(Repository.query.all()[0].to_json())
        #print(Condition.query.all()[0].content)
        #print(Uses.query.all()[0].cid, Uses.query.all()[0].rid)
        search_repo_conditions(1)
        #start_exploring()
    app.run(debug=debug)