git.ozlabs.org Git - patchwork/blob - apps/patchwork/models.py

   1 # Patchwork - automated patch tracking system
   2 # Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
   3 #
   4 # This file is part of the Patchwork package.
   5 #
   6 # Patchwork is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # Patchwork is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with Patchwork; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20 from django.db import models
  21 from django.contrib.auth.models import User
  22 from django.core.urlresolvers import reverse
  23 from django.contrib.sites.models import Site
  24 from django.conf import settings
  25 from patchwork.parser import hash_patch
  26
  27 import re
  28 import datetime, time
  29 import string
  30 import random
  31
  32 try:
  33     from email.mime.nonmultipart import MIMENonMultipart
  34     from email.encoders import encode_7or8bit
  35     import email.utils
  36 except ImportError:
  37     # Python 2.4 compatibility
  38     from email.MIMENonMultipart import MIMENonMultipart
  39     from email.Encoders import encode_7or8bit
  40     import email.Utils
  41     email.utils = email.Utils
  42
  43 class Person(models.Model):
  44     email = models.CharField(max_length=255, unique = True)
  45     name = models.CharField(max_length=255, null = True)
  46     user = models.ForeignKey(User, null = True)
  47
  48     def __str__(self):
  49         if self.name:
  50             return '%s <%s>' % (self.name, self.email)
  51         else:
  52             return self.email
  53
  54     def link_to_user(self, user):
  55         self.name = user.get_profile().name()
  56         self.user = user
  57
  58     class Meta:
  59         verbose_name_plural = 'People'
  60
  61 class Project(models.Model):
  62     linkname = models.CharField(max_length=255, unique=True)
  63     name = models.CharField(max_length=255, unique=True)
  64     listid = models.CharField(max_length=255, unique=True)
  65     listemail = models.CharField(max_length=200)
  66
  67     def __str__(self):
  68         return self.name
  69
  70 class UserProfile(models.Model):
  71     user = models.ForeignKey(User, unique = True)
  72     primary_project = models.ForeignKey(Project, null = True)
  73     maintainer_projects = models.ManyToManyField(Project,
  74             related_name = 'maintainer_project')
  75     send_email = models.BooleanField(default = False,
  76             help_text = 'Selecting this option allows patchwork to send ' +
  77                 'email on your behalf')
  78     patches_per_page = models.PositiveIntegerField(default = 100,
  79             null = False, blank = False,
  80             help_text = 'Number of patches to display per page')
  81
  82     def name(self):
  83         if self.user.first_name or self.user.last_name:
  84             names = filter(bool, [self.user.first_name, self.user.last_name])
  85             return ' '.join(names)
  86         return self.user.username
  87
  88     def contributor_projects(self):
  89         submitters = Person.objects.filter(user = self.user)
  90         return Project.objects \
  91             .filter(id__in = \
  92                     Patch.objects.filter(
  93                         submitter__in = submitters) \
  94                     .values('project_id').query)
  95
  96
  97     def sync_person(self):
  98         pass
  99
 100     def n_todo_patches(self):
 101         return self.todo_patches().count()
 102
 103     def todo_patches(self, project = None):
 104
 105         # filter on project, if necessary
 106         if project:
 107             qs = Patch.objects.filter(project = project)
 108         else:
 109             qs = Patch.objects
 110
 111         qs = qs.filter(archived = False) \
 112              .filter(delegate = self.user) \
 113              .filter(state__in = \
 114                      State.objects.filter(action_required = True) \
 115                          .values('pk').query)
 116         return qs
 117
 118     def save(self):
 119         super(UserProfile, self).save()
 120         people = Person.objects.filter(email = self.user.email)
 121         if not people:
 122             person = Person(email = self.user.email,
 123                     name = self.name(), user = self.user)
 124             person.save()
 125         else:
 126             for person in people:
 127                  person.link_to_user(self.user)
 128                  person.save()
 129
 130     def __str__(self):
 131         return self.name()
 132
 133 class State(models.Model):
 134     name = models.CharField(max_length = 100)
 135     ordering = models.IntegerField(unique = True)
 136     action_required = models.BooleanField(default = True)
 137
 138     def __str__(self):
 139         return self.name
 140
 141     class Meta:
 142         ordering = ['ordering']
 143
 144 class HashField(models.CharField):
 145     __metaclass__ = models.SubfieldBase
 146
 147     def __init__(self, algorithm = 'sha1', *args, **kwargs):
 148         self.algorithm = algorithm
 149         try:
 150             import hashlib
 151             def _construct(string = ''):
 152                 return hashlib.new(self.algorithm, string)
 153             self.construct = _construct
 154             self.n_bytes = len(hashlib.new(self.algorithm).hexdigest())
 155         except ImportError:
 156             modules = { 'sha1': 'sha', 'md5': 'md5'}
 157
 158             if algorithm not in modules.keys():
 159                 raise NameError("Unknown algorithm '%s'" % algorithm)
 160
 161             self.construct = __import__(modules[algorithm]).new
 162
 163         self.n_bytes = len(self.construct().hexdigest())
 164
 165         kwargs['max_length'] = self.n_bytes
 166         super(HashField, self).__init__(*args, **kwargs)
 167
 168     def db_type(self):
 169         return 'char(%d)' % self.n_bytes
 170
 171 class PatchMbox(MIMENonMultipart):
 172     patch_charset = 'utf-8'
 173     def __init__(self, _text):
 174         MIMENonMultipart.__init__(self, 'text', 'plain',
 175                         **{'charset': self.patch_charset})
 176         self.set_payload(_text.encode(self.patch_charset))
 177         encode_7or8bit(self)
 178
 179 class Patch(models.Model):
 180     project = models.ForeignKey(Project)
 181     msgid = models.CharField(max_length=255)
 182     name = models.CharField(max_length=255)
 183     date = models.DateTimeField(default=datetime.datetime.now)
 184     submitter = models.ForeignKey(Person)
 185     delegate = models.ForeignKey(User, blank = True, null = True)
 186     state = models.ForeignKey(State)
 187     archived = models.BooleanField(default = False)
 188     headers = models.TextField(blank = True)
 189     content = models.TextField()
 190     commit_ref = models.CharField(max_length=255, null = True, blank = True)
 191     hash = HashField(null = True, db_index = True)
 192
 193     def __str__(self):
 194         return self.name
 195
 196     def comments(self):
 197         return Comment.objects.filter(patch = self)
 198
 199     def save(self):
 200         try:
 201             s = self.state
 202         except:
 203             self.state = State.objects.get(ordering =  0)
 204
 205         if self.hash is None:
 206             self.hash = hash_patch(self.content).hexdigest()
 207
 208         super(Patch, self).save()
 209
 210     def is_editable(self, user):
 211         if not user.is_authenticated():
 212             return False
 213
 214         if self.submitter.user == user or self.delegate == user:
 215             return True
 216
 217         profile = user.get_profile()
 218         return self.project in user.get_profile().maintainer_projects.all()
 219
 220     def form(self):
 221         f = PatchForm(instance = self, prefix = self.id)
 222         return f
 223
 224     def filename(self):
 225         fname_re = re.compile('[^-_A-Za-z0-9\.]+')
 226         str = fname_re.sub('-', self.name)
 227         return str.strip('-') + '.patch'
 228
 229     def mbox(self):
 230         comment = None
 231         try:
 232             comment = Comment.objects.get(patch = self, msgid = self.msgid)
 233         except Exception:
 234             pass
 235
 236         body = ''
 237         if comment:
 238             body = comment.content.strip() + "\n"
 239
 240         responses = False
 241         for comment in Comment.objects.filter(patch = self) \
 242                 .exclude(msgid = self.msgid):
 243             body += comment.patch_responses()
 244
 245         if body:
 246             body += '\n'
 247
 248         body += self.content
 249
 250         mail = PatchMbox(body)
 251         mail['Subject'] = self.name
 252         mail['Date'] = email.utils.formatdate(
 253                         time.mktime(self.date.utctimetuple()))
 254         mail['From'] = unicode(self.submitter)
 255         mail['X-Patchwork-Id'] = str(self.id)
 256         mail.set_unixfrom('From patchwork ' + self.date.ctime())
 257
 258         return mail
 259
 260
 261     @models.permalink
 262     def get_absolute_url(self):
 263         return ('patchwork.views.patch.patch', (), {'patch_id': self.id})
 264
 265     class Meta:
 266         verbose_name_plural = 'Patches'
 267         ordering = ['date']
 268         unique_together = [('msgid', 'project')]
 269
 270 class Comment(models.Model):
 271     patch = models.ForeignKey(Patch)
 272     msgid = models.CharField(max_length=255)
 273     submitter = models.ForeignKey(Person)
 274     date = models.DateTimeField(default = datetime.datetime.now)
 275     headers = models.TextField(blank = True)
 276     content = models.TextField()
 277
 278     response_re = re.compile('^(Acked|Signed-off|Nacked)-by: .*$', re.M)
 279
 280     def patch_responses(self):
 281         return ''.join([ match.group(0) + '\n' for match in \
 282                                 self.response_re.finditer(self.content)])
 283
 284     class Meta:
 285         ordering = ['date']
 286         unique_together = [('msgid', 'patch')]
 287
 288 class Bundle(models.Model):
 289     owner = models.ForeignKey(User)
 290     project = models.ForeignKey(Project)
 291     name = models.CharField(max_length = 50, null = False, blank = False)
 292     patches = models.ManyToManyField(Patch, through = 'BundlePatch')
 293     public = models.BooleanField(default = False)
 294
 295     def n_patches(self):
 296         return self.patches.all().count()
 297
 298     def append_patch(self, patch):
 299         # todo: use the aggregate queries in django 1.1
 300         orders = BundlePatch.objects.filter(bundle = self).values('order')
 301         max_order = max([ v for (k, v) in orders])
 302
 303         bp = BundlePatch.objects.create(bundle = self, patch = patch, order = max_order + 1)
 304         bp.save()
 305
 306     class Meta:
 307         unique_together = [('owner', 'name')]
 308
 309     def public_url(self):
 310         if not self.public:
 311             return None
 312         site = Site.objects.get_current()
 313         return 'http://%s%s' % (site.domain,
 314                 reverse('patchwork.views.bundle.public',
 315                         kwargs = {
 316                                 'username': self.owner.username,
 317                                 'bundlename': self.name
 318                         }))
 319
 320     def mbox(self):
 321         return '\n'.join([p.mbox().as_string(True) \
 322                         for p in self.patches.all()])
 323
 324 class BundlePatch(models.Model):
 325     patch = models.ForeignKey(Patch)
 326     bundle = models.ForeignKey(Bundle)
 327     order = models.IntegerField()
 328
 329     class Meta:
 330         unique_together = [('bundle', 'patch'), ('bundle', 'order')]
 331
 332 class UserPersonConfirmation(models.Model):
 333     user = models.ForeignKey(User)
 334     email = models.CharField(max_length = 200)
 335     key = HashField()
 336     date = models.DateTimeField(default=datetime.datetime.now)
 337     active = models.BooleanField(default = True)
 338
 339     def confirm(self):
 340         if not self.active:
 341             return
 342         person = None
 343         try:
 344             person = Person.objects.get(email__iexact = self.email)
 345         except Exception:
 346             pass
 347         if not person:
 348             person = Person(email = self.email)
 349
 350         person.link_to_user(self.user)
 351         person.save()
 352         self.active = False
 353         self.save()
 354
 355     def save(self):
 356         max = 1 << 32
 357         if self.key == '':
 358             str = '%s%s%d' % (self.user, self.email, random.randint(0, max))
 359             self.key = self._meta.get_field('key').construct(str).hexdigest()
 360         super(UserPersonConfirmation, self).save()
 361
 362