##
##

#
# Port of requestAT code from AWIPS1
#
#    
#     SOFTWARE HISTORY
#    
#    Date            Ticket#       Engineer       Description
#    ------------    ----------    -----------    --------------------------
#    02/06/13        1447          dgilling       Initial Creation.
#    01/24/14        2504          randerso       change to use iscUtil.getLogger for consistency
#    05/15/14        #3157         dgilling       Support multiple TPC and SPC sites.
#    03/10/2015      #4129         randerso       Refactored server selection code into a reusable method
#    04/28/2015      #4027         randerso       Expunged Calendar from ActiveTableRecord
# 
#

##
# This is a base file that is not intended to be overridden.
##



import cPickle
import os
import sys
import tempfile
import time
from xml.etree import ElementTree

import IrtAccess
import siteConfig
import VTECPartners
import iscUtil
import JUtil

from com.raytheon.uf.common.activetable import ActiveTableMode
from com.raytheon.uf.common.time.util import TimeUtil
from com.raytheon.uf.edex.activetable import ActiveTable
from com.raytheon.uf.common.activetable import VTECPartners as JavaVTECPartners



logger = None

def init_logging():
    import logging
    global logger
    logger = iscUtil.getLogger("requestAT", logLevel=logging.INFO)

def execute_request_at(serverHost, serverPort, serverProtocol, mhsid, siteID, ancf, 
                         bncf, xmtScript):
    #--------------------------------------------------------------------
    # Create a message - pickled
    # (MHSsiteID, mySiteID, listOfVTECMergeSites, countDict, issueTime)
    # Note that VTEC_MERGE_SITES does not contain our site or SPC, TPC.
    #--------------------------------------------------------------------
    
    # determine my 4 letter siteid
    if siteID in ['SJU']:
        mysite4 = "TJSJ"
    elif siteID in ['AFG','AJK','HFO','GUM']:
        mysite4 = "P" + siteID
    elif siteID in ['AER','ALU']:
        mysite4 = "PAFC"
    else:
        mysite4 = "K" + siteID
    otherSites = [mysite4]
    tpcSites = JUtil.javaObjToPyVal(JavaVTECPartners.getInstance(siteID).getTpcSites())
    spcSites = JUtil.javaObjToPyVal(JavaVTECPartners.getInstance(siteID).getSpcSites())
    otherSites.extend(tpcSites)
    otherSites.extend(spcSites)

    # determine the MHS WMO id for this message
    wmoid = "TTAA00 " + mysite4
    wmoid += " " + time.strftime("%d%H%M", time.gmtime(time.time()))

    # connect to ifpServer and retrieve active table
    actTab = ActiveTable.getActiveTable(mysite4, ActiveTableMode.OPERATIONAL)

    # analyze active table to get counts
    countDict = {}
    issueTime = 0
    for i in xrange(actTab.size()):
        rec = actTab.get(i)
        # only care about our own sites that we merge
        if rec.getOfficeid() not in VTECPartners.VTEC_MERGE_SITES and \
          rec.getOfficeid() not in otherSites:
            continue

        recIssueTime = float(rec.getIssueTime().getTime() / TimeUtil.MILLIS_PER_SECOND)
        #track latest
        issueTime = max(recIssueTime, issueTime)

        cnt = countDict.get(rec.getOfficeid(), 0)  #get old count
        countDict[rec.getOfficeid()] = cnt + 1

    data = (mhsid, siteID, VTECPartners.VTEC_MERGE_SITES, countDict, issueTime)
    logger.info("Data: " + repr(data))

    tempdir = os.path.join(siteConfig.GFESUITE_HOME, "products", "ATBL")
    with tempfile.NamedTemporaryFile(suffix='.reqat', dir=tempdir, delete=False) as fp:
        fname = fp.name
        buf = cPickle.dumps(data)
        fp.write(buf)
    
    sourceServer = {'mhsid'   : mhsid, 
                    'host'    : serverHost, 
                    'port'    : serverPort,
                    'protocol': serverProtocol, 
                    'site'    : siteID}

    destSites = VTECPartners.VTEC_TABLE_REQUEST_SITES
    if not destSites:
        raise Exception('No destSites defined for VTEC_TABLE_REQUEST_SITES')
    
    irt = IrtAccess.IrtAccess(ancf, bncf, logger=logger)
    msgSendDest, xml = irt.createDestinationXML(destSites, sourceServer)

    # create the XML file
    with tempfile.NamedTemporaryFile(suffix='.xml', dir=tempdir, delete=False) as fd:
        fnameXML = fd.name
        fd.write(ElementTree.tostring(xml))

    #--------------------------------------------------------------------
    # Now send the message
    #--------------------------------------------------------------------
    irt.transmitFiles("GET_ACTIVE_TABLE2", msgSendDest, mhsid,
      [fname, fnameXML], xmtScript)


def runFromJava(serverHost, serverPort, serverProtocol, mhsid, siteID, ancf, 
                bncf, xmtScript):
    init_logging()
    
    logger.info('*********** requestAT ******************')
    startT = time.time()
    
    try:
        execute_request_at(serverHost, serverPort, serverProtocol, mhsid, 
                             siteID, ancf, bncf, xmtScript)
    except:
        logger.exception('Error requesting active table')
    
    #--------------------------------------------------------------------
    # Finish
    #--------------------------------------------------------------------
    endT = time.time()
    logger.info("Final: wctime: {0:-6.2f}, cputime: {1:-6.2f}".format(endT - startT, time.clock()))