Свести данные по схеме Marshallow / SQLAlchemy - PullRequest
Новая
       30

Свести данные по схеме Marshallow / SQLAlchemy

0 голосов
/ 14 января 2019

У меня есть некоторые проблемы с тем, как я получаю свои данные с одной из моих конечных точек, в частности, с помощью Marshmallow и SQLAlchemy

https://gist.github.com/martinmckenna/eb5eeee5869663fc8f2e52a5e7ef72c9

У меня есть отношение многие ко многим между коктейлями и ингредиентами, но у меня также есть больше данных, чем просто внешние ключи в реляционной таблице, ings_in_cocktail, такие как ounces. Когда я ПОЛУЧАЮ /cocktails/, он возвращает что-то вроде это: 

{
  "cocktails": [
    {
      "glass": "rocks",
      "ingredients": [
        {
          "ingredient": {
            "ing_type": "liquor",
            "id": 1,
            "name": "gin"
          },
          "ounces": 20
        }
      ],
      "finish": "stirred",
      "id": 1,
      "name": "gin and tonic"
    }
  ]
}

То, что я хотел бы сделать, это объединить распространение ounces свойство с ingredient dict.

Я хочу, чтобы данные выглядели следующим образом: 

{
  "cocktails": [
    {
      "glass": "rocks",
      "ingredients": [
        {
          "ing_type": "liquor",
          "id": 1,
          "name": "gin",
          "ounces": 20
        }
      ],
      "finish": "stirred",
      "id": 1,
      "name": "gin and tonic"
    }
  ]
}

После поисков в Интернете в течение нескольких часов, я не могу найти способ сделать это легко с Marshamallow. Есть ли какой-то простой способ, который я пропускаю?

Ответы [ 2 ]

0 голосов
/ 19 января 2019

В итоге я решил это так: 

class CocktailSchema(ma.ModelSchema):
    # this is responsible for returning all the ingredient data on the cocktail
    ingredients = ma.Nested(CocktailIngredientSchema, many=True, strict=True)
    ingredients = fields.Method('concat_ingredients_dicts')

    """
    at this point the ingredients field on the cocktail object looks something like this

    ingredients: [{
        ingredient: {
            name: 'white russian',
            glass: 'rocks',
            finish: 'stirred'
        },
        ounces: 2,
        action: 'muddle',
        step: 1
    }]

    what we want is to concat this data so "ingredients" just turns
    into an list of dicts
    """
    def concat_ingredients_dicts(self, obj):
        result_ingredients_list = []
        i = 0
        while i < len(list(obj.ingredients)):
            # create a dict from the fields that live in the relational table
            relational_fields_dict = {
                'ounces': obj.ingredients[i].ounces,
                'action': obj.ingredients[i].action,
                'step': obj.ingredients[i].step
            }

            # create a dict from the fields on each ingredient in the cocktail
            ingredients_dict = obj.ingredients[i].ingredient.__dict__
            ingredients_dict_extracted_values = {
                'name': ingredients_dict.get('name'),
                'type': ingredients_dict.get('ing_type'),
                'id': ingredients_dict.get('id')
            }

            # merge the two dicts together
            merged = dict()
            merged.update(ingredients_dict_extracted_values)
            merged.update(relational_fields_dict)

            # append this merged dict a result array
            result_ingredients_list.append(merged)
            i += 1
        # return the array of ingredients
        return result_ingredients_list

    class Meta:
      model = Cocktail
0 голосов
/ 14 января 2019

Вы можете использовать поле метода в IngredientSchema

https://marshmallow.readthedocs.io/en/3.0/custom_fields.html#method-fields

Пожалуйста, отметьте, как использовать это поле в документах

...