# submit.py

import sys
from optparse import OptionParser
from ProdSubmitter import ProdSubmitter

intask = ""
tag = ""
nevt = 0
clparser = OptionParser()
clparser.add_option("-d", "--debug", dest="debug",
                  help="debug level", default=0)
clparser.add_option("-f", "--data_format", dest="data_format",
                  help="data format (AOD, AOD.ESD, ...", default="")
clparser.add_option("-e", "--nevent", dest="nevent",
                  help="event count", default=0)
clparser.add_option("-g", "--group", dest="group",
                  help="physics group", default="")
clparser.add_option("-u", "--user", dest="user",
                  help="user name for submission", default="")
clparser.add_option("-s", "--submit", action="store_true", dest="submit",
                  help="true to submit (false)", default=False)
clparser.add_option("-c", "--check_url", dest="check_url",
                  help="URL for validation", default="")
(clopts, clargs) = clparser.parse_args()
data_format = clopts.data_format
nevt = int(clopts.nevent)
user = clopts.user
group = clopts.group
submit = clopts.submit
check_url = clopts.check_url
dbg = clopts.debug

# Read the other command line arguemnts.
narg = len(clargs)
if narg>1:
  intask = clargs[0]
  tag = clargs[1]
else:
  print "Usage: sys.argv[0] [-s USER] DS TAG"
  sys.exit(1)
quiet = "quiet" in clargs
show_submit_url = "show_submit_url" in clargs
show_response_html = "show_response_html" in clargs
show_response_text = "show_response_text" in clargs
show_submit_data = "show_submit_data" in clargs
if "submit" in clargs:
  submit = True

if len(intask):
  sub = ProdSubmitter(user, group, intask, tag, nevt, data_format)
  sub.dbg = dbg
  sub.html_configure_response_file = "configure_response.html"
  if not quiet:
    print "    Input task name: " + intask
    print " Transformation tag: " + tag
    #print "Configuring..."
  suburl = sub.get_url_submit()
  if not quiet:
    print "        Data format: " + sub.data_format
    print "    Events per file: " + str(sub.events_per_file())
    print "   First input file: " + str(sub.first_input_file())
    print "      # input files: " + str(sub.input_file_count())
    print "           # events: " + str(sub.event_count())
    print "      physics group: " + sub.physics_group_name()
  if show_submit_url:
    print suburl
  if show_submit_data:
    sub.show_submit_data()
if len(check_url):
  sub.check_url_submit(check_url)
if submit:
  if not quiet:
    print "Submitting..."
  resurl = sub.submit()
  print "Response: " + sub.text_submit_response
  if not quiet:
    if show_response_html:
      print sub.html_submit_response
  cnffilename = "configure.html"
  cnffile = open(cnffilename, "w")
  cnffile.write(sub.html_configure_response)
  cnffile.close()
  resfilename = "response.html"
  resfile = open(resfilename, "w")
  resfile.write(sub.html_submit_response)
  resfile.close()