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
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
# Copyright (C) 2012 Aleksey Lim
#
# 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 3 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, see <http://www.gnu.org/licenses/>.
import os
import sys
import logging
from shutil import copyfileobj
from tempfile import NamedTemporaryFile
from sugar_network import client
from sugar_network.toolkit.router import Blob, route, Request
from sugar_network.toolkit import enforce
_logger = logging.getLogger('client.journal')
_ds_root = client.profile_path('datastore')
def exists(guid):
return os.path.exists(_ds_path(guid))
def get(guid, prop):
path = _prop_path(guid, prop)
if not os.path.exists(path):
return None
with file(path, 'rb') as f:
return f.read()
class Routes(object):
_ds = None
def __init__(self):
import dbus
try:
self._ds = dbus.Interface(
dbus.SessionBus().get_object(
'org.laptop.sugar.DataStore',
'/org/laptop/sugar/DataStore'),
'org.laptop.sugar.DataStore')
except dbus.DBusException:
_logger.info(
'Cannot connect to sugar-datastore, '
'Journal integration is disabled')
@route('GET', ['journal'], mime_type='application/json', arguments={
'offset': int,
'limit': int,
'reply': ('uid', 'title', 'description', 'preview'),
'order_by': list,
})
def journal_find(self, request, response):
enforce(self._ds is not None, 'Journal is inaccessible')
import dbus
reply = request.pop('reply')
if 'preview' in reply:
reply.remove('preview')
has_preview = True
else:
has_preview = False
for key in ('timestamp', 'filesize', 'creation_time'):
value = request.get(key)
if not value or '..' not in value:
continue
start, end = value.split('..', 1)
value = {'start': start or '0', 'end': end or str(sys.maxint)}
request[key] = dbus.Dictionary(value)
if 'uid' not in reply:
reply.append('uid')
result, total = self._ds.find(request, reply, byte_arrays=True)
for item in result:
# Do not break SN like API
guid = item['guid'] = item.pop('uid')
if has_preview:
item['preview'] = _preview(guid)
return {'result': result, 'total': int(total)}
@route('GET', ['journal', None], mime_type='application/json')
def journal_get(self, request, response):
guid = request.guid
return {'guid': guid,
'title': get(guid, 'title'),
'description': get(guid, 'description'),
'preview': _preview(guid),
}
@route('GET', ['journal', None, 'preview'])
def journal_get_preview(self, request, response):
return Blob({
'blob': _prop_path(request.guid, 'preview'),
'mime_type': 'image/png',
})
@route('GET', ['journal', None, 'data'])
def journal_get_data(self, request, response):
return Blob({
'blob': _ds_path(request.guid, 'data'),
'mime_type': get(request.guid, 'mime_type') or 'application/octet',
})
@route('GET', ['journal', None, None], mime_type='application/json')
def journal_get_prop(self, request, response):
return get(request.guid, request.prop)
@route('PUT', ['journal', None], cmd='share')
def journal_share(self, request, response):
enforce(self._ds is not None, 'Journal is inaccessible')
guid = request.guid
preview_path = _prop_path(guid, 'preview')
enforce(os.access(preview_path, os.R_OK), 'No preview')
data_path = _ds_path(guid, 'data')
enforce(os.access(data_path, os.R_OK), 'No data')
subrequest = Request(method='POST', document='artifact')
subrequest.content = request.content
subrequest.content_type = 'application/json'
# pylint: disable-msg=E1101
subguid = self.fallback(subrequest, response)
subrequest = Request(method='PUT', document='artifact',
guid=subguid, prop='preview')
subrequest.content_type = 'image/png'
with file(preview_path, 'rb') as subrequest.content_stream:
self.fallback(subrequest, response)
subrequest = Request(method='PUT', document='artifact',
guid=subguid, prop='data')
subrequest.content_type = get(guid, 'mime_type') or 'application/octet'
with file(data_path, 'rb') as subrequest.content_stream:
self.fallback(subrequest, response)
def journal_update(self, guid, data=None, **kwargs):
enforce(self._ds is not None, 'Journal is inaccessible')
preview = kwargs.get('preview')
if preview:
if hasattr(preview, 'read'):
preview = preview.read()
if hasattr(preview, 'close'):
preview.close()
elif isinstance(preview, dict):
with file(preview['blob'], 'rb') as f:
preview = f.read()
import dbus
kwargs['preview'] = dbus.ByteArray(preview)
if hasattr(data, 'read'):
with NamedTemporaryFile(delete=False) as f:
copyfileobj(data, f)
data = f.name
transfer_ownership = True
elif isinstance(data, dict):
data = data['blob']
transfer_ownership = False
elif data is not None:
with NamedTemporaryFile(delete=False) as f:
f.write(data)
data = f.name
transfer_ownership = True
self._ds.update(guid, kwargs, data or '', transfer_ownership)
def journal_delete(self, guid):
enforce(self._ds is not None, 'Journal is inaccessible')
self._ds.delete(guid)
def _ds_path(guid, *args):
return os.path.join(_ds_root, guid[:2], guid, *args)
def _prop_path(guid, prop):
return _ds_path(guid, 'metadata', prop)
def _preview(guid):
return {'url': 'http://127.0.0.1:%s/journal/%s/preview' %
(client.ipc_port.value, guid),
}
|
