def setup_package(self):
config = dict()
config['SQLALCHEMY_DATABASE_URI'] = SQLALCHEMY_DATABASE_URI
config['SQLALCHEMY_ECHO'] = SQLALCHEMY_ECHO
config['SQLALCHEMY_RECORD_QUERIES'] = SQLALCHEMY_RECORD_QUERIES
config['DEBUG'] = DEBUG
db.init_connection(config)
db.drop_all()
db.create_all()
create_fixtures()
def default(self):
connection = self.pargs.connection
db.init_connection({
'SQLALCHEMY_DATABASE_URI': connection,
'SQLALCHEMY_ECHO': True
})
try:
schema = self.pargs.schema
drop = 'DROP SCHEMA IF EXISTS "%s" CASCADE;'
drop %= schema
db.session.execute(drop)
db.session.execute('CREATE SCHEMA "%s";' % schema)
db.session.execute('SET search_path TO "%s";' % schema)
db.session.execute(self.sql())
db.session.commit()
finally:
db.session.remove()
def default(self):
connection = self.pargs.connection
db.init_connection({
'SQLALCHEMY_DATABASE_URI': connection,
'SQLALCHEMY_ECHO': True
})
try:
schema = self.pargs.schema
drop = 'DROP SCHEMA IF EXISTS "%s" CASCADE;'
drop %= schema
db.session.execute(drop)
db.session.execute('CREATE SCHEMA "%s";' % schema)
db.session.execute('SET search_path TO "%s";' % schema)
db.session.execute(self.sql())
db.session.commit()
finally:
db.session.remove()
def default(self):
if self.pargs.list:
self.list_migrations()
return
if not self.pargs.migration:
print "Please select either --list or --migration <migration name>"
return
connection = self.pargs.connection
db.init_connection({
'SQLALCHEMY_DATABASE_URI': connection,
'SQLALCHEMY_ECHO': True
})
try:
schema = self.pargs.schema
db.session.execute('SET search_path TO "%s";' % schema)
db.session.execute(self.migration(self.pargs.migration))
db.session.commit()
finally:
db.session.remove()
def default(self):
if self.pargs.list:
self.list_migrations()
return
if not self.pargs.migration:
print "Please select either --list or --migration <migration name>"
return
connection = self.pargs.connection
db.init_connection({
'SQLALCHEMY_DATABASE_URI': connection,
'SQLALCHEMY_ECHO': True
})
try:
schema = self.pargs.schema
db.session.execute('SET search_path TO "%s";' % schema)
db.session.execute(self.migration(self.pargs.migration))
db.session.commit()
finally:
db.session.remove()
def init_database(app):
db.init_connection(app.config)
app.teardown_appcontext(lambda r: db.session.remove())
def init_database(app):
db.init_connection(app.config)
app.teardown_appcontext(lambda r: db.session.remove())
DIR = os.path.dirname(__file__)
DIR = os.path.abspath(DIR)
settings = os.environ.get('POOLDREST_CONFIG')
settings = settings or 'pooldrest.settings.dev'
# Create app
app = Flask(__name__)
# Configure the app
app.config.from_object('pooldrest.settings.base')
app.config.from_object(settings)
# Initialize database
db.init_connection(app.config)
# TODO: Register all application blueprint
# Rollback any pending transaction, close any open cursors and close any
# open connections when the application is torn down.
@app.teardown_appcontext
def shutdown_session(response):
db.session.remove()
@app.route('/')
def index():
return 'Hello World!'
