def _run(args):
_setup(args)
formatter = logging.Formatter("%(asctime)s %(name)s - %(levelname)s %(filename)s:%(lineno)s: %(message)s")
root = logging.getLogger('')
handler = logging.StreamHandler()
handler.setFormatter(formatter)
root.addHandler(handler)
dbi = DBMgr.getInstance(max_disconnect_poll=40)
dbi.startRequest()
info = HelperMaKaCInfo.getMaKaCInfoInstance()
useRBDB = info.getRoomBookingModuleActive()
if useRBDB:
DALManager.connect()
sm = SchedulerModule.getDBInstance()
t = sm.getTaskById(args.taskid)
t.plugLogger(logging.getLogger('console.run/%s' % args.taskid))
t.run()
if useRBDB:
DALManager.commit()
DALManager.disconnect()
dbi.endRequest()
def _run(args):
_setup(args)
formatter = logging.Formatter(
"%(asctime)s %(name)s - %(levelname)s %(filename)s:%(lineno)s: %(message)s"
)
root = logging.getLogger('')
handler = logging.StreamHandler()
handler.setFormatter(formatter)
root.addHandler(handler)
dbi = DBMgr.getInstance(max_disconnect_poll=40)
dbi.startRequest()
info = HelperMaKaCInfo.getMaKaCInfoInstance()
useRBDB = info.getRoomBookingModuleActive()
if useRBDB:
DALManager.connect()
sm = SchedulerModule.getDBInstance()
t = sm.getTaskById(args.taskid)
t.plugLogger(logging.getLogger('console.run/%s' % args.taskid))
t.run()
if useRBDB:
DALManager.commit()
DALManager.disconnect()
dbi.endRequest()
def runMigration(withRBDB=False,
prevVersion=parse_version(__version__),
specified=[],
dry_run=False,
run_from=None):
global MIGRATION_TASKS
if not dry_run:
print "\nExecuting migration...\n"
dbi = DBMgr.getInstance()
print "Probing DB connection...",
# probe DB connection
dbi.startRequest()
dbi.endRequest(False)
print "DONE!\n"
if run_from:
try:
mig_tasks_names = list(t.__name__
for (__, t, __, __) in MIGRATION_TASKS)
mti = mig_tasks_names.index(run_from)
MIGRATION_TASKS = MIGRATION_TASKS[mti:]
except ValueError:
print console.colored(
"The task {0} does not exist".format(run_from), 'red')
return 1
# go from older to newer version and execute corresponding tasks
for version, task, always, never in MIGRATION_TASKS:
if never and task.__name__ not in specified:
continue
if specified and task.__name__ not in specified:
continue
if parse_version(version) > prevVersion or always:
print console.colored("#", 'green', attrs=['bold']), \
task.__doc__.replace('\n', '').replace(' ', '').strip(),
print console.colored("(%s)" % version, 'yellow')
if dry_run:
continue
dbi.startRequest()
if withRBDB:
DALManager.connect()
task(dbi, withRBDB, prevVersion)
if withRBDB:
DALManager.commit()
dbi.endRequest()
print console.colored(" DONE\n", 'green', attrs=['bold'])
if not dry_run:
print console.colored("Database Migration successful!\n",
'green',
attrs=['bold'])
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--location', '-l', action='append', dest='locations')
args = parser.parse_args()
with DBMgr.getInstance().global_connection():
DALManager.connect()
try:
_main(args)
finally:
DALManager.disconnect()
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--location', '-l', action='append', dest='locations')
args = parser.parse_args()
with DBMgr.getInstance().global_connection():
DALManager.connect()
try:
_main(args)
finally:
DALManager.disconnect()
def runMigration(withRBDB=False, prevVersion=parse_version(__version__),
specified=[], dry_run=False, run_from=None):
global MIGRATION_TASKS
if not dry_run:
print "\nExecuting migration...\n"
dbi = DBMgr.getInstance()
print "Probing DB connection...",
# probe DB connection
dbi.startRequest()
dbi.endRequest(False)
print "DONE!\n"
if run_from:
try:
mig_tasks_names = list(t.__name__ for (__, t, __, __) in MIGRATION_TASKS)
mti = mig_tasks_names.index(run_from)
MIGRATION_TASKS = MIGRATION_TASKS[mti:]
except ValueError:
print console.colored("The task {0} does not exist".format(run_from), 'red')
return 1
# go from older to newer version and execute corresponding tasks
for version, task, always, never in MIGRATION_TASKS:
if never and task.__name__ not in specified:
continue
if specified and task.__name__ not in specified:
continue
if parse_version(version) > prevVersion or always:
print console.colored("#", 'green', attrs=['bold']), \
task.__doc__.replace('\n', '').replace(' ', '').strip(),
print console.colored("(%s)" % version, 'yellow')
if dry_run:
continue
dbi.startRequest()
if withRBDB:
DALManager.connect()
task(dbi, withRBDB, prevVersion)
if withRBDB:
DALManager.commit()
dbi.endRequest()
print console.colored(" DONE\n", 'green', attrs=['bold'])
if not dry_run:
print console.colored("Database Migration successful!\n",
'green', attrs=['bold'])
def runMigration(withRBDB=False, prevVersion=parse_version(__version__),
specified=[], dryRun=False):
if not dryRun:
print "\nExecuting migration...\n"
dbi = DBMgr.getInstance()
print "Probing DB connection...",
# probe DB connection
dbi.startRequest()
dbi.endRequest(False)
print "DONE!\n"
# go from older to newer version and execute corresponding tasks
for version, task, always, never in MIGRATION_TASKS:
if never and task.__name__ not in specified:
continue
if specified and task.__name__ not in specified:
continue
if parse_version(version) > prevVersion or always:
print console.colored("#", 'green', attrs=['bold']), \
task.__doc__.replace('\n', '').replace(' ', '').strip(),
print console.colored("(%s)" % version, 'yellow')
if dryRun:
continue
dbi.startRequest()
if withRBDB:
DALManager.connect()
task(dbi, withRBDB, prevVersion)
if withRBDB:
DALManager.commit()
dbi.endRequest()
print console.colored(" DONE\n", 'green', attrs=['bold'])
if not dryRun:
print console.colored("Database Migration successful!\n",
'green', attrs=['bold'])
def runMigration(withRBDB=False,
prevVersion=parse_version(__version__),
specified=[]):
print "\nExecuting migration...\n"
dbi = DBMgr.getInstance()
print "Probing DB connection...",
# probe DB connection
dbi.startRequest()
dbi.endRequest(False)
print "DONE!\n"
# go from older to newer version and execute corresponding tasks
for version, task, always in MIGRATION_TASKS:
if specified and task.__name__ not in specified:
continue
if parse_version(version) > prevVersion or always:
print console.colored("#", 'green', attrs=['bold']), \
task.__doc__.replace('\n', '').strip(),
print console.colored("(%s)" % version, 'yellow')
dbi.startRequest()
if withRBDB:
DALManager.connect()
task(dbi, withRBDB, prevVersion)
if withRBDB:
DALManager.commit()
dbi.endRequest()
print console.colored(" DONE\n", 'green', attrs=['bold'])
print console.colored("Database Migration successful!\n",
'green',
attrs=['bold'])
