async def _clear_dbs(root):
# make sure to completely clear db before carrying on...
async for db in iter_databases(root):
storage = db.storage
if IPostgresStorage.providedBy(storage) or ICockroachStorage.providedBy(storage):
async with storage.pool.acquire() as conn:
await conn.execute('''
DELETE from {}
WHERE zoid != '{}' AND zoid != '{}'
'''.format(storage._objects_table_name, ROOT_ID, TRASHED_ID))
async def _clear_dbs(root):
# make sure to completely clear db before carrying on...
async for db in iter_databases(root):
storage = db.storage
if IPostgresStorage.providedBy(
storage) or ICockroachStorage.providedBy(storage):
async with storage.pool.acquire() as conn:
await conn.execute('''
DELETE from {}
WHERE zoid != '{}' AND zoid != '{}'
'''.format(storage._objects_table_name, ROOT_ID, TRASHED_ID))
async def run(self, arguments, settings, app):
script = os.path.abspath(arguments.script)
spec = importlib.util.spec_from_file_location("module.name", script)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
if not hasattr(module, "run"):
logger.warn(f"Not `async def run()` function found in file {script}")
return
sig = inspect.signature(module.run)
if "container" in sig.parameters:
async for db in iter_databases():
# preload dynamic dbs
pass
async for txn, tm, container in get_containers():
await module.run(container)
await tm.commit(txn=txn)
else:
await lazy_apply(module.run, app)
async def run(self, arguments, settings, app):
async for db in iter_databases():
await self.migrate(db)
async def run(self, arguments, settings, app):
async for db in iter_databases():
await self.migrate(db)
async def run(self, arguments, settings, app):
async for db in iter_databases():
try:
await self.vacuum(db)
except Exception: # pragma: no cover
logger.error("Error vacuuming db", exc_info=True)
async def get_banner(self):
db_ids = []
async for db in utils.iter_databases():
db_ids.append('- ' + db.id)
return self.banner.format('\n'.join(db_ids))
