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
|
# Copyright 2009 One Laptop Per Child
# Author: Sayamindu Dasgupta <sayamindu@laptop.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import logging
import os, os.path
import shutil
import sqlite3
import time
import gconf
from readbookmark import Bookmark
_logger = logging.getLogger('read-activity')
class BookmarkManager:
def __init__(self, filehash, dbfile='read.db'):
self._filehash = filehash
dbpath = os.path.join(os.environ['SUGAR_ACTIVITY_ROOT'], 'data', \
dbfile)
if not os.path.exists(dbpath):
# This makes me nervous
srcpath = os.path.join(os.environ['SUGAR_BUNDLE_PATH'], 'read.db')
shutil.copy(srcpath, dbpath)
self._conn = sqlite3.connect(dbpath)
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._conn.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._conn.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
rows = self._conn.execute('select * from bookmarks where md5=? order by page', (self._filehash,))
for row in rows:
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
|
