From 63cdfd8017d09001fedf0696170775d432debb42 Mon Sep 17 00:00:00 2001 From: Isaac Cook Date: Wed, 20 Aug 2014 14:22:16 -0500 Subject: [PATCH] Resolves #71 -- add more management shell context var preimports --- manage.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/manage.py b/manage.py index 4dc08df..c80ceac 100755 --- a/manage.py +++ b/manage.py @@ -256,14 +256,24 @@ def general_cleanup_cmd(): def make_context(): """ Setup a coinserver connection fot the shell context """ - app = _request_ctx_stack.top.app conn = AuthServiceProxy( "http://{0}:{1}@{2}:{3}/" .format(app.config['coinserv']['username'], app.config['coinserv']['password'], app.config['coinserv']['address'], app.config['coinserv']['port'])) - return dict(app=app, conn=conn) + import simplecoin + print( +"""The following variables are predefined/imported: +db: the sqlalchemy database instance +t: pre-imported simplecoin.tasks +m: pre-imported simplecoin.models +conn: The primary rpc connection\n""") + return dict(app=_request_ctx_stack.top.app, + conn=conn, + m=simplecoin.models, + t=simplecoin.scheduler, + db=db) manager.add_command("shell", Shell(make_context=make_context)) manager.add_command('db', MigrateCommand)