flask-marshmallow: как пометить все поля как необязательные, только если метод PUT - PullRequest
Новая
       59

flask-marshmallow: как пометить все поля как необязательные, только если метод PUT

0 голосов
/ 18 мая 2018

Я делаю колбу успокоительного API, у меня проблемы с зефир-sqlalchemy, и webargs

короче вот моя модель sqlalchemy: 

class User(Model):
    id = Column(String, primary_key=True)
    name = Column(String(64), nullable=False)
    email = Column(String(120), nullable=False)
    password = Column(String(128))
    creation_date = Column(DateTime, default=datetime.utcnow)

и этомоя схема 

class UserSchema(ModelSchema):
    class Meta:
        model = User
        strict = True
        sqla_session = db.session

user_schema = UserSchema()

и пример моих маршрутов, использующих классную флягу и webargs: 

class UserView(FlaskView):
    trailing_slash = False
    model = User
    schema = user_schema

    @use_kwargs(schema.fields)
    def post(self, **kwargs):
        try:
            entity = self.model()

            for d in kwargs:
                if kwargs[d] is not missing:
                    entity.__setattr__(d, kwargs[d])

            db.session.add(entity)
            db.session.commit()
            o = self.schema.dump(entity).data
            return jsonify({'{}'.format(self.model.__table__.name): o})

        except IntegrityError:
            return jsonify({'message': '{} exist in the database. choose another id'
                   .format(self.model.__table__.name)}), 409


    @use_kwargs(schema.fields)
    def put(self, id, **kwargs):
        entity = self.model.query.filter_by(id=id).first_or_404()

        for d in kwargs:
            if kwargs[d] is not missing:
                entity.__setattr__(d, kwargs[d])

        db.session.commit()
        o = self.schema.dump(entity).data
        return jsonify({'{}'.format(self.model.__table__.name): o})

UserView.register(app)

Проблема: , как вы можете видеть в моей sqlalchemyВ модели некоторые поля не могут быть обнулены, поэтому моя схема зефира помечает их как обязательные.Мои методы get, index, delete и post работают отлично.Но я включил сообщение по одной причине:

, когда я пытаюсь опубликовать нового пользователя, например, без имени, http-код 422 поднимается, потому что поле name является обязательным, что я хочу, и этосделано отлично.

НО при редактировании полей с помощью запроса put мне бы хотелось, чтобы ВСЕ в моей схеме стало необязательным ... прямо сейчас, если я хочу обновить пользователя, я должен предоставить не только идентификатор ..., но ВСЕ другие необходимые данныепо умолчанию, даже если я не изменил их вообще.

короче , как пометить все поля как «необязательные», когда метод «положен»?

РЕДАКТИРОВАТЬ: Как и решение, предоставленное @Mekicha, я произвел следующие изменения:

изменим схему, чтобы обязательные поля в моей модели приняли значение None.как это: 

class UserSchema(ModelSchema):
    class Meta:
        model = User
        ...
    name = fields.Str(missing=None, required=True)
    email = fields.Email(missing=None, required=True)
    ...

изменить условие моего пут и поста с этого: 

if kwargs[d] is not missing:

на это: 

if kwargs[d] is not missing and kwargs[d] is not None:

1 Ответ

0 голосов
/ 20 мая 2018

Поскольку вы хотите сделать поля необязательными во время put, как насчет установки атрибута missing для полей.Из doc :

отсутствует десериализация, если поле не найдено во входных данных

Я думаю, что комбинация missing и allow_none (по умолчанию True при missing=None), как указано здесь: https://github.com/marshmallow-code/marshmallow/blob/dev/src/marshmallow/fields.py#L89 должно работать для вас

...