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