53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
from app import create_app, db
|
|
from flask.ext.script import Manager, Shell, Command, Option
|
|
|
|
|
|
# http://stackoverflow.com/a/24606817
|
|
class GunicornServer(Command):
|
|
"""Run the app within Gunicorn"""
|
|
|
|
def get_options(self):
|
|
from gunicorn.config import make_settings
|
|
|
|
settings = make_settings()
|
|
options = (
|
|
Option(*klass.cli, action=klass.action)
|
|
for setting, klass in settings.iteritems() if klass.cli
|
|
)
|
|
return options
|
|
|
|
def run(self, *args, **kwargs):
|
|
from gunicorn.app.wsgiapp import WSGIApplication
|
|
|
|
app = WSGIApplication()
|
|
app.app_uri = 'manage:app'
|
|
return app.run()
|
|
|
|
|
|
app = create_app(os.getenv('FLASK_CONFIG') or 'default')
|
|
manager = Manager(app)
|
|
|
|
|
|
@manager.command
|
|
def initdb():
|
|
from sqlalchemy.sql import text
|
|
cmd = """
|
|
CREATE TABLE new_communities (
|
|
id serial NOT NULL,
|
|
name character varying(50) NOT NULL,
|
|
northing double precision NOT NULL,
|
|
easting double precision NOT NULL,
|
|
latitude double precision NOT NULL,
|
|
longitude double precision NOT NULL,
|
|
data jsonb NOT NULL,
|
|
CONSTRAINT new_communities_pkey PRIMARY KEY (id)
|
|
);"""
|
|
_ = db.engine.execute(text(cmd))
|
|
|
|
manager.add_command("gunicorn", GunicornServer())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
manager.run()
|