#!/usr/bin/env python

"""
    -u
        username
    -p
        password
    -a
        appid
    -t
        toolspec directory
    -s
        rest service url
    -n
        number of times to submit toolspec
    -w
        wait for submissions to finish
    -v 
        validate only, don't submit
"""
import sys
import os
import re
import string
import subprocess
import tempfile
import getopt
import time
import requests
import xml.etree.ElementTree as ET
import pycipres.jobtemplate as JT
import pycipres.appmanage as AM
import traceback


def main(argv=None):
    if argv is None:
        argv = sys.argv

    user = password = appid = testspec = service = jobcount = None
    waitForCompletion = False
    validateOnly = False

    # Get default url, user, etc,  from our config file
    app = AM.AppManagement()
    defaults = app.getDefaultProperties()
    user = defaults.getProperty("PYCIPRES_EU")
    password = defaults.getProperty("TESTPASS")
    appid = defaults.getProperty("PYCIPRES_APPID")
    service = defaults.getProperty("URL") + "/job/"
    # defaultTestDataRoot=defaults.getProperty("TESTDATA")

    # Some command line may override defaults.  Others have no default in config file.
    options, remainder = getopt.getopt(argv[1:], "u:p:a:t:s:n:whv")
    for opt, arg in options:
        if opt in ("-u"):
            user = arg
        elif opt in ("-p"):
            password = arg
        if opt in ("-t"):
            testspec = arg
        elif opt in ("-s"):
            service = arg
        elif opt in ("-n"):
            jobcount = int(arg)
        elif opt in ("-w"):
            waitForCompletion = True
        elif opt in ("-v"):
            validateOnly = True
        elif opt in ("-a"):
            appid = int(arg)
        elif opt in ("-h"):
            print __doc__
            return 0
    if not (user and password and testspec and service and jobcount and appid):
        print __doc__
        return -1
    
    jobs = []
    for i in range(jobcount):
        tj = JT.JobTemplate(
            baseurl=service, 
            user=user,
            credentials = (user, password),
            appid=appid,
            testdir = testspec,
            validateOnly = validateOnly, 
        )

        # If job parameters are invalid throws an exception with list of errors.
        # TODO: We need to parse and display the errors.  Do parsing JobTemplate class
        try:
            tj.doPost()
            if validateOnly:
                print "Job is OK, command line is '%s'" % (tj.commandline)
            else:
                jobs.append(tj)
                print "Submitted %s" % (tj.jobhandle)
            sys.stdout.flush()
        except Exception, e:
            traceback.print_exc()
            # print e

            
    
    if waitForCompletion and not validateOnly:
        if len(jobs):
            print "Submitted and waiting for %d jobs" % ( len(jobs) ) 
        while True: 

            unfinishedJobs = 0

            # jobs[:] makes a copy of the list in jobs.  You need to work on a copy if
            # you want to remove items from a list you're iterating over.
            for job in jobs[:]:
                if job.jobIsTerminal:
                    print "%s finished. %s. Results:" % (job.jobhandle, ('OK', 'FAILED')[job.isFailed])
                    for name in job.resultFiles:
                        print name, ' is at ', job.resultFiles[name]
                    sys.stdout.flush()
                    jobs.remove(job)
                    continue
                else:
                    job.updateJobStatus()
                    if job.jobIsTerminal:
                        print "%s finished. %s. Results:" % (job.jobhandle, ('OK', 'FAILED')[job.isFailed])
                        for name in job.resultFiles:
                            print name, ' is at ', job.resultFiles[name]
                        sys.stdout.flush()
                        jobs.remove(job)
                        continue
                    unfinishedJobs += 1

            if not unfinishedJobs:
                break;
            time.sleep(30)

if __name__ == "__main__":
    sys.exit(main())