示例#1
0
 def getRobotTypeByValue(typeValue):
     entity = DBUtil.findByType(RobotType, typeValue)
     dto = RobotTypeResponseDto()
     dto.fromEntity(entity)
     print(dto.dumpModel())
示例#2
0
    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,
                                                     ignoreProperties=True)
            status, result, message, code = GenericHelperService.handleMethod(
                Robot, method, model, id)
            if status:
                robotResponse = SimpleRobotResponseDto()
                robotResponse.fromEntity(result)
                return Utils.JsonResponse(robotResponse.getJson(), 200)
            else:
                if code is None:
                    code = 500
                return Utils.JsonMessage(message, code)
        else:
            if modelDto is not None:
                robotRequestDto = RobotRequestDto().serialize(
                    modelDto, ignoreProperties=False)
                Logger.info(robotRequestDto)
                serialized = JSONSerializator().serialize(modelDto)
                if Utils.check_object_propertis(RobotRequestDto(), serialized):
                    entity = Robot.createFromRequestDto(
                        robotRequestDto, str(int(dt.now().timestamp())))
                    status, model, message, code = GenericHelperService.handleMethod(
                        Robot, method, entity, id)
                    if status:
                        e = DBUtil.findByName(Robot, entity.name)
                        robotResponseDto = RobotResponseDto()
                        robotResponseDto.fromEntity(e)
                        robotType = DBUtil.findByType(RobotType,
                                                      robotResponseDto.typeId)
                        robotTypeDto = RobotTypeResponseDto()
                        robotTypeDto.fromEntity(robotType)
                        robotResponseDto.setTypeDto(robotTypeDto)
                        return Utils.JsonResponse(robotResponseDto.getJson(),
                                                  201)
                    else:
                        return Utils.JsonMessage(message, code)
                else:
                    return Utils.JsonMessage(
                        "JSON format is not valid. Check properties", 500)
            else:
                if method == 'GET':
                    status, result, message, code = GenericHelperService.handleMethod(
                        Robot, method, None, None)
                    dtoList = []
                    for e in result:
                        dto = SimpleRobotResponseDto()
                        dto.fromEntity(e)
                        dtoList.append(dto.getJson())

                    listResponse = {'robots': dtoList}
                    return Utils.JsonResponse(listResponse, 200)
                return Utils.JsonMessage(
                    "ID and body cannot be empty!".format(id), 500)