entity.py.vm 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. # -*- coding: utf-8 -*-
  2. # @Author : {{ table.function_author }}
  3. # @FileName: {{ table.class_name }}.py
  4. # @Time : {{ datetime }}
  5. from typing import Optional, Annotated
  6. from datetime import datetime
  7. from pydantic import Field, BeforeValidator
  8. from ruoyi_common.base.model import BaseEntity
  9. from ruoyi_common.base.transformer import to_datetime
  10. class {{ table.class_name }}(BaseEntity):
  11. """
  12. {{ table.table_comment }}对象
  13. """
  14. {%- for column in table.columns %}
  15. {%- if column.java_type == 'String' or column.java_type == 'str' %}
  16. # {{ column.column_comment }}
  17. {{ underscore(column.java_field) }}: Optional[str] = Field(default=None, description="{{ column.column_comment }}")
  18. {%- elif column.java_type == 'Integer' or column.java_type == 'int' %}
  19. # {{ column.column_comment }}
  20. {{ underscore(column.java_field) }}: Optional[int] = Field(default=None, description="{{ column.column_comment }}")
  21. {%- elif column.java_type == 'Long' %}
  22. # {{ column.column_comment }}
  23. {{ underscore(column.java_field) }}: Optional[int] = Field(default=None, description="{{ column.column_comment }}")
  24. {%- elif column.java_type == 'Float' or column.java_type == 'Double' %}
  25. # {{ column.column_comment }}
  26. {{ underscore(column.java_field) }}: Optional[float] = Field(default=None, description="{{ column.column_comment }}")
  27. {%- elif column.java_type == 'Boolean' or column.java_type == 'bool' %}
  28. # {{ column.column_comment }}
  29. {{ underscore(column.java_field) }}: Optional[bool] = Field(default=None, description="{{ column.column_comment }}")
  30. {%- elif column.java_type == 'Date' or column.java_type == 'DateTime' %}
  31. # {{ column.column_comment }}
  32. {{ underscore(column.java_field) }}: Annotated[Optional[datetime], BeforeValidator(to_datetime())] = Field(default=None, description="{{ column.column_comment }}")
  33. {%- else %}
  34. # {{ column.column_comment }}
  35. {{ underscore(column.java_field) }}: Optional[str] = Field(default=None, description="{{ column.column_comment }}")
  36. {%- endif %}
  37. {%- endfor %}
  38. # 页码
  39. page_num: Optional[int] = Field(default=1, description="页码")
  40. # 每页数量
  41. page_size: Optional[int] = Field(default=10, description="每页数量")