def main()

in components/processing/libs/processor-msg/src/processors/msg/msg_generator.py [0:0]

• 26 lines of code
• 2 McCabe index (conditional complexity)

def main():
    load_dotenv()

    parser = argparse.ArgumentParser(
        prog="msg_generator",
        description="Generate .msg files for parsing test purposes",
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
    )
    parser.add_argument(
        "--output_dir",
        default=f"gs://{os.getenv('GCS_INPUT_BUCKET')}/input",
        type=str,
        help="Output directory for .msg files",
    )
    parser.add_argument(
        "--count", type=int, default=1, help="Count of .msg files to produce"
    )
    parser.add_argument(
        "--name-prefix",
        type=str,
        default=f"gen-{uuid.uuid1()}",
        help="Prefix of filename",
    )

    args = parser.parse_args()

    generator = MSGGenerator()
    for i in range(args.count):
        generator.save(GCSPath(f"{args.output_dir}/{args.name_prefix}-{i:08d}.msg"))