def getAllF(args, header, exportCsv=False): try: token = header.get(AUTH_TOKEN) status = TaskExecutionService.checkToken(token) if status is not None: return status except: return Utils.JsonMessage("Internal server error", 500) taskID = args.get(FILTER_TASK_ID) robotID = args.get(FILTER_ROBOT_ID) status = args.get(FILTER_STATUS) durationFrom = args.get(FILTER_DURATION_FROM) durationTo = args.get(FILTER_DURATION_TO) dateFrom = args.get(FILTER_DATE_FROM) dateTo = args.get(FILTER_DATE_TO) entityList = DBUtil.taskExecutionFilter(TaskExecution, taskId=taskID, robotId=robotID, status=status, durationFrom=durationFrom, durationTo=durationTo, dateFrom=dateFrom, dateTo=dateTo) dtoList = [] for e in entityList: dto = TaskExecutedResponseDto() dto.fromEntity(e) dtoList.append(dto.getJson()) jsonList = { 'executed': dtoList } if exportCsv: csvData = ""#StringIO() for e in entityList: csvData += "{};{};{};{};{};{};{};{};{};{}\n".format(e.id, e.uuid, e.task_id, e.robot_id, e.date, e.start_time, e.end_time, e.duration, e.status, e.created) print(csvData) return csvData else: return Utils.JsonResponse(jsonList, 200)
def getAll(header): try: token = header.get(AUTH_TOKEN) status = TaskExecutionService.checkToken(token) if status is not None: return status except: return Utils.JsonMessage("Internal server error", 500) entityList = DBUtil.findAll(TaskExecution) dtoList = [] for e in entityList: dto = TaskExecutedResponseDto() dto.fromEntity(e) dtoList.append(dto.getJson()) jsonList = { 'executed': dtoList } return Utils.JsonResponse(jsonList, 200)
def action(method, modelDto, header, id=None): token = None try: token = header.get(AUTH_TOKEN) except: pass if token is None: return Utils.JsonMessage("Token parameter not found!", 500) if not LoginService.validateJWTToken(token): return Utils.JsonMessage("Unauthorized", 401) if id is not None: model = None if modelDto is not None: model = JSONSerializator().serialize(modelDto) status, result, message, code = GenericHelperService.handleMethod( Task, method, model, id) if status: taskResponse = SimpleTaskResponseDto() taskResponse.fromEntity(result) return Utils.JsonResponse(taskResponse.getJson(), 200) else: if code is None: code = 500 return Utils.JsonMessage(message, code) else: if modelDto is not None: taskRequestDto = TaskRequestDto().serialize(modelDto) serialized = JSONSerializator().serialize(modelDto) if Utils.check_object_propertis(TaskRequestDto(), serialized): entity = Task.createFromRequestDto( taskRequestDto, str(int(dt.now().timestamp()))) status, model, message, code = GenericHelperService.handleMethod( Task, method, entity, id) if status: e = DBUtil.findByName(Task, entity.name) taskResponseDto = SimpleTaskResponseDto() taskResponseDto.fromEntity(e) return Utils.JsonResponse(taskResponseDto.getJson(), 201) else: return Utils.JsonMessage(message, code) else: return Utils.JsonMessage( "JSON format is not valid. Check properties".format( id), 500) else: if method == 'GET': status, result, message, code = GenericHelperService.handleMethod( Task, method, None, None) dtoList = [] for e in result: dto = SimpleTaskResponseDto() dto.fromEntity(e) dtoList.append(dto.getJson()) listResponse = {'tasks': dtoList} return Utils.JsonResponse(listResponse, 200) return Utils.JsonMessage( "ID and body cannot be empty!".format(id), 500)