def process(self, request):
self.payloadValidationHelper = PayloadValidationHelper(self.config)
# 0. Logs the request
self._logRequest(request)
# 1. Validates the payload
payload = self.payloadValidationHelper._getValidatedPayload(request)
if not payload:
return self._responseFailure(400, "payload invalid.", payload, request)
# 2. Convert the 'body' HTML to a plain text version to send along to the email provider.
self._updateBodyHTMLsafe(payload)
# 3. Send email via providers (try each if fails)
# 4. Log the requests and responses
providerRequestHandler = ProviderRequestHandler()
for priority in self.providerConfigsInPriority:
providerInfo = self.providerConfigsInPriority[priority]
statusCode, response = providerRequestHandler._sendRequest(payload, providerInfo)
if statusCode == 200:
return self._responseSuccess(statusCode, response, payload, request)
else:
# ToDo: this will log twice for the last failure
self._log("Log", statusCode, response, payload, request)
if statusCode == 200:
return self._responseSuccess(statusCode, response, payload, request)
else:
return self._responseFailure(statusCode, response, payload, request)
class EmailRequestHandler:
def __init__(self, config):
self.config = config
self.payloadValidationSchema = self.config.getPayloadValidationConfig()
# a list of provider info sorted by priority
self.providerConfigsInPriority = self.config.getProviderConfigsInPriority()
# Log
logFilePath = config.getLogPath()
logging.basicConfig(filename=logFilePath, level=logging.INFO)
# Splunk Log
splunkLogPath = config.getSplunkLogPath()
if os.path.isfile(splunkLogPath):
self.splunkLog = open(splunkLogPath, 'a')
else:
self.splunkLog = open(splunkLogPath, 'w')
def process(self, request):
self.payloadValidationHelper = PayloadValidationHelper(self.config)
# 0. Logs the request
self._logRequest(request)
# 1. Validates the payload
payload = self.payloadValidationHelper._getValidatedPayload(request)
if not payload:
return self._responseFailure(400, "payload invalid.", payload, request)
# 2. Convert the 'body' HTML to a plain text version to send along to the email provider.
self._updateBodyHTMLsafe(payload)
# 3. Send email via providers (try each if fails)
# 4. Log the requests and responses
providerRequestHandler = ProviderRequestHandler()
for priority in self.providerConfigsInPriority:
providerInfo = self.providerConfigsInPriority[priority]
statusCode, response = providerRequestHandler._sendRequest(payload, providerInfo)
if statusCode == 200:
return self._responseSuccess(statusCode, response, payload, request)
else:
# ToDo: this will log twice for the last failure
self._log("Log", statusCode, response, payload, request)
if statusCode == 200:
return self._responseSuccess(statusCode, response, payload, request)
else:
return self._responseFailure(statusCode, response, payload, request)
def _updateBodyHTMLsafe(self, payload):
body = self._getValueFromDict("body", payload)
if body:
payload["safeBody"] = Markup(body).striptags() # remove HTML tags
# payload["safeBody"] = Markup.escape(body) # escape HTML tags
def _responseSuccess(self, statusCode, response, payload, request):
self._log("Log", statusCode, response, payload, request)
response = Response(
"{\"message\": \"Mail has sent successfully.\", \"response\": "+ response +"}",
status=statusCode,
mimetype='application/json'
)
return response
def _responseFailure(self, statusCode, response, payload, request):
self._log("Log", statusCode, response, payload, request)
response = Response(
"{\"message\": \"Error while processing the request\", \"response\": "+ response +"}",
status=statusCode,
mimetype='application/json'
)
return response
# ToDo: move this to proper util or helper class
def _logRequest(self, request):
level = "Log"
env = self.config.getEnv()
emailServiceVersion = self.config.getEmailServiceVersion()
ip = request.remote_addr
userAgent = request.headers["User-Agent"]
payload = self.payloadValidationHelper._getPayload(request)
data = json.dumps(payload)
asciidata = self._unicodeToAscii(data)
ts = time.time()
now = datetime.datetime.fromtimestamp(ts).strftime('%Y-%m-%d %H:%M:%S')
contents = "logType="+ level +", v="+ emailServiceVersion +", env="+ env +", type=request, ip="+ ip +", userAgent="+ userAgent +", payload="+ asciidata +", timeStamp="+ str(now)
self.splunkLog.write(contents + "\n")
# ToDo: move this to proper util or helper class
def _log(self, level, statusCode, response, payload, request):
env = self.config.getEnv()
emailServiceVersion = self.config.getEmailServiceVersion()
ip = request.remote_addr
userAgent = request.headers["User-Agent"]
data = json.dumps(payload)
asciidata = self._unicodeToAscii(data)
ts = time.time()
now = datetime.datetime.fromtimestamp(ts).strftime('%Y-%m-%d %H:%M:%S')
contents = "logType="+ level +", v="+ emailServiceVersion +", env="+ env +", type=response, statusCode="+ str(statusCode) +", response="+ response +", ip="+ ip +", userAgent="+ userAgent +", payload="+ asciidata +", timeStamp="+ str(now)
self.splunkLog.write(contents + "\n")
# ToDo: move this to proper util or helper class
def _unicodeToAscii(self, data):
udata = data.decode("utf-8")
asciidata = udata.encode("ascii", "ignore")
return asciidata
# ToDo: move this to proper util or helper class
def _getValueFromDict(self, key, payload):
if key in payload:
return payload[key]
else:
return None
