diff options
| author | Colin Wilk <colin.wilk@tum.de> | 2023-10-02 01:10:42 +0200 |
|---|---|---|
| committer | Colin Wilk <colin.wilk@tum.de> | 2023-10-02 01:10:42 +0200 |
| commit | da105eb3dd66935401c7740a132b3d18d77294ec (patch) | |
| tree | 9188c81f6a2381038fd4404989efc963b59c358a /src/lib/dataclasses.py | |
| parent | 45301553e15e95afe3139f8efa4885c1462d436d (diff) | |
| download | szuruboorupy-da105eb3dd66935401c7740a132b3d18d77294ec.tar.gz szuruboorupy-da105eb3dd66935401c7740a132b3d18d77294ec.zip | |
Add pydantic to dataclasses
Verifies the type annotations on the classes.
This commit also fixes a bug where an invalid type was returned for
suggestions of a Tag. The return type was a dict of a Micro Tag
Structure but it was expected to be a string (see szurubooru.py)
Signed-off-by: Colin Wilk <colin.wilk@tum.de>
Diffstat (limited to 'src/lib/dataclasses.py')
| -rw-r--r-- | src/lib/dataclasses.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/lib/dataclasses.py b/src/lib/dataclasses.py index 16413b7..6313fca 100644 --- a/src/lib/dataclasses.py +++ b/src/lib/dataclasses.py @@ -3,8 +3,9 @@ Module for collection of dataclasses that map Szurubooru objects to python class """ -from dataclasses import dataclass, field -from typing import List +from pydantic.dataclasses import dataclass +from dataclasses import field +from typing import List, Optional @dataclass @@ -40,7 +41,7 @@ class Tag: name: str version: int = -1 - description: str = "" + description: Optional[str] = "" category: str = "General" implications: List[str] = field(default_factory=list) suggestions: List[str] = field(default_factory=list) |