def localads_mwm_to_csv(): parser = argparse.ArgumentParser( description="Prepares CSV files for uploading to localads database " "from mwm files.") parser.add_argument("mwm", help="path to mwm files") parser.add_argument( "--osm2ft", help="path to osm2ft files (default is the same as mwm)") parser.add_argument("--output", default=".", help="path to generated files ('.' by default)") types_default = os.path.join(os.path.dirname(__file__), "..", "..", "..", "data", "types.txt") parser.add_argument("--types", default=types_default, help="path to omim/data/types.txt") parser.add_argument("--threads", type=int, default=1, help="number of threads to process files") parser.add_argument("--mwm_version", type=int, required=True, help="Mwm version") args = parser.parse_args(sys.argv[2:]) if not args.osm2ft: args.osm2ft = args.mwm create_csv(args.output, args.mwm, args.osm2ft, args.types, args.mwm_version, args.threads)
def stage_localads(env): create_csv(env.localads_path, env.mwm_path, env.mwm_path, env.types_path, env.mwm_version, multiprocessing.cpu_count()) with tarfile.open(f"{env.localads_path}.tar.gz", "w:gz") as tar: for filename in os.listdir(env.localads_path): tar.add(os.path.join(env.localads_path, filename), arcname=filename)
def stage_localads(env): create_csv(env.localads_path, env.mwm_path, env.mwm_path, env.types_path, env.mwm_version, multiprocessing.cpu_count()) make_tarfile(f"{env.localads_path}.tar.gz", env.localads_path)