from flask import Flask, Blueprint, jsonify
from flask_sqlalchemy import SQLAlchemy
from app.models.sql_factory import connection_url, load_sql_datasets

from app.server.api import api

db = SQLAlchemy()

def create_app(script_info=None):
  app = Flask(__name__, static_url_path='')
  app.config['SQLALCHEMY_DATABASE_URI'] = connection_url
  app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

  db.init_app(app)
  datasets = load_sql_datasets(replace=False, base_model=db.Model)

  app.register_blueprint(api, url_prefix='/api')

  @app.route('/', methods=['GET'])
  def index():
    return app.send_static_file('index.html')

  @app.shell_context_processor
  def shell_context():
    return { 'app': app, 'db': db }

  @app.route("/site-map")
  def site_map():
    links = []
    for rule in app.url_map.iter_rules():
      # url = url_for(rule.endpoint, **(rule.defaults or {}))
      # print(url)
      links.append((rule.endpoint))
    return(jsonify(links))

  return app