def initialize(engine, create=False, drop=False):
global _session
_session = scoped_session(sessionmaker())
initialize_model(_session, engine)
if drop:
logger.warning("dropping all tables in {engine.url!s}".format(engine=engine))
Model.metadata.drop_all()
if create:
logger.info("creating tables in {engine.url!s}".format(engine=engine))
Model.metadata.create_all(engine)
return _session
def setUp(self):
self.engine = create_engine('sqlite://')
self.session = scoped_session(sessionmaker())
self.logger = logging.getLogger('batteries.tests')
initialize_model(self.session, self.engine)
warnings.filterwarnings('error')
warnings.filterwarnings(
'ignore',
r"^Dialect sqlite\+pysqlite does \*not\* support Decimal objects natively\, "
"and SQLAlchemy must convert from floating point - rounding errors and other "
"issues may occur\. Please consider storing Decimal numbers as strings or "
"integers on this platform for lossless storage\.$",
SAWarning, r'^sqlalchemy\.sql\.type_api$')
Model.metadata.create_all(self.engine)
def initialize(engine, create=False, drop=False):
global session
import scramble.models.base
session = scoped_session(sessionmaker(extension=ZopeTransactionExtension()))
initialize_model(session, engine)
if drop:
logger.warning("dropping all tables in {engine.url!s}".format(engine=engine))
Model.metadata.drop_all()
if create:
logger.info("creating tables in {engine.url!s}".format(engine=engine))
Model.metadata.create_all(engine)
return session
def initialize(e, create=False, drop=False, **kwargs):
global Session
global session
global engine
import bonanza.models.base
Session = sessionmaker(**kwargs)
engine = e
session = scoped_session(Session)
initialize_model(session, engine)
if drop:
logger.warning("dropping all tables in {engine.url!s}".format(engine=engine))
Model.metadata.drop_all()
if create:
logger.info("creating tables in {engine.url!s}".format(engine=engine))
Model.metadata.create_all(engine)
return session
