def _main():
parser = _make_parser()
args = parser.parse_args()
if args.debug:
logger.setLevel(logging.DEBUG)
from indra_db.databases import logger as db_logger
db_logger.setLevel(logging.DEBUG)
print("Getting %s database." % args.database)
db = get_db(args.database)
assert db is not None
db.grab_session()
s3_cache = S3Path.from_string(args.cache)
pa = DbPreassembler(args.batch,
s3_cache,
stmt_type=args.stmt_type,
yes_all=args.yes_all)
desc = 'Continuing' if args.continuing else 'Beginning'
print("%s to %s preassembled corpus." % (desc, args.task))
if args.task == 'create':
pa.create_corpus(db, args.continuing)
elif args.task == 'update':
pa.supplement_corpus(db, args.continuing)
else:
raise IndraDBPreassemblyError('Unrecognized task: %s.' % args.task)
def wrapper(*args, **kwargs):
# Make the database manager logger quieter.
from indra_db.databases import logger as dbm_logger
original_level = dbm_logger.level
dbm_logger.setLevel(logging.WARNING)
try:
ret = func(*args, **kwargs)
finally:
# Make sure we go back to ordinary logging
dbm_logger.setLevel(original_level)
return ret
def _main():
parser = _make_parser()
args = parser.parse_args()
if args.debug:
logger.setLevel(logging.DEBUG)
from indra_db.databases import logger as db_logger
db_logger.setLevel(logging.DEBUG)
print("Getting %s database." % args.database)
db = get_db(args.database)
assert db is not None
db.grab_session()
pm = PreassemblyManager(args.num_procs, args.batch)
desc = 'Continuing' if args.continuing else 'Beginning'
print("%s to %s preassembled corpus." % (desc, args.task))
if args.task == 'create':
pm.create_corpus(db, args.continuing)
elif args.task == 'update':
pm.supplement_corpus(db, args.continuing)
else:
raise IndraDBPreassemblyError('Unrecognized task: %s.' % args.task)