data_management.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  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. # 创建人编号
  9. user_id: str
  10. # 项目编号
  11. project_id: str
  12. class DataManagementCreate(DataManagementBase):
  13. dag_uuid: str
  14. node_id: str
  15. out_pin: str
  16. class Config:
  17. schema_extra = {
  18. "example": {
  19. "name": "test",
  20. "user_name": "test",
  21. "user_id": "test",
  22. "project_id": "test",
  23. "dag_uuid": "test",
  24. "node_id": "test",
  25. "out_pin": "0",
  26. }
  27. }
  28. class DataManagement(DataManagementBase):
  29. id: int
  30. # 表格名称
  31. table_name: str
  32. # 创建时间
  33. create_time: int
  34. class Config:
  35. orm_mode = True
  36. class DataManagementSelect(BaseModel):
  37. table_name: str
  38. page: Optional[int] = 0
  39. size: Optional[int] = 100
  40. class Config:
  41. schema_extra = {
  42. "example": {
  43. "table_name": "train",
  44. "page": 0,
  45. "size": 100
  46. }
  47. }