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)