-
-
Notifications
You must be signed in to change notification settings - Fork 21
/
Copy pathui.py
160 lines (141 loc) · 5.03 KB
/
ui.py
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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
from functools import partial
from typing import TYPE_CHECKING, Any, Iterator
from calibre.gui2 import Dispatcher
from calibre.gui2.actions import InterfaceAction
from calibre.gui2.threaded_jobs import ThreadedJob
from PyQt6.QtGui import QIcon
from .custom_x_ray import CustomXRayDialog
from .error_dialogs import job_failed, unsupported_ww_lang_dialog
from .metadata import MetaDataResult, check_metadata
from .parse_job import ParseJobData, do_job
from .send_file import SendFile, device_connected
from .utils import donate
load_translations() # type: ignore
if TYPE_CHECKING:
_: Any
class WordDumb(InterfaceAction):
name = "WordDumb"
action_spec = ("WordDumb", None, "Good morning Krusty Crew!", None)
action_type = "current"
action_add_menu = True
action_menu_clone_qaction = _("Create Word Wise and X-Ray")
def genesis(self):
self.qaction.setIcon(get_icons("starfish.svg", "WordDumb")) # type: ignore
self.menu = self.qaction.menu()
self.qaction.triggered.connect(partial(run, self.gui, True, True))
self.create_menu_action(
self.menu,
"Word Wise",
_("Create Word Wise"),
triggered=partial(run, self.gui, True, False),
)
self.create_menu_action(
self.menu,
"X-Ray",
_("Create X-Ray"),
triggered=partial(run, self.gui, False, True),
)
self.menu.addSeparator()
self.create_menu_action(
self.menu,
"Customize X-Ray",
_("Customize X-Ray"),
icon=QIcon.ic("polish.png"),
triggered=self.open_custom_x_ray_dialog,
)
self.create_menu_action(
self.menu,
"Preferences",
_("Preferences"),
icon=QIcon.ic("config.png"),
triggered=self.config,
)
self.menu.addSeparator()
self.create_menu_action(
self.menu,
"Donate",
_("Donate"),
icon=QIcon.ic("donate.png"),
description="I need about tree-fiddy.",
triggered=donate,
)
self.qaction.setMenu(self.menu)
def config(self):
self.interface_action_base_plugin.do_user_config(self.gui)
def open_custom_x_ray_dialog(self) -> None:
for md_result in get_metadata_of_selected_books(self.gui, True):
custom_x_dlg = CustomXRayDialog(
md_result.book_paths[0], md_result.mi.get("title"), self.gui
)
if custom_x_dlg.exec():
custom_x_dlg.x_ray_model.save_data()
def get_metadata_of_selected_books(
gui: Any, custom_x_ray: bool
) -> Iterator[MetaDataResult]:
return filter(
None,
[
check_metadata(gui, book_id, custom_x_ray)
for book_id in map(
gui.library_view.model().id,
gui.library_view.selectionModel().selectedRows(),
)
],
)
def run(gui: Any, create_ww: bool, create_x: bool) -> None:
if not create_ww and not create_x:
return
for md_result in get_metadata_of_selected_books(gui, False):
for book_fmt, book_path, support_ww in zip(
md_result.book_fmts, md_result.book_paths, md_result.support_ww_list
):
if create_ww and not support_ww:
create_ww = False
unsupported_ww_lang_dialog()
if create_x and not md_result.support_x_ray:
create_x = False
if not create_ww and not create_x:
continue
title = (
f"{md_result.mi.get('title')}({md_result.book_fmt})"
if len(md_result.book_fmts) > 1
else md_result.mi.get("title")
)
notif = []
if create_ww:
notif.append(_("Word Wise"))
if create_x:
notif.append("X-Ray")
notif = _(" and ").join(notif)
job_data = ParseJobData(
book_id=md_result.book_id,
book_path=book_path,
mi=md_result.mi,
book_fmt=book_fmt,
book_lang=md_result.book_lang,
create_ww=create_ww,
create_x=create_x,
)
job = ThreadedJob(
"WordDumb's dumb job",
_("Generating {} for {}").format(notif, title),
do_job,
(job_data,),
{},
Dispatcher(
partial(
done,
gui=gui,
notif=_("{} generated for {}").format(notif, title),
)
),
killable=False,
)
gui.job_manager.run_threaded_job(job)
def done(job, gui=None, notif=None):
if job_failed(job, gui):
return
if package_name := device_connected(gui, job.result.book_fmt):
SendFile(gui, job.result, package_name, notif).send_files(None)
else:
gui.status_bar.show_message(notif)