116 lines
3.4 KiB
Python
Executable File
116 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import logging
|
|
import code
|
|
|
|
import web
|
|
from sqlalchemy.orm import scoped_session, sessionmaker
|
|
from api.handlers import check_client_content_type
|
|
from api.models import engine
|
|
from db import load_db_driver, syncdb
|
|
from unit_test import TestRunner
|
|
from urls import urls
|
|
|
|
logging.basicConfig(level="DEBUG")
|
|
|
|
app = web.application(urls, locals())
|
|
app.add_processor(load_db_driver)
|
|
app.add_processor(check_client_content_type)
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers(
|
|
dest="action", help='actions'
|
|
)
|
|
run_parser = subparsers.add_parser(
|
|
'run', help='run application locally'
|
|
)
|
|
run_parser.add_argument(
|
|
'-p', '--port', dest='port', action='store', type=str,
|
|
help='application port', default='8000'
|
|
)
|
|
runwsgi_parser = subparsers.add_parser(
|
|
'runwsgi', help='run WSGI application'
|
|
)
|
|
runwsgi_parser.add_argument(
|
|
'-p', '--port', dest='port', action='store', type=str,
|
|
help='application port', default='8000'
|
|
)
|
|
test_parser = subparsers.add_parser(
|
|
'test', help='run unit tests'
|
|
)
|
|
syncdb_parser = subparsers.add_parser(
|
|
'syncdb', help='sync application database'
|
|
)
|
|
shell_parser = subparsers.add_parser(
|
|
'shell', help='open python REPL'
|
|
)
|
|
loaddata_parser = subparsers.add_parser(
|
|
'loaddata', help='load data from fixture'
|
|
)
|
|
loaddata_parser.add_argument(
|
|
'fixture', action='store', help='json fixture to load'
|
|
)
|
|
params, other_params = parser.parse_known_args()
|
|
sys.argv.pop(1)
|
|
|
|
if params.action == "syncdb":
|
|
logging.info("Syncing database...")
|
|
syncdb()
|
|
logging.info("Done")
|
|
elif params.action == "test":
|
|
logging.info("Running tests...")
|
|
TestRunner.run()
|
|
logging.info("Done")
|
|
elif params.action == "loaddata":
|
|
from fixtures import fixman
|
|
if os.path.exists(params.fixture):
|
|
logging.info("Uploading fixtures...")
|
|
with open(params.fixture, "r") as f:
|
|
fixman.upload_fixture(f)
|
|
logging.info("Done")
|
|
else:
|
|
parser.print_help()
|
|
elif params.action in ("run", "runwsgi"):
|
|
from rpc import threaded
|
|
import eventlet
|
|
eventlet.monkey_patch()
|
|
q = threaded.rpc_queue
|
|
rpc_thread = threaded.RPCThread()
|
|
|
|
if params.action == "run":
|
|
app = web.application(urls, locals(), autoreload=True)
|
|
else:
|
|
app = web.application(urls, locals())
|
|
|
|
app.add_processor(load_db_driver)
|
|
app.add_processor(check_client_content_type)
|
|
|
|
if params.action == "run":
|
|
sys.argv.insert(1, params.port)
|
|
app.run()
|
|
else:
|
|
logging.info("Running WSGI app...")
|
|
server = web.httpserver.WSGIServer(
|
|
("0.0.0.0", int(params.port)),
|
|
app.wsgifunc()
|
|
)
|
|
try:
|
|
rpc_thread.start()
|
|
server.start()
|
|
except KeyboardInterrupt:
|
|
logging.info("Stopping WSGI app...")
|
|
q.put("exit")
|
|
server.stop()
|
|
logging.info("Done")
|
|
elif params.action == "shell":
|
|
orm = scoped_session(sessionmaker(bind=engine))
|
|
code.interact(local={'orm': orm})
|
|
orm.commit()
|
|
else:
|
|
parser.print_help()
|