import argparse
import os
if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="An analyst for quick ML applications.",
                                     add_help=False)

    parser.add_argument('-icd', action='store', dest='CODES',default=
                    '780.93,733.00,530.85,530.81,493.90,333.94,327.23,288.60,278.00,275.42'  
                    ',250.00,250.40,729.1,585.9,571.8,564.1,473.9,472.0,428.9,331.0,285.9,277.7'
                    ',276.1,272.4,272.1,256.4,311',type=str, 
            help='Comma-separated list of icd-9 codes to run')
    parser.add_argument('-ml',action='store',dest='MLS', type=str, 
            default='LogisticRegression')
    parser.add_argument('--short',action='store_false',dest='LONG', default=True)
    parser.add_argument('--local',action='store_true',dest='LOCAL', default=False)
    parser.add_argument('--norare',action='store_true',dest='NORARE', default=False)
    parser.add_argument('-n_trials',action='store',dest='NTRIALS', default=1)
    parser.add_argument('-m',action='store',dest='M', default=16000)
    parser.add_argument('-cutoffs',action='store',dest='CUTOFFS', type=str, default='')
    parser.add_argument('-trials',action='store',dest='TRIALS', type=str, default='')
    parser.add_argument('-results',action='store',dest='RDIR', type=str, default='/project/moore/users/lacava/geis-ehr/results')
    parser.add_argument('-dir',action='store',dest='DIR', type=str, 
                        default='/project/moore/users/lacava/geis-ehr/phased/')
   
    args = parser.parse_args()
  
    if len(args.TRIALS)>0:
        args.NTRIALS = len(args.TRIALS.split(','))

    codes = args.CODES.split(',')
    cutoffs = args.CUTOFFS.split(',')

    q = 'moore_long' if args.LONG else 'moore_normal'

    if args.LOCAL:
        lpc_options = ''
    else:
        lpc_options = '--lsf -q {Q} -m {M} -n_jobs 1'.format(Q=q,M=args.M)
    
    for c in codes:
        for delay in cutoffs:
            if delay != '':
                dataset = args.DIR + 'icd9_' + c + '_cutoff' + delay
            else:
                dataset = args.DIR + 'icd9_' + c 

            if len(args.TRIALS)>0: 
                cmd = ('python analyze.py {DATA} -ml {ML} -n_trials {NTR} {RARE} '
                        '-trials {TR} {LPC}').format(DATA=dataset,
                                                    ML=args.MLS,
                                                    TR=args.TRIALS,
                                                    NTR=args.NTRIALS,
                                                    RARE='--norare' if args.NORARE else '',
                                                    LPC=lpc_options)
            else:
                cmd = ('python analyze.py {DATA} -ml {ML} -n_trials {NTR} {RARE} '
                        '-results {RDIR} {LPC}').format(DATA=dataset,
                                        ML=args.MLS,
                                        NTR=args.NTRIALS,
                                        RARE='--norare' if args.NORARE else '',
                                        RDIR=args.RDIR,
                                        LPC=lpc_options)
            print(cmd)
            os.system(cmd)