1 files changed, 16 insertions, 7 deletions
diff --git a/megapixels/app/server/create.py b/megapixels/app/server/create.py
index 1119ee8f..9efed669 100644
--- a/megapixels/app/server/create.py
+++ b/megapixels/app/server/create.py
@@ -1,10 +1,8 @@
-from flask import Flask, Blueprint
+from flask import Flask, Blueprint, jsonify
 from flask_sqlalchemy import SQLAlchemy
-from app.models.sql_factory import connection_url
+from app.models.sql_factory import connection_url, load_sql_datasets
 
-from app.server.api import router as api_router
-
-# from app.server.views.assets import assets
+from app.server.api import api
 
 db = SQLAlchemy()
 
@@ -14,8 +12,10 @@ def create_app(script_info=None):
   app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 
   db.init_app(app)
-  app.register_blueprint(api)
-  
+  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')
@@ -24,4 +24,13 @@ def create_app(script_info=None):
   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