41 lines
945 B
Python
41 lines
945 B
Python
from ninja import Schema, ModelSchema
|
|
from pydantic import EmailStr, Field
|
|
|
|
from .models import User, RoleChoices
|
|
|
|
|
|
class UserListSchema(ModelSchema):
|
|
@classmethod
|
|
def from_orm(cls, obj):
|
|
raw_password = obj.raw_password if obj.role == RoleChoices.NORMAL else ""
|
|
return cls(
|
|
username=obj.username,
|
|
raw_password=raw_password,
|
|
role=obj.role,
|
|
date_joined=obj.date_joined,
|
|
last_login=obj.last_login,
|
|
is_active=obj.is_active,
|
|
)
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = [
|
|
"username",
|
|
"raw_password",
|
|
"role",
|
|
"date_joined",
|
|
"last_login",
|
|
"is_active",
|
|
]
|
|
|
|
|
|
class UserRegistrationSchema(Schema):
|
|
username: str
|
|
email: EmailStr
|
|
password: str = Field(min_length=6, max_length=20)
|
|
|
|
|
|
class UserLoginSchema(Schema):
|
|
username: str
|
|
password: str
|