-
Notifications
You must be signed in to change notification settings - Fork 0
/
epubreader.py
181 lines (146 loc) · 6.97 KB
/
epubreader.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
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
from gi.repository import GObject, GLib, Gdk, Gtk, Gio, WebKit
import epubserver
from readersettings import ReaderSettings
NONE, WAITING, ACCEPT, REJECT = range(4)
class DNDWebView(WebKit.WebView):
def __init__(self, *args):
WebKit.WebView.__init__(self, *args)
self.accept_dnd = NONE
self.dnd_data = None
def do_drag_motion(self, context, x, y, time, user_data=None):
if self.accept_dnd is REJECT:
Gdk.drag_status(context, 0, time)
elif self.accept_dnd is NONE:
target = self.drag_dest_find_target(context, None)
if target == 0: #Gdk.NONE:
self.accept_dnd = REJECT
Gdk.drag_status(context, 0, time)
else:
self.drag_get_data(context, target, time)
self.accept_dnd = WAITING
return True # Use drag_status to say yes or no; otherwise drag_leave called
def do_drag_data_received(self, context, x, y, data, info, time):
self.dnd_data = data.get_text().strip()
if self.dnd_data.endswith('.epub'):
self.accept_dnd = ACCEPT
Gdk.drag_status(context, Gdk.DragAction.COPY, time)
else:
self.accept_dnd = REJECT
Gdk.drag_status(context, 0, time)
def do_drag_drop(self, context, x, y, time, user_data=None):
Gtk.drag_finish(context, True, False, time)
def do_drag_leave(self, context, time, user_data=None):
self.accept_dnd = NONE
class EpubReader(Gtk.ApplicationWindow):
def __init__(self, application, filename=""):
Gtk.ApplicationWindow.__init__(self, application=application,
default_width=450, default_height=600)
self.application = application
self.load_file_lazy(filename)
self.establish_actions()
sw = Gtk.ScrolledWindow()
sw.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
self.view = DNDWebView()
self.view.connect('drag-drop', self.on_drag_drop)
sw.add(self.view)
self.add(sw)
self.view.connect('title-changed', self.on_title_changed)
self.view.connect('console-message', self.on_console_message)
self.hb = Gtk.HeaderBar()
self.hb.set_show_close_button(True)
self.settings_button = Gtk.Button.new_from_icon_name('emblem-system-symbolic', Gtk.IconSize.BUTTON)
self.settings_button.connect('clicked', self.on_settings)
self.hb.pack_end(self.settings_button)
self.toc_button = Gtk.Button.new_from_icon_name('view-list-symbolic', Gtk.IconSize.BUTTON)
self.toc_button.connect('clicked', self.on_toc)
self.hb.pack_start(self.toc_button)
self.open_button = Gtk.Button.new_from_icon_name('document-open-symbolic', Gtk.IconSize.BUTTON)
self.open_button.connect('clicked', self.application.on_open)
self.hb.pack_start(self.open_button)
self.set_titlebar(self.hb)
self.set_title("Berg")
# Work-around for application title
# http://stackoverflow.com/questions/9324163/how-to-set-application-title-in-gnome-shell
self.set_wmclass("Berg", "Berg")
self.connect('key-press-event', self.on_key_press)
self.connect('configure-event', self.on_configure)
self._size = (0, 0)
self._resize_timeout = None
self.settings = ReaderSettings(self)
self.show_all()
self.toc_button.hide()
self.settings_button.hide()
self.spawn_server()
def establish_actions(self):
action_group = Gtk.ActionGroup('main')
action_group.add_actions((
('quit', Gtk.STOCK_QUIT, "Quit", "<control>w", None, self.on_quit),
('settings',Gtk.STOCK_PREFERENCES, "Settings", None, None, self.on_settings),
('reload', Gtk.STOCK_REFRESH, "Refresh", "<control>r", None, self.on_reload)
))
# Note: ui_manager must be property of self, or accelerators don't work. This is probably
# because it would be garbage collected otherwise.
self.ui_manager = Gtk.UIManager()
self.ui_manager.insert_action_group(action_group)
self.ui_manager.add_ui_from_string('''
<ui>
<accelerator action="quit"/>
<accelerator action="reload"/>
</ui>
''')
self.ui_manager.ensure_update()
self.add_accel_group(self.ui_manager.get_accel_group())
def spawn_server(self):
self.server = epubserver.EpubServer()
self.port = self.server.get_port()
def load_file_lazy(self, filename):
GLib.idle_add(self.load_file, filename)
def load_file(self, filename):
self.view.load_uri("http://localhost:%i/?load=%s" % (self.port, filename))
if filename:
self.open_button.hide()
self.settings_button.show()
self.toc_button.show()
def on_drag_drop(self, widget, context, x, y, time, data=None):
filename = self.view.dnd_data
if filename.startswith('file://'):
filename = filename[7:]
self.application.load_file(filename)
def on_quit(self, *args):
self.destroy()
def on_settings(self, *args):
self.settings.show()
def on_reload(self, *args):
self.view.reload()
def on_toc(self, *args):
self.view.execute_script('reader.showTOC();')
def on_key_press(self, widget, event):
# Check that none of Shift, Control, Alt are pressed
if ((Gdk.ModifierType.SHIFT_MASK | Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.MOD1_MASK)
& event.state) == 0:
if event.keyval in (Gdk.KEY_Right, Gdk.KEY_Down, Gdk.KEY_space, Gdk.KEY_period):
self.change_page(1)
return True
if event.keyval in (Gdk.KEY_Left, Gdk.KEY_Up, Gdk.KEY_BackSpace, Gdk.KEY_comma):
self.change_page(-1)
return True
return False
def on_configure(self, widget, event):
if self._size == (event.width, event.height):
return
self._size = (event.width, event.height)
if self._resize_timeout is not None:
GObject.source_remove(self._resize_timeout)
self._resize_timeout = GObject.timeout_add(500,
lambda *args: self.view.execute_script('reader.resized();'))
def on_title_changed(self, web_view, frame, title):
self.set_title(title)
def set_title(self, title):
self.hb.set_title(title)
def on_console_message(self, web_view, message, line, source_id):
if message == 'Ready':
self.settings.update_styles()
return True
return not self.application.debug
def change_page(self, direction=1):
self.view.execute_script('reader.moveTo({direction: %i})' % direction)