diff options
Diffstat (limited to 'src/lib')
| -rw-r--r-- | src/lib/dataclasses.py | 7 | ||||
| -rw-r--r-- | src/lib/szurubooru.py | 2 |
2 files changed, 5 insertions, 4 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) diff --git a/src/lib/szurubooru.py b/src/lib/szurubooru.py index 1219cfb..38c6265 100644 --- a/src/lib/szurubooru.py +++ b/src/lib/szurubooru.py @@ -533,7 +533,7 @@ class Szurubooru: category=c["category"], description=c["description"], implications=list(map(lambda a: a["names"][0], c["implications"])), - suggestions=c["suggestions"], + suggestions=list(map(lambda a: a["names"][0], c["suggestions"])), ) def tag_exists(self, name: str) -> bool: |