def check_version(ev):
""" ApplicationCreated event handler """
if not Version.__table__.exists():
return
versions = dict((v.package, v.version_num)
for v in ptah.get_session().query(Version).all())
packages = ptah.get_cfg_storage(MIGRATION_ID).keys()
has_steps = False
log = logging.getLogger('ptah.alembic')
for pkg in packages:
version = versions.get(pkg)
script = ScriptDirectory(pkg)
for sc in script.walk_revisions():
if sc.is_head:
if sc.revision != version:
has_steps = True
log.error("Package '%s' current revision: '%s', head: '%s'",
pkg, version, sc.revision)
break
if has_steps:
config.shutdown()
log.error("Please run `ptah-migrate` script. Stopping...")
raise SystemExit(1)
def check_version(ev):
""" ApplicationCreated event handler """
if not Version.__table__.exists():
return
versions = dict((v.package, v.version_num)
for v in ptah.get_session().query(Version).all())
packages = ptah.get_cfg_storage(MIGRATION_ID).keys()
has_steps = False
log = logging.getLogger('ptah.alembic')
for pkg in packages:
version = versions.get(pkg)
script = ScriptDirectory(pkg)
for sc in script.walk_revisions():
if sc.is_head:
if sc.revision != version:
has_steps = True
log.error(
"Package '%s' current revision: '%s', head: '%s'", pkg,
version, sc.revision)
break
if has_steps:
config.shutdown()
log.error("Please run `ptah-migrate` script. Stopping...")
raise SystemExit(1)
def ptah_initialize(cfg):
""" Initialize ptah package."""
from pyramid.exceptions import ConfigurationExecutionError
cfg.begin()
try:
auth_service.set_userid(SUPERUSER_URI)
# initialize settings
init_settings(cfg, cfg.registry.settings)
except config.StopException:
config.shutdown()
raise
finally:
cfg.end()
def tearDown(self):
config.shutdown()
config.cleanup_system(self.__class__.__module__)
sm = self.p_config
sm.__init__('base')
testing.tearDown()
def tearDown(self):
config.shutdown()
Base.tearDown(self)
shutil.rmtree(self.dir1)
shutil.rmtree(self.dir2)
