def parse_args()

in tools/url-checker/create_test_files.py [0:0]


def parse_args():
    """Parse command-line arguments."""
    parser = argparse.ArgumentParser(
        description="Create test files for the URL checker with various file types."
    )
    parser.add_argument(
        "--dir", 
        default="test_files",
        help="Directory where test files will be created (relative to script location)"
    )
    parser.add_argument(
        "--clean", 
        action="store_true", 
        help="Remove existing test files before creating new ones"
    )
    parser.add_argument(
        "--file-count",
        type=int,
        default=5, 
        help="Base number of files to create per type (some types may have more/fewer)"
    )
    parser.add_argument(
        "--complexity",
        type=int,
        default=3,
        choices=[1, 2, 3, 4, 5],
        help="Complexity level of directory structure (1=simple, 5=very complex)"
    )
    return parser.parse_args()