api_v1.4/bootstrap.py

"""
Main
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2021 Project Author name guillermo.gutierrezmorote@concordia.ca
Project Collaborator name peteryefi@gmail.com
"""

import datetime

import flask
import yaml
from pathlib import Path
from flasgger import LazyJSONEncoder, Swagger
from flask import Response
from flask_restful import Api
import threading

import hub_api.helpers.session_helper as sh
from hub_api.control.session import SessionStart, SessionEnd, KeepSessionAlive
from hub_api.control.uptime import Uptime
from hub_api.persistence.retrofit_results import RetrofitResults
from hub_api.workflow.insel_montly_energy_balance import InselMonthlyEnergyBalance
from hub_api.workflow.costs import Costs
from hub_api.workflow.energy_plus import EnergyPlus
from hub_api.energy_plus.idf_generator import IdfGenerator


sh.begin_time = datetime.datetime.now()
app = flask.Flask('cerc_api')
app.json_provider_class = LazyJSONEncoder
api = Api(app)

api.add_resource(Uptime, '/v1.4/uptime')

# Session
api.add_resource(SessionStart, '/v1.4/session/start')
api.add_resource(SessionEnd, '/v1.4/session/end')
api.add_resource(KeepSessionAlive, '/v1.4/session/keep-alive')

# persistence
api.add_resource(RetrofitResults, '/v1.4/persistence/retrofit-results')

# energy plus
api.add_resource(IdfGenerator, '/v1.4/energy-plus/idf-generator')

# workflows
api.add_resource(Costs, '/v1.4/workflow/costs')
api.add_resource(EnergyPlus, '/v1.4/workflow/energy-plus')
api.add_resource(InselMonthlyEnergyBalance,
                 '/v1.4/workflow/insel-monthly-energy-balance')

yml_path = Path('./hub/docs/openapi-specs.yml').resolve()

with open(yml_path, "r") as stream:
    swagger_config = {
        "headers": [],
        "specs": [
            {
                "endpoint": '/api/apispec',
                "route": '/api/apispec/apispec.json',
                "rule_filter": lambda rule: True,  # all in
                "model_filter": lambda tag: True,  # all in
            }
        ],
        "static_url_path": "/api/static",
        "specs_route": "/api/api-docs/",
        "openapi": "3.0.0"
    }
    try:
        Swagger(app, template=yaml.safe_load(stream), config=swagger_config)
    except yaml.YAMLError as exc:
        print(f'error: {exc}')


@app.route("/")
def home():
    return Response(headers={'Access-Control-Allow-Origin': '*'})


sh.debug_mode = True

threading.Thread(target=sh.expired_sessions_collector,
                 daemon=True, args="5").start()
app.run(port=15789, host="0.0.0.0", debug=sh.debug_mode)