Estoy tratando de hacer una API con Flask-RestX que pueda mostrar una respuesta como esta,

{
  "id": 1,
  "msg": "How are things",
  "score": 345,
  "additional": {
    "goal": 56,
    "title": "ASking how"
  }
}

Cuando los datos son así (en mi caso, no controlo el formato de los datos),

data = {
    "id":1,
    "msg":"How are things",
    "goal": 56,
    "score":345,
    "title":"Asking how"
}

Pero la respuesta que obtengo con el código actual es incorrecta, muestra valores nulos

{
  "id": 1,
  "msg": "How are things",
  "score": 345,
  "additional": {
    "goal": null,
    "title": null
  }
}

Código completo -

from flask import Flask
from flask_restx import Resource, Api, fields

app = Flask(__name__)
api = Api(app)

data = {
    "id":1,
    "msg":"How are things",
    "goal": 56,
    "score":345,
    "title":"Asking how"
}

extra = api.model('Extra', {
    'goal': fields.Integer,
    'title': fields.String
    })

model = api.model('Model', {
    'id' : fields.Integer,
    'msg' : fields.String,
    'score': fields.Integer,
    'additional' : fields.Nested(extra)
  })


@api.route('/hello')
class HelloWorld(Resource):
    @api.marshal_with(model)
    def get(self):
        return data

if __name__ == '__main__':
    app.run(debug=True, port=4000)

Soy completamente nuevo en Flask-RestX / Flask-RestPlus. Por favor, dígame, ¿cómo puedo lograr esto sin cambiar el formato de los datos?

1
realsdx 9 feb. 2021 a las 20:13

1 respuesta

La mejor respuesta

De acuerdo con comentario de waynetech arriba, lo siguiente funcionó para mí:

from flask import Flask
from flask_restx import Resource, Api, fields

app = Flask(__name__)
api = Api(app)

data = {
    "id":1,
    "msg":"How are things",
    "score":345,
    "additional": {
        "goal": 56,
        "title": "Asking how",
    }
}

extra = api.model('Extra', {
    'goal': fields.Integer,
    'title': fields.String
    })

model = api.model('Model', {
    'id' : fields.Integer,
    'msg' : fields.String,
    'score': fields.Integer,
    'additional' : fields.Nested(extra)
  })


@api.route('/hello')
class HelloWorld(Resource):

    @api.marshal_with(model)
    def get(self):
        return data

    @api.marshal_with(model)
    def post(self):
        data = api.payload

        return data

if __name__ == '__main__':
    app.run(debug=True, port=4000)

Aquí está el script de prueba rápida que utilicé:

from pprint import pprint
import requests

payload = {
    "id": 1,
    "msg":"Test",
    "score":345,
    "additional": {
        "goal": 56,
        "title": "Test",
    }
}

url = 'http://127.0.0.1:4000/hello'
r = requests.post(url, json=payload)

if r.status_code == 200:
    print(r.status_code)
    pprint(r.json())
else:
    print("FAIL: ", r.status_code)
1
Danny Blaker 8 mar. 2021 a las 06:11