args_parser.py

import argparse

def parse_args():
    # Initial parsing to get the method
    parser = argparse.ArgumentParser(description="Script to generate binarization thresholds.")
    parser.add_argument(
        '--method',
        type=str,
        choices=['lpt_x', 
                 'rdm_sampling', 
                 'tgt_sampling', 
                 'rf_classifier', 
                 'mlp_classifier', 
                 'single_fixed_thres',
                 'mean_pred_thres'
                 ],
        required=True,
        help="Choose one of the available options: lpt_x, rdm_sampling, tgt_sampling, rf_classifier, mlp_classifier, single_fixed_thres, mean_pred_thres"
    )
    parser.add_argument(
        '--species_set',
        type=str,
        choices=['iucn', 'snt', 'all', 'custom'],
        default='iucn',
        help="Choose the species set."
    )
    parser.add_argument("--model_path", type=str, required=True, help="Model path.")
    parser.add_argument("--exp_name", type=str, default='test', help="Experiment name")

    # First parse to get the method
    args, _ = parser.parse_known_args()

    # Conditionally add arguments based on the method
    if args.method == 'lpt_x':
        parser.add_argument(
            '--lpt_level',
            type=float,
            default=5.0,
            help="Specify the level for lpt-x method"
        )
    elif args.method == 'rdm_sampling':
        # Create a mutually exclusive group for raw_number and factor_presences
        rdm_group = parser.add_mutually_exclusive_group(required=True)
        
        rdm_group.add_argument(
            '--raw_number',
            type=int,
            default=None,
            help="Specify the raw number of absences to generate, e.g. 100, 1000, 10000..."
        )
        rdm_group.add_argument(
            '--factor_presences',
            type=float,
            default=None,
            help="Specify a factor (proportion) of the number of presences to generate absences."
        )
    elif args.method == 'single_fixed_thres':
        parser.add_argument(
            '--threshold',
            type=float,
            default=0.5,
            help="Specify the level for the single fixed threshold."
        )

    # Conditionally add species IDs if species_set is 'custom'
    if args.species_set == 'custom':
        parser.add_argument(
            '--species_ids',
            type=int,
            nargs='+',
            required=True,
            help="Specify the species IDs as a list of integers when 'custom' is chosen as the species set."
        )

    # Parse all arguments including the conditional ones
    args = parser.parse_args()

    print(args)

    return args

if __name__ == "__main__":
    args = parse_args()