Как объединить вложенную схему в один JSON объект с Flask Marshmallow - PullRequest
Новая
       89

Как объединить вложенную схему в один JSON объект с Flask Marshmallow

0 голосов
/ 05 февраля 2020

У меня проблема с объединением двух схем в один вложенный JSON объект. Этот API-интерфейс возвратил JSON в результате запроса к 3 связанным таблицам. После того, как я сделал запрос, я использую две схемы Marshmellow, на которые ссылается здесь . Но ответ не такой, как ожидалось, вот ответ: 

{
  "message": "success",
  "device": [
    {
      "device_name": "Kamar Tidur Utama"
    }
  ],
  "sensor": [
    {
      "value": 23.3683,
      "sensor_name": "Temperature"
    },
    {
      "value": 0.0,
      "sensor_name": "Motion"
    },
    {
      "value": 90.12,
      "sensor_name": "Humidity"
    },
    {
      "value": 15.8667,
      "sensor_name": "Current 1"
    },
    {
      "value": 15.0333,
      "sensor_name": "Current 2"
    }
  ]
}

Я хотел, чтобы объект «датчик» был помещен внутри «устройства». У многих устройств есть много датчиков, подобных этому: 

{
  "message": "success",
  "device": [
    {
        "device_name": "Kamar Tidur Utama"
        "sensor": [
            {
            "value": 23.3683,
            "sensor_name": "Temperature"
            },
            {
            "value": 0.0,
            "sensor_name": "Motion"
            },
            {
            "value": 90.12,
            "sensor_name": "Humidity"
            },
            {
            "value": 15.8667,
            "sensor_name": "Current 1"
            },
            {
            "value": 15.0333,
            "sensor_name": "Current 2"
            }
        ]
    }
  ]
}

Это три мои родственные модели:

Device.py (метод, который выполняет запрос к API) 

 @classmethod
    def get_device_sensor(cls):
        device_sensor_schema = DeviceSensorSchema(many=True)
        sensor_value_schema = SensorValueSchema(many=True)

        device = Device.query.join(SensorData, Device.id == SensorData.device_id)\
            .add_columns(Device.name.label('device_name'))
        print(device)
        sensor = Sensor.query.join(SensorData, Sensor.id == SensorData.sensor_id)\
            .add_columns(Sensor.name.label('sensor_name'), SensorData.value.label('value'))

        res_device = device_sensor_schema.dump(device, many=True)
        res_sensor = sensor_value_schema.dump(sensor, many=True)
        if device:
            return jsonify({'message': 'success', 'device': res_device, 'sensor': res_sensor})
        else:
            return {'message': 'Table device is empty'}, 404

Device.py (схема двух маршмеллоу) 

class SensorValueSchema(ma.Schema):
    sensor_name = fields.String()
    value = fields.Float()
    ordered=True

class DeviceSensorSchema(Schema):
    device_name = fields.String()
    sensors = fields.List(fields.Nested(SensorValueSchema))

Device.py (класс модели) 

class Device(db.Model):
    __tablename__ = "devices"
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    name = db.Column(db.String(255))
    serial_number = db.Column(db.String(255), unique=True)
    used_relay = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime(timezone=True), server_default=db.func.now())
    updated_at = db.Column(db.DateTime(timezone=True), onupdate=db.func.now())
    switches = db.relationship('Switch', backref='device', lazy='dynamic')
    power_usages = db.relationship('PowerUsage', backref='device', lazy='dynamic')
    power_expenses = db.relationship('PowerExpense', backref='device', lazy='dynamic')
    behaviors_dataset = db.relationship('BehaviorDataset', backref='device', lazy='dynamic')
    usage_histories = db.relationship('UsageHistory', backref='device', lazy='dynamic')
    sensors_data = db.relationship('SensorData', backref='device', lazy='dynamic')
    notifications = db.relationship('Notification', backref='device', lazy='dynamic')
    device_serial_number = db.relationship('DeviceSerialNumber', backref='device', lazy='dynamic')

Sensor.py (класс модели) 

class Sensor(db.Model):
    __tablename__ = "sensors"
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(120))
    created_at = db.Column(db.DateTime(timezone=True), server_default=db.func.now())
    updated_at = db.Column(db.DateTime(timezone=True), onupdate=db.func.now())
    sensors_data = db.relationship('SensorData', backref='sensor', lazy='dynamic')

SensorData.py (класс модели) 

class SensorData(db.Model):
    __tablename__ = "sensors_data"
    id = db.Column(db.Integer, primary_key=True)
    device_id = db.Column(db.Integer, db.ForeignKey('devices.id'))
    sensor_id = db.Column(db.Integer, db.ForeignKey('sensors.id'))
    value = db.Column(db.Float)
    created_at = db.Column(db.DateTime(timezone=True), server_default=db.func.now())
    updated_at = db.Column(db.DateTime(timezone=True), onupdate=db.func.now())

1 Ответ

0 голосов
/ 06 февраля 2020

Вы должны назначить sensor объект для каждой записи в списке device. Мой дурак sh способ достижения желаемого результата: 

 @classmethod
    def get_device_sensor(cls):
        device_sensor_schema = DeviceSensorSchema(many=True)
        sensor_value_schema = SensorValueSchema(many=True)

        device = Device.query.join(SensorData, Device.id == SensorData.device_id)\
            .add_columns(Device.name.label('device_name'))
        print(device)
        sensor = Sensor.query.join(SensorData, Sensor.id == SensorData.sensor_id)\
            .add_columns(Sensor.name.label('sensor_name'), SensorData.value.label('value'))

        res_device[0].sensors = sensor
        res_device = device_sensor_schema.dump(device, many=True)

        if device:
            return jsonify({'message': 'success', 'device': res_device})
        else:
            return {'message': 'Table device is empty'}, 404

Чтобы лучше подходить к этому, вы бы использовали сопоставленный объект SQLAlchemy. В этом случае Device.sensors_data.

...