8dc2f6c2c4
Change-Id: Ic990a5d04971f6374e68d87ad0ceb09991734000 Former-commit-id:da8796f655[formerly084b44e957] [formerly4463098852[formerly 2cda7c79a5f8f061a4e284bb7dd0fd4cd04a2245]] Former-commit-id:4463098852Former-commit-id:a2bcc8bffd
123 lines
4.7 KiB
Python
123 lines
4.7 KiB
Python
##
|
|
# This software was developed and / or modified by Raytheon Company,
|
|
# pursuant to Contract DG133W-05-CQ-1067 with the US Government.
|
|
#
|
|
# U.S. EXPORT CONTROLLED TECHNICAL DATA
|
|
# This software product contains export-restricted data whose
|
|
# export/transfer/disclosure is restricted by U.S. law. Dissemination
|
|
# to non-U.S. persons whether in the United States or abroad requires
|
|
# an export license or other authorization.
|
|
#
|
|
# Contractor Name: Raytheon Company
|
|
# Contractor Address: 6825 Pine Street, Suite 340
|
|
# Mail Stop B8
|
|
# Omaha, NE 68106
|
|
# 402.291.0100
|
|
#
|
|
# See the AWIPS II Master Rights File ("Master Rights File.pdf") for
|
|
# further licensing information.
|
|
##
|
|
|
|
|
|
#
|
|
# Main processing module of pypies. Receives the http request through WSGI,
|
|
# deserializes the request, processes it, and serializes the response
|
|
#
|
|
#
|
|
# SOFTWARE HISTORY
|
|
#
|
|
# Date Ticket# Engineer Description
|
|
# ------------ ---------- ----------- --------------------------
|
|
# 08/17/10 njensen Initial Creation.
|
|
#
|
|
#
|
|
#
|
|
|
|
from werkzeug import Request, Response, ClosingIterator
|
|
import time, logging, os
|
|
import pypies
|
|
from pypies import IDataStore
|
|
import dynamicserialize
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.pypies.request import *
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.pypies.response import *
|
|
|
|
logger = pypies.logger
|
|
timeMap = pypies.timeMap
|
|
|
|
from pypies.impl import H5pyDataStore
|
|
datastore = H5pyDataStore.H5pyDataStore()
|
|
|
|
datastoreMap = {
|
|
StoreRequest: (datastore.store, "StoreRequest"),
|
|
RetrieveRequest: (datastore.retrieve, "RetrieveRequest"),
|
|
DatasetNamesRequest: (datastore.getDatasets, "DatasetNamesRequest"),
|
|
DatasetDataRequest: (datastore.retrieveDatasets, "DatasetDataRequest"),
|
|
GroupsRequest: (datastore.retrieveGroups, "GroupsRequest"),
|
|
DeleteRequest: (datastore.delete, "DeleteRequest"),
|
|
DeleteFilesRequest: (datastore.deleteFiles, "DeleteFilesRequest"),
|
|
CreateDatasetRequest: (datastore.createDataset, "CreateDatasetRequest"),
|
|
RepackRequest: (datastore.repack, "RepackRequest"),
|
|
CopyRequest: (datastore.copy, "CopyRequest")
|
|
}
|
|
|
|
@Request.application
|
|
def pypies_response(request):
|
|
timeMap.clear()
|
|
try:
|
|
startTime = time.time()
|
|
try:
|
|
obj = dynamicserialize.deserialize(request.data)
|
|
except:
|
|
msg = 'Error deserializing request: ' + IDataStore._exc()
|
|
logger.error(msg)
|
|
resp = ErrorResponse()
|
|
resp.setError(msg)
|
|
return __prepareResponse(resp)
|
|
timeMap['deserialize']=time.time()-startTime
|
|
|
|
clz = obj.__class__
|
|
if logger.isEnabledFor(logging.DEBUG):
|
|
logger.debug(str(clz) + ": " + obj.getFilename())
|
|
success = False
|
|
if datastoreMap.has_key(clz):
|
|
try:
|
|
resp = datastoreMap[clz][0](obj)
|
|
success = True
|
|
except:
|
|
msg = 'Error processing ' + datastoreMap[clz][1] +' on file ' + obj.getFilename() + ': ' + IDataStore._exc()
|
|
logger.error(msg)
|
|
resp = ErrorResponse()
|
|
resp.setError(msg)
|
|
else:
|
|
msg = 'IDataStore unable to process request of type ' + str(obj.__class__)
|
|
logger.error(msg)
|
|
resp = ErrorResponse()
|
|
resp.setError(msg)
|
|
|
|
startSerialize = time.time()
|
|
httpResp = __prepareResponse(resp)
|
|
if success:
|
|
endTime = time.time()
|
|
timeMap['serialize'] = endTime - startSerialize
|
|
timeMap['total'] = endTime - startTime
|
|
logger.info({'request':datastoreMap[clz][1], 'time':timeMap, 'file':obj.getFilename()})
|
|
#logger.info("pid=" + str(os.getpid()) + " " + datastoreMap[clz][1] + " on " + obj.getFilename() + " processed in " + ('%.3f' % (t1-t0)) + " seconds")
|
|
return httpResp
|
|
except:
|
|
# Absolutely should not reach this, if we do, need to fix code
|
|
logger.error("Uncaught exception! " + IDataStore._exc())
|
|
return Response("Very bad uncaught exception, check pypies log")
|
|
|
|
|
|
def __prepareResponse(resp):
|
|
try:
|
|
serializedResp = dynamicserialize.serialize(resp)
|
|
except:
|
|
resp = ErrorResponse()
|
|
errorMsg = 'Error serializing response: ' + IDataStore._exc()
|
|
logger.error(errorMsg)
|
|
resp.setError(errorMsg)
|
|
# hopefully the error response serializes ok, if not you're kind of screwed
|
|
serializedResp = dynamicserialize.serialize(resp)
|
|
return Response(serializedResp)
|
|
|