forked from mirrors/cbz
Release v3.3.0
This commit is contained in:
hyugogirubato
+128
-63
@@ -1,72 +1,137 @@
|
||||
from cbz.constants import YesNo, Manga, AgeRating, Format, PageType, FIELDS, PAGE_FIELDS
|
||||
from cbz.utils import _get, _set
|
||||
from cbz.constants import COMIC_FIELDS, PAGE_FIELDS, Format, YesNo, Manga, AgeRating, LanguageISO, Rating, PageType
|
||||
from cbz.utils import verify_attr, default_attr
|
||||
|
||||
|
||||
class ComicModel:
|
||||
title: str = ''
|
||||
series: str = ''
|
||||
number: str = ''
|
||||
count: int = -1
|
||||
volume: int = -1
|
||||
alternate_series: str = ''
|
||||
alternate_number: str = ''
|
||||
alternate_count: int = -1
|
||||
summary: str = ''
|
||||
notes: str = ''
|
||||
year: int = -1
|
||||
month: int = -1
|
||||
day: int = -1
|
||||
writer: str = ''
|
||||
penciller: str = ''
|
||||
inker: str = ''
|
||||
colorist: str = ''
|
||||
letterer: str = ''
|
||||
cover_artist: str = ''
|
||||
editor: str = ''
|
||||
translator: str = ''
|
||||
publisher: str = ''
|
||||
imprint: str = ''
|
||||
genre: str = ''
|
||||
tags: str = ''
|
||||
web: str = ''
|
||||
language_iso: str = ''
|
||||
format: Format = Format.UNKNOWN
|
||||
ean: str = ''
|
||||
black_white: YesNo = YesNo.UNKNOWN
|
||||
manga: Manga = Manga.UNKNOWN
|
||||
characters: str = ''
|
||||
teams: str = ''
|
||||
locations: str = ''
|
||||
scan_information: str = ''
|
||||
story_arc: str = ''
|
||||
story_arc_number: str = ''
|
||||
series_group: str = ''
|
||||
age_rating: AgeRating = AgeRating.UNKNOWN
|
||||
community_rating: int = -1
|
||||
main_character_or_team: str = ''
|
||||
review: str = ''
|
||||
_filepath: str = ''
|
||||
class BaseModel:
|
||||
|
||||
def __init__(self, kwargs: dict):
|
||||
_set(self, FIELDS, kwargs)
|
||||
def __init__(self, fields: dict, **kwargs):
|
||||
"""
|
||||
Initializes the BaseModel instance.
|
||||
|
||||
def _get(self) -> dict:
|
||||
return _get(self, FIELDS)
|
||||
Args:
|
||||
fields (dict): A dictionary mapping attribute names to tuples containing attribute display names
|
||||
and their expected types.
|
||||
**kwargs: Additional keyword arguments for initializing attributes.
|
||||
|
||||
Attributes:
|
||||
__fields (dict): Stores the fields dictionary passed during initialization.
|
||||
"""
|
||||
self.__fields = fields
|
||||
for key, (_, field_type) in self.__fields.items():
|
||||
# Set default values for each attribute based on its type
|
||||
setattr(self, key, kwargs.get(key, default_attr(field_type)))
|
||||
|
||||
def __setattr__(self, key: str, value: any) -> None:
|
||||
"""
|
||||
Sets the value of an attribute and verifies its type.
|
||||
|
||||
Args:
|
||||
key (str): The name of the attribute to set.
|
||||
value (any): The value to assign to the attribute.
|
||||
|
||||
Raises:
|
||||
TypeError: If the assigned value does not match the expected type for the attribute.
|
||||
"""
|
||||
try:
|
||||
field_type = self.__fields[key][1]
|
||||
# Convert value to the specified type if necessary
|
||||
if field_type not in (int, str, bool):
|
||||
value = field_type(value)
|
||||
# Verify that the assigned value matches the expected type
|
||||
verify_attr(field_type, key, value)
|
||||
except (AttributeError, KeyError):
|
||||
pass
|
||||
super().__setattr__(key, value)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
"""
|
||||
Returns a string representation of the object.
|
||||
|
||||
Returns:
|
||||
str: A string representation of the object, displaying its class name and attribute key-value pairs.
|
||||
"""
|
||||
return '{name}({items})'.format(
|
||||
name=self.__class__.__name__,
|
||||
items=', '.join([f'{k}={repr(v)}' for k, v in self.__dict__.items() if not k.startswith('_')])
|
||||
)
|
||||
|
||||
|
||||
class PageModel:
|
||||
class ComicModel(BaseModel):
|
||||
"""
|
||||
Model for representing comic book metadata.
|
||||
"""
|
||||
title: str
|
||||
series: str
|
||||
number: int
|
||||
count: int
|
||||
volume: int
|
||||
alternate_series: str
|
||||
alternate_number: int
|
||||
alternate_count: int
|
||||
summary: str
|
||||
notes: str
|
||||
year: int
|
||||
month: int
|
||||
day: int
|
||||
writer: str
|
||||
penciller: str
|
||||
inker: str
|
||||
colorist: str
|
||||
letterer: str
|
||||
cover_artist: str
|
||||
editor: str
|
||||
translator: str
|
||||
publisher: str
|
||||
imprint: str
|
||||
genre: str
|
||||
tags: str
|
||||
web: str
|
||||
format: Format
|
||||
ean: str
|
||||
black_white: YesNo
|
||||
manga: Manga
|
||||
characters: str
|
||||
teams: str
|
||||
locations: str
|
||||
scan_information: str
|
||||
story_arc: str
|
||||
story_arc_number: int
|
||||
series_group: str
|
||||
age_rating: AgeRating
|
||||
main_character_or_team: str
|
||||
review: str
|
||||
language_iso: LanguageISO
|
||||
community_rating: Rating
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
"""
|
||||
Initializes a ComicModel instance.
|
||||
|
||||
Args:
|
||||
**kwargs: Keyword arguments used to initialize attributes of the ComicModel.
|
||||
"""
|
||||
super(ComicModel, self).__init__(COMIC_FIELDS, **kwargs)
|
||||
|
||||
|
||||
class PageModel(BaseModel):
|
||||
"""
|
||||
Model for representing comic book pages.
|
||||
"""
|
||||
type: PageType
|
||||
double: bool
|
||||
image_size: int
|
||||
key: str
|
||||
bookmark: str
|
||||
image_width: int
|
||||
image_height: int
|
||||
suffix: str
|
||||
_content: bytes
|
||||
type: PageType = PageType.STORY
|
||||
double: bool = False
|
||||
_image_size: int = 0
|
||||
key: str = ''
|
||||
bookmark: str = ''
|
||||
_image_width: int = 0
|
||||
_image_height: int = 0
|
||||
__content: bytes
|
||||
|
||||
def __init__(self, kwargs: dict):
|
||||
_set(self, PAGE_FIELDS, kwargs)
|
||||
def __init__(self, **kwargs):
|
||||
"""
|
||||
Initializes a PageModel instance.
|
||||
|
||||
def _get(self) -> dict:
|
||||
return _get(self, PAGE_FIELDS)
|
||||
Args:
|
||||
**kwargs: Keyword arguments used to initialize attributes of the PageModel.
|
||||
"""
|
||||
super(PageModel, self).__init__(PAGE_FIELDS, **kwargs)
|
||||
|
||||
Reference in New Issue
Block a user