ebookRen/models_core.py at master · Piotr-Grechuta/ebookRen · GitHub

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
from __future__ import annotations

from dataclasses import dataclass, field
from pathlib import Path


@dataclass(frozen=True)
class Candidate:
    score: int
    series: str
    volume: tuple[int, str] | None
    title_override: str | None
    source: str


@dataclass
class EpubMetadata:
    path: Path
    stem: str
    segments: list[str]
    core: str
    title: str = ""
    creators: list[str] = field(default_factory=list)
    identifiers: list[str] = field(default_factory=list)
    subjects: list[str] = field(default_factory=list)
    meta_series: str = ""
    meta_volume: tuple[int, str] | None = None
    errors: list[str] = field(default_factory=list)


@dataclass
class RenameMove:
    source: Path
    temp: Path | None
    destination: Path
    record: object | None = None
    operation: str = "rename"


@dataclass
class UndoPlan:
    folder: Path
    moves: list[RenameMove]
    total_rows: int


@dataclass
class InferenceResult:
    record: object
    base_confidence: int
    title_from_core: bool


@dataclass(frozen=True)
class RankedOnlineMatch:
    providers: list[str]
    sources: list[str]
    title: str
    authors: list[str]
    identifiers: list[str]
    score: int
    reason: str
    series: str = ""
    volume: tuple[int, str] | None = None
    genre: str = ""
    cycle_source: str = ""


@dataclass(frozen=True)
class OnlineCandidate:
    provider: str
    source: str
    title: str
    authors: list[str]
    identifiers: list[str]
    score: int
    reason: str
    series: str = ""
    volume: tuple[int, str] | None = None
    genre: str = ""
    cycle_source: str = ""


@dataclass(frozen=True)
class LubimyczytacResult:
    title: str
    authors: list[str]
    series: str = ""
    volume: tuple[int, str] | None = None
    url: str = ""
    genres: list[str] = field(default_factory=list)
    cycle_source: str = ""


@dataclass(frozen=True)
class OnlineVerification:
    checked: bool
    author_confirmed: bool
    title_confirmed: bool
    series_confirmed: bool
    volume_confirmed: bool
    providers: list[str]


@dataclass(frozen=True)
class OnlineRoleEvidence:
    authors: dict[str, str]
    titles: dict[str, str]
    series: dict[str, str]
    volumes: set[tuple[int, str]]


@dataclass(frozen=True)
class OnlineSeriesEvidence:
    series: str = ""
    volume: tuple[int, str] | None = None
    cycle_source: str = ""
    authoritative: bool = False


@dataclass(frozen=True)
class HybridLocalParse:
    title_hint: str = ""
    author_hint: str = ""
    volume_hint: tuple[int, str] | None = None
    source: str = ""
    confidence: int = 0


@dataclass(frozen=True)
class LocalPrototype:
    path: Path
    author: str
    series: str
    volume: tuple[int, str] | None
    title: str
    genre: str
    source: str
    confidence: int
    title_from_core: bool = False
    author_from_trailing_core: bool = False