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