def main(): """ Calls importmail.py prior to importing to DB """ parser = argparse.ArgumentParser() parser.add_argument('--light', action='store_true', default=False, help='Import without downloading any new files') parser.add_argument('--username', default='') parser.add_argument('--password', default='') parser.add_argument('--lastmail', action='store_true', default=False) args = parser.parse_args() collection = retrieve_ims_xml() if not args.light: logging.info("Importing data from mail...") importmail.main(args.username, args.password, args.lastmail) united_path = "static/data/united/" total = 0 logging.info("Loading United accidents...") for united_file in os.listdir(united_path): if united_file.endswith(".csv"): total += import_to_db(collection, united_path + united_file) logging.info("\tImported {0} items".format(total)) update_db(collection)
def main(): """ Calls importmail.py prior to importing to DB """ parser = argparse.ArgumentParser() parser.add_argument('--light', action='store_true', default=False, help='Import without downloading any new files') parser.add_argument('--username', default='') parser.add_argument('--password', default='') parser.add_argument('--lastmail', action='store_true', default=False) args = parser.parse_args() if not args.light: importmail.main(args.username, args.password, args.lastmail) united_path = "static/data/united/" total = 0 for united_file in os.listdir(united_path): if united_file.endswith(".csv"): total += import_to_db(united_path + united_file) print("\tImported {0} items".format(total))