| author | Production Moz <dev@sopinspace.com> |
| Tue, 31 May 2011 17:52:28 +0200 | |
| changeset 344 | 9787360440db |
| parent 327 | d858c3bb07fe |
| child 347 | 0a359382701b |
| permissions | -rw-r--r-- |
|
156
6d447220fd1e
bug fix: removed \r in textversion's content because of a comment position bug in pre_edit. Added a save function to TextVersion to do so.
rbernard
parents:
155
diff
changeset
|
1 |
import re |
| 0 | 2 |
from cm.converters.pandoc_converters import \ |
3 |
CHOICES_INPUT_FORMATS as CHOICES_INPUT_FORMATS_PANDOC, \ |
|
4 |
DEFAULT_INPUT_FORMAT as DEFAULT_INPUT_FORMAT_PANDOC, pandoc_convert |
|
5 |
from cm.models_base import PermanentModel, KeyManager, Manager, KeyModel, AuthorModel |
|
6 |
from cm.models_utils import * |
|
7 |
from cm.utils.dj import absolute_reverse |
|
8 |
from cm.utils.date import datetime_to_user_str |
|
| 175 | 9 |
from cm.utils.html import on_content_receive |
| 0 | 10 |
from cm.utils.comment_positioning import compute_new_comment_positions |
11 |
from django import forms |
|
12 |
from django.db.models import Q |
|
13 |
from django.template.loader import render_to_string |
|
14 |
from django.conf import settings |
|
15 |
from django.template import RequestContext |
|
16 |
from django.contrib.auth.models import Permission |
|
17 |
from django.contrib.contenttypes import generic |
|
18 |
from django.contrib.contenttypes.models import ContentType |
|
19 |
from django.core.files.base import ContentFile |
|
20 |
from django.core.urlresolvers import reverse |
|
21 |
from django.template.defaultfilters import timesince |
|
22 |
from django.db import models |
|
23 |
from django.utils.translation import ugettext as _, ugettext_lazy, ugettext_noop |
|
24 |
from tagging.fields import TagField |
|
25 |
import pickle |
|
26 |
from django.db import connection |
|
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
27 |
from datetime import datetime |
| 0 | 28 |
|
29 |
class TextManager(Manager): |
|
30 |
def create_text(self, title, format, content, note, name, email, tags, user=None, state='approved', **kwargs): |
|
| 175 | 31 |
content = on_content_receive(content, format) |
| 0 | 32 |
text = self.create(name=name, email=email, user=user, state=state) |
33 |
text_version = TextVersion.objects.create(title=title, format=format, content=content, text=text, note=note, name=name, email=email, tags=tags, user=user) |
|
34 |
return text |
|
35 |
||
36 |
def create_new_version(self, text, title, format, content, note, name, email, tags, user=None, **kwargs): |
|
37 |
text_version = TextVersion.objects.create(title=title, format=format, content=content, text=text, note=note, name=name, email=email, tags=tags, user=user) |
|
38 |
return text_version |
|
39 |
||
40 |
class Text(PermanentModel, AuthorModel): |
|
41 |
modified = models.DateTimeField(auto_now=True) |
|
42 |
created = models.DateTimeField(auto_now_add=True) |
|
43 |
||
44 |
private_feed_key = models.CharField(max_length=20, db_index=True, unique=True, blank=True, null=True, default=None) |
|
45 |
||
46 |
# denormalized fields |
|
47 |
last_text_version = models.ForeignKey("TextVersion", related_name='related_text', null=True, blank=True) |
|
48 |
title = models.TextField() |
|
49 |
||
50 |
objects = TextManager() |
|
51 |
||
52 |
def get_latest_version(self): |
|
53 |
return self.last_text_version |
|
54 |
||
55 |
def fetch_latest_version(self): |
|
56 |
versions = self.get_versions() |
|
57 |
if versions: |
|
58 |
return versions[0] |
|
59 |
else: |
|
60 |
return None |
|
61 |
||
62 |
def update_denorm_fields(self): |
|
63 |
real_last_text_version = self.fetch_latest_version() |
|
64 |
||
| 289 | 65 |
try: |
66 |
last_text_version = self.last_text_version |
|
67 |
except TextVersion.DoesNotExist: |
|
68 |
# the text version has just been deleted |
|
69 |
last_text_version = None |
|
70 |
||
| 0 | 71 |
modif = False |
| 289 | 72 |
if real_last_text_version and real_last_text_version != last_text_version: |
| 0 | 73 |
self.last_text_version = real_last_text_version |
74 |
modif = True |
|
75 |
||
76 |
if real_last_text_version and real_last_text_version.title and real_last_text_version.title != self.title: |
|
77 |
self.title = real_last_text_version.title |
|
78 |
modif = True |
|
79 |
||
80 |
if real_last_text_version and real_last_text_version.modified != self.modified: |
|
81 |
self.modified = real_last_text_version.modified |
|
82 |
modif = True |
|
83 |
||
84 |
if modif: |
|
85 |
self.save() |
|
86 |
||
87 |
||
88 |
def get_title(self): |
|
89 |
return self.get_latest_version().title |
|
90 |
||
91 |
def get_versions(self): |
|
92 |
""" |
|
93 |
Versions with most recent first |
|
94 |
""" |
|
95 |
versions = TextVersion.objects.filter(text__exact=self).order_by('-created') |
|
96 |
# TODO: use new postgresql 8.4 row_number as extra select to do that |
|
| 145 | 97 |
#for index in xrange(len(versions)): |
98 |
# v = versions[index] |
|
99 |
# # version_number is 1-based |
|
100 |
# setattr(v, 'version_number', len(versions) - index) |
|
| 0 | 101 |
return versions |
102 |
||
103 |
def get_version(self, version_number): |
|
104 |
""" |
|
105 |
Get version number 'version_number' (1-based) |
|
106 |
""" |
|
107 |
version = TextVersion.objects.filter(text__exact=self).order_by('created')[version_number - 1:version_number][0] |
|
108 |
return version |
|
109 |
||
110 |
def get_inversed_versions(self): |
|
111 |
versions = TextVersion.objects.filter(text__exact=self).order_by('created') |
|
112 |
# TODO: use new postgresql 8.4 row_number as extra select to do that |
|
| 145 | 113 |
#for index in xrange(len(versions)): |
114 |
# v = versions[index] |
|
115 |
# # version_number is 1-based |
|
116 |
# setattr(v, 'version_number', index + 1) |
|
| 0 | 117 |
return versions |
118 |
||
119 |
def get_versions_number(self): |
|
120 |
return self.get_versions().count() |
|
121 |
||
122 |
def is_admin(self, adminkey=None): |
|
123 |
if adminkey and self.adminkey == adminkey: |
|
124 |
return True |
|
125 |
else: |
|
126 |
return False |
|
127 |
||
| 145 | 128 |
def revert_to_version(self, text_version_key): |
129 |
text_version = TextVersion.objects.get(key=text_version_key) |
|
130 |
if text_version.text != self: |
|
131 |
raise Exception('Invalid revert attempt') |
|
| 0 | 132 |
new_text_version = TextVersion.objects.duplicate(text_version, True) |
133 |
return new_text_version |
|
134 |
||
| 103 | 135 |
def edit(self, new_title, new_format, new_content, new_tags=None, new_note=None, keep_comments=True, cancel_modified_scopes=True, new_version=True): |
| 0 | 136 |
text_version = self.get_latest_version() |
137 |
||
138 |
if new_version: |
|
139 |
text_version = TextVersion.objects.duplicate(text_version, keep_comments) |
|
| 103 | 140 |
text_version.edit(new_title, new_format, new_content, new_tags, new_note, keep_comments, cancel_modified_scopes) |
| 0 | 141 |
return text_version |
142 |
||
143 |
def __unicode__(self): |
|
144 |
return self.title |
|
145 |
||
146 |
DEFAULT_INPUT_FORMAT = getattr(settings, 'DEFAULT_INPUT_FORMAT', DEFAULT_INPUT_FORMAT_PANDOC) |
|
147 |
CHOICES_INPUT_FORMATS = getattr(settings, 'CHOICES_INPUT_FORMATS', CHOICES_INPUT_FORMATS_PANDOC) |
|
148 |
||
| 24 | 149 |
class TextVersionManager(KeyManager): |
| 0 | 150 |
|
151 |
def duplicate(self, text_version, duplicate_comments=True): |
|
152 |
#import pdb;pdb.set_trace() |
|
153 |
old_comment_set = set(text_version.comment_set.all()) |
|
154 |
text_version.id = None |
|
| 24 | 155 |
|
156 |
# generate new key |
|
157 |
text_version.key = self._gen_key() |
|
158 |
text_version.adminkey = self._gen_adminkey() |
|
159 |
||
| 0 | 160 |
text_version.save() |
161 |
||
162 |
duplicate_text_version = text_version |
|
163 |
||
164 |
if duplicate_comments: |
|
165 |
old_comment_map = {} |
|
166 |
while len(old_comment_set): |
|
167 |
for c in old_comment_set: |
|
168 |
if not c.reply_to or c.reply_to.id in old_comment_map: |
|
169 |
old_id = c.id |
|
170 |
old_comment_set.remove(c) |
|
171 |
reply_to = None |
|
172 |
if c.reply_to: |
|
173 |
reply_to = old_comment_map[c.reply_to.id] |
|
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
174 |
c2 = Comment.objects.duplicate(c, duplicate_text_version, reply_to, keep_dates=True) |
| 0 | 175 |
old_comment_map[old_id] = c2 |
176 |
break |
|
177 |
||
178 |
return duplicate_text_version |
|
179 |
||
| 24 | 180 |
class TextVersion(AuthorModel, KeyModel): |
| 0 | 181 |
modified = models.DateTimeField(auto_now=True) |
182 |
created = models.DateTimeField(auto_now_add=True) |
|
183 |
||
184 |
title = models.TextField(ugettext_lazy("Title")) |
|
185 |
format = models.CharField(ugettext_lazy("Format"), max_length=20, blank=False, default=DEFAULT_INPUT_FORMAT, choices=CHOICES_INPUT_FORMATS) |
|
186 |
content = models.TextField(ugettext_lazy("Content")) |
|
187 |
tags = TagField(ugettext_lazy("Tags"), max_length=1000) |
|
188 |
||
189 |
note = models.CharField(ugettext_lazy("Note"), max_length=100, null=True, blank=True) |
|
190 |
||
191 |
mod_posteriori = models.BooleanField(ugettext_lazy('Moderation a posteriori?'), default=True) |
|
192 |
||
193 |
text = models.ForeignKey("Text") |
|
194 |
||
195 |
objects = TextVersionManager() |
|
196 |
||
197 |
def get_content(self, format='html'): |
|
|
259
0371caf8bcc6
always use pandoc but in raw mode for html->html convert
raph
parents:
252
diff
changeset
|
198 |
return pandoc_convert(self.content, self.format, format) |
| 0 | 199 |
# def _get_comments(self, user = None, filter_reply = 0): |
200 |
# """ |
|
201 |
# get comments viewable by this user (user = None or user = AnonymousUser => everyone) |
|
202 |
# filter_reply = 0: comments and replies |
|
203 |
# 1: comments |
|
204 |
# 2: replies |
|
205 |
# """ |
|
206 |
# from cm.security import has_perm_on_text # should stay here to avoid circular dependencies |
|
207 |
# |
|
208 |
# if has_perm(user, 'can_view_unapproved_comment', self.text): |
|
209 |
# comments = self.comment_set.all() |
|
210 |
# elif has_perm(user, 'can_view_approved_comment', self.text): |
|
211 |
# comments = self.comment_set.filter(visible=True) |
|
|
17
a4be0b8a905d
bug fix : can view own comment on comment add on mod priori
reno
parents:
16
diff
changeset
|
212 |
# elif has_perm(user, 'can_view_comment_own', self.text): |
| 0 | 213 |
# comments = self.comment_set.filter(user=user) |
214 |
# else: |
|
215 |
# return Comment.objects.none() # empty queryset |
|
216 |
# if filter_reply: |
|
217 |
# comments = comments.filter) |
|
218 |
# return comments |
|
219 |
# |
|
220 |
# def get_comments_as_json(self, user = None): |
|
221 |
# return simplejson.dumps(self._get_comments(user, filter_reply=0)) |
|
222 |
# |
|
223 |
# def get_comments_and_replies(self, user = None): |
|
224 |
# return (self.get_comments(user), |
|
225 |
# self.get_replies(user)) |
|
226 |
# |
|
227 |
def get_comments(self): |
|
228 |
"Warning: data access without security" |
|
229 |
return self.comment_set.filter(reply_to=None, deleted=False) |
|
230 |
||
231 |
def get_replies(self): |
|
232 |
"Warning: data access without security" |
|
233 |
return self.comment_set.filter(~Q(reply_to == None), Q(deleted=False)) |
|
234 |
||
235 |
def __unicode__(self): |
|
236 |
return '<%d> %s' % (self.id, self.title) |
|
237 |
||
| 175 | 238 |
def edit(self, new_title, new_format, new_content, new_tags=None, new_note=None, keep_comments=True, cancel_modified_scopes=True): |
239 |
new_content = on_content_receive(new_content, new_format) |
|
| 0 | 240 |
if not keep_comments : |
241 |
self.comment_set.all().delete() |
|
242 |
elif self.content != new_content or new_format != self.format: |
|
243 |
comments = self.get_comments() ; |
|
244 |
tomodify_comments, toremove_comments = compute_new_comment_positions(self.content, self.format, new_content, new_format, comments) |
|
| 103 | 245 |
#print "tomodify_comments" |
246 |
#print tomodify_comments |
|
| 0 | 247 |
#print "toremove_comments",len(toremove_comments) |
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
248 |
[comment.save(keep_dates=True) for comment in tomodify_comments] |
| 103 | 249 |
if cancel_modified_scopes : |
250 |
[comment.remove_scope() for comment in toremove_comments] |
|
251 |
else : |
|
252 |
[comment.delete() for comment in toremove_comments] |
|
253 |
||
| 0 | 254 |
self.title = new_title |
255 |
if new_tags: |
|
256 |
self.tags = new_tags |
|
257 |
if new_note: |
|
258 |
self.note = new_note |
|
259 |
self.content = new_content |
|
260 |
self.format = new_format |
|
261 |
self.save() |
|
| 145 | 262 |
|
263 |
def get_next_version(self): |
|
264 |
other_versions = TextVersion.objects.filter(text__exact=self.text).order_by('created').filter(created__gt=self.created) |
|
265 |
return other_versions[0] if other_versions else None |
|
266 |
if other_versions: |
|
267 |
return |
|
268 |
else: |
|
269 |
return None |
|
270 |
||
271 |
def get_previous_version(self): |
|
272 |
other_versions = TextVersion.objects.filter(text__exact=self.text).order_by('-created').filter(created__lt=self.created) |
|
273 |
return other_versions[0] if other_versions else None |
|
274 |
||
275 |
def get_version_number(self): |
|
276 |
return TextVersion.objects.filter(text__exact=self.text).order_by('created').filter(created__lte=self.created).count() |
|
|
156
6d447220fd1e
bug fix: removed \r in textversion's content because of a comment position bug in pre_edit. Added a save function to TextVersion to do so.
rbernard
parents:
155
diff
changeset
|
277 |
|
| 0 | 278 |
class CommentManager(Manager): |
279 |
||
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
280 |
def duplicate(self, comment, text_version, reply_to=None, keep_dates=False): |
| 0 | 281 |
comment.id = None |
282 |
comment.text_version = text_version |
|
283 |
if reply_to: |
|
284 |
comment.reply_to = reply_to |
|
285 |
self.update_keys(comment) |
|
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
286 |
comment.save(keep_dates=keep_dates) |
| 0 | 287 |
return comment |
288 |
||
|
130
f257ad33bed5
model change+migration (add id_key to comment): run ./bin/django migrate
raph
parents:
103
diff
changeset
|
289 |
from cm.models_base import KEY_MAX_SIZE, generate_key |
|
f257ad33bed5
model change+migration (add id_key to comment): run ./bin/django migrate
raph
parents:
103
diff
changeset
|
290 |
|
| 0 | 291 |
class Comment(PermanentModel, AuthorModel): |
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
292 |
modified = models.DateTimeField() |
|
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
293 |
created = models.DateTimeField() |
| 0 | 294 |
|
|
130
f257ad33bed5
model change+migration (add id_key to comment): run ./bin/django migrate
raph
parents:
103
diff
changeset
|
295 |
# key to identify same comments across versions |
|
f257ad33bed5
model change+migration (add id_key to comment): run ./bin/django migrate
raph
parents:
103
diff
changeset
|
296 |
id_key = models.CharField(max_length=KEY_MAX_SIZE, db_index=True, default=generate_key) |
|
f257ad33bed5
model change+migration (add id_key to comment): run ./bin/django migrate
raph
parents:
103
diff
changeset
|
297 |
|
| 0 | 298 |
text_version = models.ForeignKey("TextVersion") |
299 |
||
300 |
# comment_set will be replies |
|
301 |
reply_to = models.ForeignKey("Comment", null=True, blank=True) |
|
302 |
||
303 |
title = models.TextField() |
|
304 |
content = models.TextField() |
|
305 |
content_html = models.TextField() |
|
306 |
||
307 |
format = models.CharField(_("Format:"), max_length=20, blank=False, default=DEFAULT_INPUT_FORMAT, choices=CHOICES_INPUT_FORMATS) |
|
308 |
||
309 |
tags = TagField() |
|
310 |
||
311 |
start_wrapper = models.IntegerField(null=True, blank=True) |
|
312 |
end_wrapper = models.IntegerField(null=True, blank=True) |
|
313 |
start_offset = models.IntegerField(null=True, blank=True) |
|
314 |
end_offset = models.IntegerField(null=True, blank=True) |
|
315 |
||
316 |
objects = CommentManager() |
|
317 |
||
| 327 | 318 |
def save(self, keep_dates=False, **kwargs): |
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
319 |
if not keep_dates: |
|
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
320 |
now = datetime.now() |
|
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
321 |
if not self.id: |
|
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
322 |
self.created = now |
|
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
323 |
self.modified = now |
| 327 | 324 |
super(PermanentModel, self).save(**kwargs) |
|
72
b0c777412d0a
keep comment dates (created / modified) when duplicating (fixes #12)
raph
parents:
38
diff
changeset
|
325 |
|
| 0 | 326 |
def __unicode__(self): |
| 262 | 327 |
return '<%d> %s [st_wrp:%s, st_ofs:%s, e_wrp:%s, e_ofs:%s]' % (self.id, self.title, self.start_wrapper , self.start_offset, self.end_wrapper, self.end_offset, ) |
| 0 | 328 |
|
329 |
def is_reply(self): |
|
330 |
return self.reply_to != None |
|
331 |
||
| 16 | 332 |
def is_thread_full_visible(self, own_user=None): |
333 |
""" |
|
334 |
own_user: comment belonging to this user are also visible |
|
335 |
""" |
|
336 |
if self.state == 'approved' or (own_user and self.user == own_user): |
|
337 |
if self.reply_to==None: |
|
338 |
return True |
|
339 |
else: |
|
340 |
return self.reply_to.is_thread_full_visible(own_user) |
|
341 |
return False |
|
342 |
||
| 103 | 343 |
def is_scope_removed(self): |
344 |
#when scope is "removed" we will have |
|
345 |
#self.start_wrapper == self.end_wrapper == self.start_offset == self.end_offset == -1 |
|
346 |
return (self.start_wrapper == -1) |
|
347 |
||
| 0 | 348 |
def top_comment(self): |
349 |
if self.reply_to == None : |
|
350 |
return self |
|
351 |
else : |
|
352 |
return self.reply_to.top_comment() |
|
353 |
||
354 |
def depth(self): |
|
355 |
if self.reply_to == None : |
|
356 |
return 0 |
|
357 |
else : |
|
358 |
return 1 + self.reply_to.depth() |
|
359 |
||
360 |
def delete(self): |
|
361 |
PermanentModel.delete(self) |
|
362 |
# delete replies |
|
363 |
[c.delete() for c in self.comment_set.all()] |
|
| 103 | 364 |
|
365 |
def remove_scope(self): |
|
366 |
self.start_wrapper = self.end_wrapper = self.start_offset = self.end_offset = -1 |
|
367 |
self.save() |
|
368 |
||
| 0 | 369 |
# http://docs.djangoproject.com/en/dev/topics/files/#topics-files |
370 |
||
371 |
# default conf values |
|
372 |
DEFAULT_CONF = { |
|
373 |
'workspace_name' : 'Workspace', |
|
374 |
'site_url' : settings.SITE_URL, |
|
375 |
'email_from' : settings.DEFAULT_FROM_EMAIL, |
|
376 |
} |
|
377 |
||
378 |
from cm.role_models import change_role_model |
|
379 |
||
380 |
class ConfigurationManager(models.Manager): |
|
381 |
def set_workspace_name(self, workspace_name): |
|
| 140 | 382 |
if workspace_name: |
383 |
#self.set_key('workspace_name', _(u"%(workspace_name)s's workspace") %{'workspace_name':workspace_name}) |
|
384 |
self.set_key('workspace_name', workspace_name) |
|
| 0 | 385 |
|
386 |
def get_key(self, key, default_value=None): |
|
387 |
try: |
|
388 |
return self.get(key=key).value |
|
389 |
except Configuration.DoesNotExist: |
|
390 |
return DEFAULT_CONF.get(key, default_value) |
|
| 160 | 391 |
|
392 |
def del_key(self, key): |
|
393 |
try: |
|
394 |
self.get(key=key).delete() |
|
395 |
except Configuration.DoesNotExist: |
|
396 |
return None |
|
| 0 | 397 |
|
398 |
def set_key(self, key, value): |
|
399 |
conf, created = self.get_or_create(key=key) |
|
400 |
if created or conf.value != value: |
|
401 |
conf.value = value |
|
402 |
conf.save() |
|
403 |
if key == 'workspace_role_model': |
|
404 |
change_role_model(value) |
|
405 |
||
406 |
def __getitem__(self, key): |
|
| 160 | 407 |
if not key.startswith('f_'): |
408 |
return self.get_key(key, None) |
|
409 |
else: |
|
410 |
return getattr(self,key)() |
|
411 |
||
412 |
def f_get_logo_url(self): |
|
413 |
key = self.get_key('workspace_logo_file_key', None) |
|
414 |
if key: |
|
415 |
attach = Attachment.objects.get(key=key) |
|
416 |
return attach.data.url |
|
417 |
else: |
|
418 |
return None |
|
| 0 | 419 |
|
420 |
import base64 |
|
421 |
||
422 |
class Configuration(models.Model): |
|
423 |
key = models.TextField(blank=False) # , unique=True cannot be added: creates error on mysql (?) |
|
424 |
raw_value = models.TextField(blank=False) |
|
425 |
||
426 |
def get_value(self): |
|
427 |
return pickle.loads(base64.b64decode(self.raw_value.encode('utf8'))) |
|
428 |
||
429 |
def set_value(self, value): |
|
430 |
self.raw_value = base64.b64encode(pickle.dumps(value, 0)).encode('utf8') |
|
431 |
||
432 |
value = property(get_value, set_value) |
|
433 |
||
434 |
objects = ConfigurationManager() |
|
435 |
||
436 |
def __unicode__(self): |
|
437 |
return '%s: %s' % (self.key, self.value) |
|
438 |
||
439 |
ApplicationConfiguration = Configuration.objects |
|
440 |
||
441 |
class AttachmentManager(KeyManager): |
|
442 |
def create_attachment(self, text_version, filename, data): |
|
443 |
attach = self.create(text_version=text_version) |
|
444 |
ff = ContentFile(data) |
|
445 |
attach.data.save(filename, ff) |
|
446 |
return attach |
|
447 |
||
448 |
class Attachment(KeyModel): |
|
449 |
data = models.FileField(upload_to="attachments/%Y/%m/%d/", max_length=1000) |
|
| 160 | 450 |
text_version = models.ForeignKey(TextVersion, null=True) |
| 0 | 451 |
|
452 |
objects = AttachmentManager() |
|
453 |
||
454 |
class NotificationManager(KeyManager): |
|
| 12 | 455 |
def create_notification(self, text, type, active, email_or_user): |
456 |
notification = self.create(text=text, type=type, active=active) |
|
457 |
notification.set_email_or_user(email_or_user) |
|
458 |
return notification |
|
| 0 | 459 |
|
| 12 | 460 |
def get_notifications(self, text, type, email_or_user): |
| 0 | 461 |
if isinstance(email_or_user,unicode): |
462 |
prev_notifications = Notification.objects.filter(text=text, type=type, email=email_or_user) |
|
463 |
else: |
|
464 |
prev_notifications = Notification.objects.filter(text=text, type=type, user=email_or_user) |
|
| 12 | 465 |
|
| 0 | 466 |
if prev_notifications: |
467 |
return prev_notifications[0] |
|
468 |
else: |
|
469 |
return None |
|
470 |
||
| 12 | 471 |
def set_notification(self, text, type, active, email_or_user): |
472 |
notification = self.get_notifications(text, type, email_or_user) |
|
473 |
if notification == None : |
|
474 |
self.create_notification(text, type, active, email_or_user) |
|
475 |
else : |
|
476 |
notification.active = active |
|
| 0 | 477 |
notification.save() |
478 |
||
479 |
class Notification(KeyModel, AuthorModel): |
|
480 |
text = models.ForeignKey(Text, null=True, blank=True) |
|
481 |
type = models.CharField(max_length=30, null=True, blank=True) |
|
482 |
active = models.BooleanField(default=True) # active = False means user desactivation |
|
483 |
||
484 |
objects = NotificationManager() |
|
485 |
||
486 |
def desactivate_notification_url(self): |
|
487 |
return reverse('desactivate-notification', args=[self.adminkey]) |
|
488 |
||
489 |
def desactivate(self): |
|
490 |
if self.type=='own': |
|
491 |
self.active = False |
|
492 |
self.save() |
|
493 |
else: |
|
494 |
self.delete() |
|
495 |
||
| 266 | 496 |
def __unicode__(self): |
497 |
return u"%s: %s %s %s %s" % (self.__class__.__name__, self.user, self.text, self.type, self.active ) |
|
498 |
||
| 0 | 499 |
# right management |
500 |
class UserRoleManager(models.Manager): |
|
501 |
def create_userroles_text(self, text): |
|
502 |
# make sure every user has a userrole on this text |
|
503 |
for user in User.objects.all(): |
|
504 |
userrole, _ = self.get_or_create(user=user, text=text) |
|
505 |
# anon user |
|
506 |
userrole, _ = self.get_or_create(user=None, text=text) |
|
507 |
# anon global user |
|
508 |
global_userrole, _ = self.get_or_create(user=None, text=None) |
|
509 |
||
510 |
class UserRole(models.Model): |
|
511 |
role = models.ForeignKey("Role", null=True, blank=True) |
|
512 |
||
513 |
# user == null => anyone |
|
514 |
user = models.ForeignKey(User, null=True, blank=True) |
|
515 |
||
516 |
# text == null => any text (workspace role) |
|
517 |
text = models.ForeignKey(Text, null=True, blank=True) |
|
518 |
||
519 |
objects = UserRoleManager() |
|
520 |
||
521 |
class Meta: |
|
522 |
unique_together = (('role', 'user', 'text',)) |
|
523 |
||
524 |
def __unicode__(self): |
|
525 |
if self.role: |
|
526 |
rolename = _(self.role.name) |
|
527 |
else: |
|
528 |
rolename = '' |
|
529 |
||
530 |
if self.user: |
|
531 |
return u"%s: %s %s %s" % (self.__class__.__name__, self.user.username, self.text, rolename) |
|
532 |
else: |
|
533 |
return u"%s: *ALL* %s %s" % (self.__class__.__name__, self.text, rolename) |
|
534 |
||
535 |
def __repr__(self): |
|
536 |
return self.__unicode__() |
|
537 |
||
538 |
from cm.models_base import generate_key |
|
539 |
from cm.utils.misc import update |
|
540 |
||
541 |
class Role(models.Model): |
|
542 |
""" |
|
543 |
'Static' application roles |
|
544 |
""" |
|
545 |
name = models.CharField(ugettext_lazy('name'), max_length=50, unique=True) |
|
546 |
description = models.TextField(ugettext_lazy('description')) |
|
547 |
#order = models.IntegerField(unique=True) |
|
548 |
permissions = models.ManyToManyField(Permission, related_name="roles") |
|
549 |
||
550 |
global_scope = models.BooleanField('global scope', default=False) # applies to global scope only |
|
551 |
anon = models.BooleanField('anonymous', default=False) # role possible for anonymous users? |
|
552 |
||
553 |
def __unicode__(self): |
|
554 |
return _(self.name) |
|
555 |
||
556 |
def __hash__(self): |
|
557 |
return self.id |
|
558 |
||
559 |
def name_i18n(self): |
|
560 |
return _(self.name) |
|
561 |
||
562 |
from django.utils.safestring import mark_safe |
|
563 |
||
564 |
class RegistrationManager(KeyManager): |
|
565 |
def activate_user(self, activation_key): |
|
566 |
""" |
|
567 |
Validates an activation key and activates the corresponding |
|
568 |
``User`` if valid. |
|
569 |
If the key is valid , returns the ``User`` as second arg |
|
570 |
First is boolean indicating if user has just been activated |
|
571 |
""" |
|
572 |
# Make sure the key we're trying conforms to the pattern of a |
|
573 |
# SHA1 hash; if it doesn't, no point trying to look it up in |
|
574 |
# the database. |
|
575 |
try: |
|
576 |
profile = self.get(admin_key=activation_key) |
|
577 |
except self.model.DoesNotExist: |
|
578 |
return False, False |
|
579 |
user = profile.user |
|
580 |
activated = False |
|
581 |
if not user.is_active: |
|
582 |
user.is_active = True |
|
583 |
user.save() |
|
584 |
activated = True |
|
585 |
return (activated, user) |
|
586 |
||
587 |
def _create_manager(self, email, username, password, first_name, last_name): |
|
588 |
if username and email and password and len(User.objects.filter(username=username)) == 0: |
|
589 |
user = User.objects.create(username=username, email=email, first_name=first_name, last_name=last_name, is_active=True) |
|
590 |
user.set_password(password) |
|
591 |
user.save() |
|
592 |
||
593 |
profile = UserProfile.objects.create(user=user) |
|
594 |
||
595 |
manager = Role.objects.get(name='Manager') |
|
596 |
UserRole.objects.create(text=None, user=user, role=manager) |
|
597 |
return user |
|
598 |
else: |
|
599 |
return None |
|
600 |
||
601 |
||
602 |
def create_inactive_user(self, email, send_invitation, **kwargs): |
|
|
305
df2c3202fd8e
do not lock table with backends other than postgresql (postgresql specific syntax) (fixes #52)
raph
parents:
289
diff
changeset
|
603 |
if 'postgresql' in settings.DATABASE_ENGINE: |
|
df2c3202fd8e
do not lock table with backends other than postgresql (postgresql specific syntax) (fixes #52)
raph
parents:
289
diff
changeset
|
604 |
#prevent concurrent access |
|
df2c3202fd8e
do not lock table with backends other than postgresql (postgresql specific syntax) (fixes #52)
raph
parents:
289
diff
changeset
|
605 |
cursor = connection.cursor() |
|
df2c3202fd8e
do not lock table with backends other than postgresql (postgresql specific syntax) (fixes #52)
raph
parents:
289
diff
changeset
|
606 |
sql = "LOCK TABLE auth_user IN EXCLUSIVE MODE" |
|
df2c3202fd8e
do not lock table with backends other than postgresql (postgresql specific syntax) (fixes #52)
raph
parents:
289
diff
changeset
|
607 |
cursor.execute(sql) |
| 0 | 608 |
|
609 |
try: |
|
610 |
user_with_email = User.objects.get(email__iexact=email) |
|
611 |
except User.DoesNotExist: |
|
612 |
user = User.objects.create(username=email, email=email) |
|
613 |
profile = UserProfile.objects.create(user=user) |
|
614 |
update(user, kwargs) |
|
615 |
update(profile, kwargs) |
|
616 |
||
617 |
user.is_active = False |
|
618 |
user.save() |
|
619 |
profile.save() |
|
620 |
||
621 |
note = kwargs.get('note', None) |
|
622 |
if send_invitation: |
|
623 |
profile.send_activation_email(note) |
|
624 |
return user |
|
625 |
else: |
|
626 |
return user_with_email |
|
627 |
||
628 |
||
629 |
from cm.utils.mail import send_mail |
|
630 |
||
631 |
class UserProfile(KeyModel): |
|
632 |
modified = models.DateTimeField(auto_now=True) |
|
633 |
created = models.DateTimeField(auto_now_add=True) |
|
634 |
||
635 |
user = models.ForeignKey(User, unique=True) |
|
636 |
||
637 |
allow_contact = models.BooleanField(ugettext_lazy(u'Allow contact'), default=True, help_text=ugettext_lazy(u"Allow email messages from other users")) |
|
638 |
preferred_language = models.CharField(ugettext_lazy(u'Preferred language'), max_length=2, default="en") |
|
639 |
is_temp = models.BooleanField(default=False) |
|
640 |
is_email_error = models.BooleanField(default=False) |
|
641 |
is_suspended = models.BooleanField(ugettext_lazy(u'Suspended access'), default=False) # used to disable access or to wait for approval when registering |
|
642 |
||
| 38 | 643 |
tags = TagField(ugettext_lazy("Tags"), max_length=1000) |
644 |
||
| 0 | 645 |
objects = RegistrationManager() |
646 |
||
647 |
class Meta: |
|
648 |
permissions = ( |
|
649 |
("can_create_user", "Can create user"), |
|
650 |
("can_delete_user", "Can delete user"), |
|
651 |
) |
|
652 |
||
653 |
def __unicode__(self): |
|
654 |
return unicode(self.user) |
|
655 |
||
656 |
def global_userrole(self): |
|
657 |
try: |
|
658 |
return UserRole.objects.get(user=self.user, text=None) |
|
659 |
except UserRole.DoesNotExist: |
|
660 |
return None |
|
661 |
||
662 |
def admin_print(self): |
|
663 |
if self.is_suspended: |
|
664 |
if self.user.is_active: |
|
665 |
return mark_safe('%s (%s)' % (self.user.username, _(u'suspended'),)) |
|
666 |
else: |
|
667 |
return mark_safe('%s (%s)' % (self.user.username, _(u'waiting approval'),)) |
|
668 |
else: |
|
669 |
if self.user.is_active: |
|
670 |
return mark_safe('%s' % self.user.username) |
|
671 |
else: |
|
672 |
email_username = self.user.email.split('@')[0] |
|
673 |
return mark_safe('%s (%s)' % (self.user.username, _(u'pending'),)) |
|
674 |
||
675 |
def simple_print(self): |
|
676 |
if self.user.is_active: |
|
677 |
return self.user.username |
|
678 |
else: |
|
679 |
return self.user.email |
|
680 |
||
681 |
def send_activation_email(self, note=None): |
|
682 |
self._send_act_invit_email(note=note, template='email/activation_email.txt') |
|
683 |
||
684 |
def send_invitation_email(self, note=None): |
|
685 |
self._send_act_invit_email(note=note, template='email/invitation_email.txt') |
|
686 |
||
687 |
def _send_act_invit_email(self, template, note=None): |
|
688 |
subject = _(u'Invitation') |
|
689 |
||
690 |
activate_url = reverse('user-activate', args=[self.adminkey]) |
|
691 |
message = render_to_string(template, |
|
692 |
{ |
|
693 |
'activate_url' : activate_url, |
|
694 |
'note' : note, |
|
695 |
'CONF': ApplicationConfiguration |
|
696 |
}) |
|
697 |
||
698 |
send_mail(subject, message, ApplicationConfiguration['email_from'], [self.user.email]) |
|
699 |
||
700 |
from django.db.models import signals |
|
701 |
||
702 |
#def create_profile(sender, **kwargs): |
|
703 |
# created = kwargs['created'] |
|
704 |
# if created: |
|
705 |
# user = kwargs['instance'] |
|
706 |
# UserProfile.objects.create(user = user) |
|
707 |
||
708 |
def delete_profile(sender, **kwargs): |
|
709 |
user_profile = kwargs['instance'] |
|
710 |
user = user_profile.user |
|
711 |
user.delete() |
|
712 |
||
713 |
#signals.post_save.connect(create_profile, sender=User) |
|
714 |
signals.post_delete.connect(delete_profile, sender=UserProfile) |
|
715 |
||
716 |
class ActivityManager(models.Manager): |
|
717 |
pass |
|
718 |
||
719 |
class Activity(models.Model): |
|
720 |
created = models.DateTimeField(auto_now_add=True) |
|
721 |
originator_user = models.ForeignKey(User, related_name='originator_activity', null=True, blank=True, default=None) |
|
722 |
text = models.ForeignKey(Text, null=True, blank=True, default=None) |
|
723 |
text_version = models.ForeignKey(TextVersion, null=True, blank=True, default=None) |
|
724 |
comment = models.ForeignKey(Comment, null=True, blank=True, default=None) |
|
725 |
user = models.ForeignKey(User, null=True, blank=True, default=None) |
|
726 |
type = models.CharField(max_length=30) |
|
727 |
ip = models.IPAddressField(null=True, blank=True, default=None) |
|
728 |
||
729 |
objects = ActivityManager() |
|
730 |
||
731 |
# viewable activities (i.e. now 'text-view') |
|
732 |
VIEWABLE_ACTIVITIES = { |
|
733 |
'view_comments' : ['comment_created', 'comment_removed'], |
|
734 |
'view_users' : ['user_created', 'user_activated', 'user_refused', 'user_enabled', 'user_approved', 'user_suspended'], |
|
735 |
'view_texts' : ['text_created', 'text_removed', 'text_edited', 'text_edited_new_version'], |
|
736 |
} |
|
737 |
ACTIVITIES_TYPES = reduce(list.__add__, VIEWABLE_ACTIVITIES.values()) |
|
738 |
||
739 |
IMGS = { |
|
740 |
'text_created' : u'page_add_small.png', |
|
741 |
'text_removed' : u'page_delete_small.png', |
|
742 |
'text_edited' : u'page_save_small.png', |
|
743 |
'text_edited_new_version' : u'page_save_small.png', |
|
744 |
'user_created' : u'user_add_small.png', |
|
745 |
'user_enabled' : u'user_add_small.png', |
|
746 |
'user_refused': u'user_delete_small.png', |
|
747 |
'user_suspended': u'user_delete_small.png', |
|
748 |
'user_approved': u'user_add_small.png', |
|
749 |
'user_activated' : u'user_go.png', |
|
750 |
'comment_created' : u'note_add_small.png', |
|
751 |
'comment_removed' : u'note_delete_small.png', |
|
752 |
} |
|
753 |
||
754 |
#type/msg |
|
755 |
MSGS = { |
|
| 227 | 756 |
'text_edited' : ugettext_lazy(u'Text %(link_to_text)s edited by %(creator)s'), |
757 |
'text_edited_new_version' : ugettext_lazy(u'Text %(link_to_text)s edited (new version created) by %(creator)s'), |
|
758 |
'text_created' : ugettext_lazy(u'Text %(link_to_text)s added by %(creator)s'), |
|
| 5 | 759 |
'text_removed' : ugettext_lazy(u'Text %(link_to_text)s removed'), |
| 227 | 760 |
'comment_created' : ugettext_lazy(u'Comment %(link_to_comment)s added on text %(link_to_text)s by %(creator)s'), |
| 5 | 761 |
'comment_removed' : ugettext_lazy(u'Comment %(link_to_comment)s removed from text %(link_to_text)s'), |
762 |
'user_created' : ugettext_lazy(u'User %(username)s added'), |
|
763 |
'user_enabled' : ugettext_lazy(u'User %(username)s access to workspace enabled'), |
|
764 |
'user_refused' : ugettext_lazy(u'User %(username)s access to workspace refused'), |
|
765 |
'user_suspended' : ugettext_lazy(u'User %(username)s access to workspace suspended'), |
|
766 |
'user_activated' : ugettext_lazy(u'User %(username)s access to workspace activated'), |
|
767 |
'user_approved' : ugettext_lazy(u'User %(username)s has activated his account'), |
|
| 0 | 768 |
} |
769 |
||
770 |
def is_same_user(self, other_activity): |
|
771 |
if (self.originator_user != None or other_activity.originator_user != None) and self.originator_user != other_activity.originator_user: |
|
772 |
return False |
|
773 |
else: |
|
774 |
return self.ip != None and self.ip == other_activity.ip |
|
775 |
||
776 |
def linkable_text_title(self, html=True, link=True): |
|
777 |
# html: whether or not output sould be html |
|
778 |
format_args = {'link':absolute_reverse('text-view', args=[self.text.key]), 'title':self.text.title} |
|
779 |
if html and not self.text.deleted : |
|
780 |
return mark_safe(u'<a href="%(link)s">%(title)s</a>' % format_args) |
|
781 |
else: |
|
782 |
if link and not self.text.deleted: |
|
783 |
return u'%(title)s (%(link)s)' % format_args |
|
784 |
else: |
|
785 |
return self.text.title ; |
|
786 |
||
787 |
def linkable_comment_title(self, html=True, link=True): |
|
788 |
if self.comment: |
|
| 144 | 789 |
format_args = {'link':absolute_reverse('text-view-show-comment', args=[self.text.key, self.comment.id_key]), 'title':self.comment.title} |
| 0 | 790 |
if html and not self.comment.deleted and not self.text.deleted: |
791 |
return mark_safe(u'<a href="%(link)s">%(title)s</a>' % format_args) |
|
792 |
else : |
|
793 |
if link and not self.comment.deleted and not self.text.deleted: |
|
794 |
return u'%(title)s (%(link)s)' % format_args |
|
795 |
else: |
|
796 |
return self.comment.title ; |
|
797 |
else: |
|
798 |
return u'' |
|
799 |
||
800 |
def __unicode__(self): |
|
801 |
return u"%s %s %s %s %s" % (self.type, self.originator_user, self.text, self.comment, self.user) |
|
802 |
||
803 |
def img_name(self): |
|
804 |
return self.IMGS.get(self.type) |
|
805 |
||
806 |
def printable_data_nohtml_link(self): |
|
807 |
return self.printable_data(html=False, link=True) |
|
808 |
||
809 |
def printable_data(self, html=True, link=True): |
|
810 |
msg = self.MSGS.get(self.type, None) |
|
811 |
if msg: |
|
812 |
return mark_safe(msg % { |
|
813 |
'link_to_text' : self.linkable_text_title(html=html, link=link) if self.text else None, |
|
814 |
'link_to_comment' : self.linkable_comment_title(html=html, link=link) if self.comment else None, |
|
815 |
'username' : self.user.username if self.user else None, |
|
| 227 | 816 |
'creator' : self.originator_user.username if self.originator_user else ugettext_lazy(u'anonymous'), |
| 0 | 817 |
}) |
818 |
return '' |
|
819 |
||
820 |
def printable_metadata(self, html=True): |
|
821 |
ret = [] |
|
822 |
if self.type == 'user_activated': |
|
823 |
ret.append(_(u'by "%(username)s"') % {'username' : self.originator_user.username}) |
|
824 |
ret.append(' ') |
|
825 |
ret.append(_(u"%(time_since)s ago") % {'time_since':timesince(self.created)}) |
|
826 |
return ''.join(ret) |
|
827 |
||
828 |
def printable_metadata_absolute(self, html=True): |
|
829 |
ret = [] |
|
830 |
if self.type == 'user_activated': |
|
831 |
ret.append(_(u'by "%(username)s"') % {'username' : self.originator_user.username}) |
|
832 |
ret.append(u' ') |
|
833 |
ret.append(datetime_to_user_str(self.created)) |
|
834 |
return u''.join(ret) |
|
835 |
||
836 |
import cm.denorm_engine |
|
837 |
import cm.admin |
|
838 |
import cm.main |
|
839 |
import cm.activity |
|
840 |
import cm.notifications |
|
841 |
||
842 |
# we fill username with email so we need a bigger value |
|
843 |
User._meta.get_field('username').max_length = 75 |