File: enum_specialized.py

package info (click to toggle)
python-datamodel-code-generator 0.45.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 9,052 kB
  • sloc: python: 29,621; makefile: 15
file content (53 lines) | stat: -rw-r--r-- 1,008 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# generated by datamodel-codegen:
#   filename:  subclass_enum.json
#   timestamp: 2019-07-26T00:00:00+00:00

from __future__ import annotations

from enum import Enum, IntEnum, StrEnum
from typing import Optional

from pydantic import BaseModel


class IntEnumModel(IntEnum):
    integer_1 = 1
    integer_2 = 2
    integer_3 = 3


class FloatEnum(Enum):
    number_1_1 = 1.1
    number_2_1 = 2.1
    number_3_1 = 3.1


class StrEnumModel(StrEnum):
    field_1 = '1'
    field_2 = '2'
    field_3 = '3'


class NonTypedEnum(Enum):
    field_1 = '1'
    field_2 = '2'
    field_3 = '3'


class BooleanEnum(Enum):
    boolean_True = True
    boolean_False = False


class UnknownEnum(Enum):
    a = 'a'
    b = 'b'


class Model(BaseModel):
    IntEnum: Optional[IntEnumModel] = None
    FloatEnum: Optional[FloatEnum] = None
    StrEnum: Optional[StrEnumModel] = None
    NonTypedEnum: Optional[NonTypedEnum] = None
    BooleanEnum: Optional[BooleanEnum] = None
    UnknownEnum: Optional[UnknownEnum] = None