pydantic 从 model 中排除多个字段

Question

In pydantic is there a cleaner way to exclude multiple fields from the model, something like:

class User(UserBase):
    class Config:        
        exclude = ['user_id', 'some_other_field']

I am aware that following works, but I was looking for something cleaner like django.

class User(UserBase):

    class Config:       
        fields = {'user_id': {'exclude':True}, 
                   'some_other_field': {'exclude':True}
                 }

Answer 1

Pydantic will exclude the class variables which begin with an underscore. so if it fits your use case, you can rename your attribues.

class User(UserBase):
    _user_id=str
    some_other_field=str
    ....

Answer 2

I wrote something like this for my json:

from pydantic import BaseModel


class CustomBase(BaseModel):
    def json(self, **kwargs):
        include = getattr(self.Config, "include", set())
        if len(include) == 0:
            include = None
        exclude = getattr(self.Config, "exclude", set())
        if len(exclude) == 0:
            exclude = None
        return super().json(include=include, exclude=exclude, **kwargs)

    

class User(CustomBase):
    name :str = ...
    family :str = ...

    class Config:
        exclude = {"family"}


u = User(**{"name":"milad","family":"vayani"})

print(u.json())

you can overriding dict and other method like.

Answer 3

A possible solution is creating a new class based in the baseclass using create_model:

from pydantic import BaseModel, create_model

def exclude_id(baseclass, to_exclude: list):
    # Here we just extract the fields and validators from the baseclass
    fields = baseclass.__fields__
    validators = {'__validators__': baseclass.__validators__}
    new_fields = {key: (item.type_, ... if item.required else None)
                  for key, item in fields.items() if key not in to_exclude}
    return create_model(f'{baseclass.__name__}Excluded', **new_fields, __validators__=validators)


class User(BaseModel):
    ID: str
    some_other: str

list_to_exclude = ['ID']
UserExcluded = exclude_id(User, list_to_exclude)

UserExcluded(some_other='hola')

Which will return:

> UserExcluded(some_other='hola')

Which is a copy of the baseclass but with no parameter 'ID'.

If you have the id in the validators you may want also to exclude those validators.