jshbrmn/irt-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add basic error handling

Browse Source
This commit is contained in:
Josh Burman
2021-12-10 19:45:03 +00:00
parent d107b66e4f
commit e0fca68f46
3 changed files with 37 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
app/services/loft_service.py
View File

@ -3,6 +3,7 @@ import os, json, random, io, logging
from pulp import LpProblem, LpVariable, LpMinimize, LpStatus, lpSum
from helpers import aws_helper, tar_helper, csv_helper, service_helper, solver_helper
from lib.errors.item_generation_error import ItemGenerationError
from models.solver_run import SolverRun
from models.solution import Solution
@ -12,9 +13,12 @@ from services.base import Base
class LoftService(Base):
def process(self):
self.solver_run = SolverRun.parse_obj(self.retreive_attributes_from_message())
self.solution = self.generate_solution()
self.result = self.stream_to_s3_bucket()
try:
self.solver_run = SolverRun.parse_obj(self.retreive_attributes_from_message())
self.solution = self.generate_solution()
self.result = self.stream_to_s3_bucket()
except ItemGenerationError as error:
self.result = self.stream_to_s3_bucket(error)
def retreive_attributes_from_message(self):
logging.info('Retrieving attributes from message...')
@ -96,12 +100,18 @@ class LoftService(Base):
return solution
def stream_to_s3_bucket(self):
def stream_to_s3_bucket(self, error = None):
self.file_name = f'{service_helper.key_to_uuid(self.key)}.csv'
logging.info('Streaming %s to s3 bucket - %s', self.file_name, os.environ['S3_PROCESSED_BUCKET'])
solution_file = None
# setup writer buffer and write processed forms to file
buffer = io.StringIO()
solution_file = service_helper.solution_to_file(buffer, self.solver_run.total_form_items, self.solution.forms)
if error:
logging.info('Streaming %s error response to s3 bucket - %s', self.file_name, os.environ['S3_PROCESSED_BUCKET'])
solution_file = service_helper.error_to_file(buffer, error)
else:
logging.info('Streaming %s to s3 bucket - %s', self.file_name, os.environ['S3_PROCESSED_BUCKET'])
solution_file = service_helper.solution_to_file(buffer, self.solver_run.total_form_items, self.solution.forms)
# upload generated file to s3 and return result
return aws_helper.file_stream_upload(solution_file, self.file_name, os.environ['S3_PROCESSED_BUCKET'])