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
|
import logging
import sqlite3
import time
import gconf
from readbookmark import Bookmark
_logger = logging.getLogger('read-activity')
class BookmarkManager:
def __init__(self, filehash, dbpath='read.db'):
self._filehash = filehash
self._conn = sqlite3.connect(dbpath)
self._cur = self._conn.cursor()
self._bookmarks = []
self._populate_bookmarks()
def add_bookmark(self, page, title, local=1):
# locale = 0 means that this is a bookmark originally
# created by the person who originally shared the file
timestamp = time.time()
client = gconf.client_get_default()
user = client.get_string("/desktop/sugar/user/nick")
color = client.get_string("/desktop/sugar/user/color")
t = (self._filehash, page, title, timestamp, user, color, local)
self._cur.execute('insert into bookmarks values (?, ?, ?, ?, ?, ?, ?)', t)
self._conn.commit()
self._resync_bookmark_cache()
def del_bookmark(self, page):
client = gconf.client_get_default()
user = client.get_string("/desktop/sugar/user/nick")
# We delete only the locally made bookmark
t = (self._filehash, page, user)
self._cur.execute('delete from bookmarks where md5=? and page=? and user=?', t)
self._conn.commit()
self._resync_bookmark_cache()
def _populate_bookmarks(self):
# TODO: Figure out if caching the entire set of bookmarks is a good idea or not
self._cur.execute('select * from bookmarks where md5=? order by page', (self._filehash,))
for row in self._cur:
self._bookmarks.append(Bookmark(row))
def get_bookmarks_for_page(self, page):
bookmarks = []
for bookmark in self._bookmarks:
if bookmark.belongstopage(page):
bookmarks.append(bookmark)
return bookmarks
def _resync_bookmark_cache(self):
# To be called when a new bookmark has been added/removed
self._bookmarks = []
self._populate_bookmarks()
def get_prev_bookmark_for_page(self, page, wrap = True):
if not len(self._bookmarks):
return None
if page <= self._bookmarks[0].page_no and wrap:
return self._bookmarks[-1]
else:
for i in range(page-1, -1, -1):
for bookmark in self._bookmarks:
if bookmark.belongstopage(i):
return bookmark
return None
def get_next_bookmark_for_page(self, page, wrap = True):
if not len(self._bookmarks):
return None
if page >= self._bookmarks[-1].page_no and wrap:
return self._bookmarks[0]
else:
for i in range(page+1, self._bookmarks[-1].page_no + 1):
for bookmark in self._bookmarks:
if bookmark.belongstopage(i):
return bookmark
return None
|
