Run subprocess future asynchronously.

tasks.py

@@ -1,6 +1,5 @@
 import os
 import logging
-import functools
 import concurrent.futures
 
 from invoke import task, run
@@ -42,7 +41,6 @@ def apify(file_name, table_name=None, primary_name='id', insert=True):
 @task
 def serve(host='0.0.0.0', port=5000, debug=False):
     from sandman import app
-    from sandman.model import activate
     app.json_encoder = utils.APIJSONEncoder
     app.config['SERVER_PORT'] = port
     app.config['SQLALCHEMY_DATABASE_URI'] = config.SQLA_URI
@@ -61,10 +59,13 @@ def protect_admin():
     blueprint = aws.make_blueprint()
     app.register_blueprint(blueprint)
 
-    # Load bucket in a separate process, then activate sandman in the main process
-    with concurrent.futures.ProcessPoolExecutor() as pool:
-        future = pool.submit(aws.fetch_bucket)
-        callback = functools.partial(activate, base=utils.ReadOnlyModel, browser=False)
-        future.add_done_callback(callback)
+    # Activate sandman with existing models
+    utils.activate(admin=True)
+
+    # Load bucket in a separate process, then re-activate sandman in the main process
+    pool = concurrent.futures.ProcessPoolExecutor()
+    future = pool.submit(aws.fetch_bucket)
+    future.add_done_callback(utils.activate)
+    pool.shutdown(wait=False)
 
     app.run(host=host, port=port, debug=debug)

utils.py

@@ -4,8 +4,8 @@
 import sqlalchemy as sa
 
 from sandman import db
-from sandman.model import activate
 from sandman.model.models import Model
+from sandman.model import activate as sandman_activate
 
 import config
 
@@ -21,6 +21,9 @@ def default(self, o):
             return o.isoformat()
         return super(APIJSONEncoder, self).default(o)
 
+def activate(base=ReadOnlyModel, browser=False, admin=False, **kwargs):
+    sandman_activate(base=base, browser=browser, admin=admin, **kwargs)
+
 def drop_table(tablename, metadata=None, engine=None):
     metadata = metadata or sa.MetaData()
     engine = engine or sa.create_engine(config.SQLA_URI)