data_management.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from typing import List, Optional
  2. from pydantic import BaseModel
  3. class DataManagementBase(BaseModel):
  4. # 数据名称
  5. name: str
  6. # 创建人名称
  7. user_name: str
  8. class DataManagementCreate(DataManagementBase):
  9. dag_uuid: str
  10. node_id: str
  11. out_pin: str
  12. class Config:
  13. schema_extra = {
  14. "example": {
  15. "name": "test",
  16. "user_name": "test",
  17. "dag_uuid": "test",
  18. "node_id": "test",
  19. "out_pin": "0",
  20. }
  21. }
  22. class DataManagement(DataManagementBase):
  23. id: int
  24. # 表格名称
  25. table_name: str
  26. # 创建时间
  27. create_time: int
  28. # 创建人编号
  29. user_id: str
  30. # 项目编号
  31. project_id: str
  32. class Config:
  33. orm_mode = True
  34. class DataManagementSelect(BaseModel):
  35. table_name: str
  36. page: Optional[int] = 0
  37. size: Optional[int] = 100
  38. class Config:
  39. schema_extra = {
  40. "example": {
  41. "table_name": "train",
  42. "page": 0,
  43. "size": 100
  44. }
  45. }