def import_log_to_database(): parser = argparse.ArgumentParser(description="Import Log-Files into Database.") parser.add_argument('database', help="Database to import to") parser.add_argument('files', metavar='File', type=argparse.FileType('r'), nargs='+') parser.add_argument('--import-parts', help="import raw parts", action="store_true") args = parser.parse_args() if args.import_parts: print("also adding parts!") else: print("not adding parts") # open database session = setup_connection(create_db=True, path=args.database) # add files for f in args.files: print("parsing " + f.name) tmp = read_from_file(f) incident = parse_incident(session, tmp[0], tmp[1], include_parts=args.import_parts) print("adding " + f.name + " to db") session.add(incident) session.commit() # close database session.close()
def tmp(f): return parse_incident(read_from_file(open(f, 'r')))