user.py 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. from typing import Optional
  2. from pydantic import BaseModel, EmailStr
  3. from datetime import datetime
  4. from typing import List
  5. class UserBase(BaseModel):
  6. mobile: str
  7. name: Optional[str] = None
  8. english_name: Optional[str] = None
  9. status: str = "PENDING"
  10. role: Optional[str] = "ORDINARY_USER"
  11. is_deleted: int = 0
  12. class UserCreate(UserBase):
  13. password: str
  14. admin_password: Optional[str] = None
  15. class UserRegister(BaseModel):
  16. mobile: str
  17. password: str
  18. sms_code: str # Added field
  19. name: str # Chinese Name
  20. class PromoteUserRequest(BaseModel):
  21. password: str
  22. captcha_id: str
  23. captcha_code: str
  24. class BatchResetEnglishNameRequest(BaseModel):
  25. user_ids: List[int]
  26. admin_password: str
  27. class UserUpdate(BaseModel):
  28. password: Optional[str] = None
  29. mobile: Optional[str] = None
  30. name: Optional[str] = None
  31. english_name: Optional[str] = None
  32. status: Optional[str] = None
  33. role: Optional[str] = None
  34. is_deleted: Optional[int] = None
  35. admin_password: Optional[str] = None
  36. class UserSyncRequest(BaseModel):
  37. mobile: str
  38. password: Optional[str] = None
  39. status: Optional[str] = None
  40. mapped_key: Optional[str] = None # External User ID
  41. mapped_email: Optional[str] = None # External User Email
  42. is_active: Optional[bool] = None # True=Active, False=Disabled. None=No Change
  43. class UserInDBBase(UserBase):
  44. id: int
  45. created_at: datetime
  46. updated_at: datetime
  47. class Config:
  48. from_attributes = True
  49. class User(UserInDBBase):
  50. pass
  51. class UserInDB(UserInDBBase):
  52. password_hash: str
  53. class UserList(BaseModel):
  54. total: int
  55. items: List[User]
  56. class UserSyncSimple(BaseModel):
  57. mobile: str
  58. name: Optional[str] = None
  59. english_name: Optional[str] = None
  60. class Config:
  61. from_attributes = True
  62. class UserSyncList(BaseModel):
  63. total: int
  64. items: List[UserSyncSimple]