X-Git-Url: https://git.ozlabs.org/?a=blobdiff_plain;f=apps%2Fpatchwork%2Fmodels.py;fp=apps%2Fpatchwork%2Fmodels.py;h=0000000000000000000000000000000000000000;hb=ad2762cf775a8dde508de47164d6429f3fd724f1;hp=54b86566d7bd9a07b0dabb18c978b3d9b44f01f1;hpb=f09e982f58384946111d4157fd2b7c2b31b78612;p=patchwork diff --git a/apps/patchwork/models.py b/apps/patchwork/models.py deleted file mode 100644 index 54b8656..0000000 --- a/apps/patchwork/models.py +++ /dev/null @@ -1,386 +0,0 @@ -# Patchwork - automated patch tracking system -# Copyright (C) 2008 Jeremy Kerr -# -# This file is part of the Patchwork package. -# -# Patchwork is free software; you can redistribute it and/or modify -# it under the terms of the GNU General Public License as published by -# the Free Software Foundation; either version 2 of the License, or -# (at your option) any later version. -# -# Patchwork is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with Patchwork; if not, write to the Free Software -# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - -from django.db import models -from django.contrib.auth.models import User -from django.core.urlresolvers import reverse -from django.contrib.sites.models import Site -from django.conf import settings -from patchwork.parser import hash_patch - -import re -import datetime, time -import random - -class Person(models.Model): - email = models.CharField(max_length=255, unique = True) - name = models.CharField(max_length=255, null = True, blank = True) - user = models.ForeignKey(User, null = True, blank = True, - on_delete = models.SET_NULL) - - def __unicode__(self): - if self.name: - return u'%s <%s>' % (self.name, self.email) - else: - return self.email - - def link_to_user(self, user): - self.name = user.profile.name() - self.user = user - - class Meta: - verbose_name_plural = 'People' - -class Project(models.Model): - linkname = models.CharField(max_length=255, unique=True) - name = models.CharField(max_length=255, unique=True) - listid = models.CharField(max_length=255, unique=True) - listemail = models.CharField(max_length=200) - web_url = models.CharField(max_length=2000, blank=True) - scm_url = models.CharField(max_length=2000, blank=True) - webscm_url = models.CharField(max_length=2000, blank=True) - send_notifications = models.BooleanField(default=False) - - def __unicode__(self): - return self.name - - def is_editable(self, user): - if not user.is_authenticated(): - return False - return self in user.profile.maintainer_projects.all() - - class Meta: - ordering = ['linkname'] - - -class UserProfile(models.Model): - user = models.OneToOneField(User, unique = True, related_name='profile') - primary_project = models.ForeignKey(Project, null = True, blank = True) - maintainer_projects = models.ManyToManyField(Project, - related_name = 'maintainer_project') - send_email = models.BooleanField(default = False, - help_text = 'Selecting this option allows patchwork to send ' + - 'email on your behalf') - patches_per_page = models.PositiveIntegerField(default = 100, - null = False, blank = False, - help_text = 'Number of patches to display per page') - - def name(self): - if self.user.first_name or self.user.last_name: - names = filter(bool, [self.user.first_name, self.user.last_name]) - return u' '.join(names) - return self.user.username - - def contributor_projects(self): - submitters = Person.objects.filter(user = self.user) - return Project.objects.filter(id__in = - Patch.objects.filter( - submitter__in = submitters) - .values('project_id').query) - - def sync_person(self): - pass - - def n_todo_patches(self): - return self.todo_patches().count() - - def todo_patches(self, project = None): - - # filter on project, if necessary - if project: - qs = Patch.objects.filter(project = project) - else: - qs = Patch.objects - - qs = qs.filter(archived = False) \ - .filter(delegate = self.user) \ - .filter(state__in = - State.objects.filter(action_required = True) - .values('pk').query) - return qs - - def __unicode__(self): - return self.name() - -def _user_saved_callback(sender, created, instance, **kwargs): - try: - profile = instance.profile - except UserProfile.DoesNotExist: - profile = UserProfile(user = instance) - profile.save() - -models.signals.post_save.connect(_user_saved_callback, sender = User) - -class State(models.Model): - name = models.CharField(max_length = 100) - ordering = models.IntegerField(unique = True) - action_required = models.BooleanField(default = True) - - def __unicode__(self): - return self.name - - class Meta: - ordering = ['ordering'] - -class HashField(models.CharField): - __metaclass__ = models.SubfieldBase - - def __init__(self, algorithm = 'sha1', *args, **kwargs): - self.algorithm = algorithm - try: - import hashlib - def _construct(string = ''): - return hashlib.new(self.algorithm, string) - self.construct = _construct - self.n_bytes = len(hashlib.new(self.algorithm).hexdigest()) - except ImportError: - modules = { 'sha1': 'sha', 'md5': 'md5'} - - if algorithm not in modules.keys(): - raise NameError("Unknown algorithm '%s'" % algorithm) - - self.construct = __import__(modules[algorithm]).new - - self.n_bytes = len(self.construct().hexdigest()) - - kwargs['max_length'] = self.n_bytes - super(HashField, self).__init__(*args, **kwargs) - - def db_type(self, connection=None): - return 'char(%d)' % self.n_bytes - -def get_default_initial_patch_state(): - return State.objects.get(ordering=0) - -class Patch(models.Model): - project = models.ForeignKey(Project) - msgid = models.CharField(max_length=255) - name = models.CharField(max_length=255) - date = models.DateTimeField(default=datetime.datetime.now) - submitter = models.ForeignKey(Person) - delegate = models.ForeignKey(User, blank = True, null = True) - state = models.ForeignKey(State, default=get_default_initial_patch_state) - archived = models.BooleanField(default = False) - headers = models.TextField(blank = True) - content = models.TextField(null = True, blank = True) - pull_url = models.CharField(max_length=255, null = True, blank = True) - commit_ref = models.CharField(max_length=255, null = True, blank = True) - hash = HashField(null = True, blank = True) - - def __unicode__(self): - return self.name - - def comments(self): - return Comment.objects.filter(patch = self) - - def save(self): - try: - s = self.state - except: - self.state = State.objects.get(ordering = 0) - - if self.hash is None and self.content is not None: - self.hash = hash_patch(self.content).hexdigest() - - super(Patch, self).save() - - def is_editable(self, user): - if not user.is_authenticated(): - return False - - if self.submitter.user == user or self.delegate == user: - return True - - return self.project.is_editable(user) - - def filename(self): - fname_re = re.compile('[^-_A-Za-z0-9\.]+') - str = fname_re.sub('-', self.name) - return str.strip('-') + '.patch' - - @models.permalink - def get_absolute_url(self): - return ('patchwork.views.patch.patch', (), {'patch_id': self.id}) - - class Meta: - verbose_name_plural = 'Patches' - ordering = ['date'] - unique_together = [('msgid', 'project')] - -class Comment(models.Model): - patch = models.ForeignKey(Patch) - msgid = models.CharField(max_length=255) - submitter = models.ForeignKey(Person) - date = models.DateTimeField(default = datetime.datetime.now) - headers = models.TextField(blank = True) - content = models.TextField() - - response_re = re.compile( \ - '^(Tested|Reviewed|Acked|Signed-off|Nacked|Reported)-by: .*$', - re.M | re.I) - - def patch_responses(self): - return ''.join([ match.group(0) + '\n' for match in - self.response_re.finditer(self.content)]) - - class Meta: - ordering = ['date'] - unique_together = [('msgid', 'patch')] - -class Bundle(models.Model): - owner = models.ForeignKey(User) - project = models.ForeignKey(Project) - name = models.CharField(max_length = 50, null = False, blank = False) - patches = models.ManyToManyField(Patch, through = 'BundlePatch') - public = models.BooleanField(default = False) - - def n_patches(self): - return self.patches.all().count() - - def ordered_patches(self): - return self.patches.order_by('bundlepatch__order') - - def append_patch(self, patch): - # todo: use the aggregate queries in django 1.1 - orders = BundlePatch.objects.filter(bundle = self).order_by('-order') \ - .values('order') - - if len(orders) > 0: - max_order = orders[0]['order'] - else: - max_order = 0 - - # see if the patch is already in this bundle - if BundlePatch.objects.filter(bundle = self, patch = patch).count(): - raise Exception("patch is already in bundle") - - bp = BundlePatch.objects.create(bundle = self, patch = patch, - order = max_order + 1) - bp.save() - - class Meta: - unique_together = [('owner', 'name')] - - def public_url(self): - if not self.public: - return None - site = Site.objects.get_current() - return 'http://%s%s' % (site.domain, - reverse('patchwork.views.bundle.bundle', - kwargs = { - 'username': self.owner.username, - 'bundlename': self.name - })) - - @models.permalink - def get_absolute_url(self): - return ('patchwork.views.bundle.bundle', (), { - 'username': self.owner.username, - 'bundlename': self.name, - }) - -class BundlePatch(models.Model): - patch = models.ForeignKey(Patch) - bundle = models.ForeignKey(Bundle) - order = models.IntegerField() - - class Meta: - unique_together = [('bundle', 'patch')] - ordering = ['order'] - -class EmailConfirmation(models.Model): - validity = datetime.timedelta(days = settings.CONFIRMATION_VALIDITY_DAYS) - type = models.CharField(max_length = 20, choices = [ - ('userperson', 'User-Person association'), - ('registration', 'Registration'), - ('optout', 'Email opt-out'), - ]) - email = models.CharField(max_length = 200) - user = models.ForeignKey(User, null = True) - key = HashField() - date = models.DateTimeField(default = datetime.datetime.now) - active = models.BooleanField(default = True) - - def deactivate(self): - self.active = False - self.save() - - def is_valid(self): - return self.date + self.validity > datetime.datetime.now() - - def save(self): - max = 1 << 32 - if self.key == '': - str = '%s%s%d' % (self.user, self.email, random.randint(0, max)) - self.key = self._meta.get_field('key').construct(str).hexdigest() - super(EmailConfirmation, self).save() - -class EmailOptout(models.Model): - email = models.CharField(max_length = 200, primary_key = True) - - def __unicode__(self): - return self.email - - @classmethod - def is_optout(cls, email): - email = email.lower().strip() - return cls.objects.filter(email = email).count() > 0 - -class PatchChangeNotification(models.Model): - patch = models.ForeignKey(Patch, primary_key = True) - last_modified = models.DateTimeField(default = datetime.datetime.now) - orig_state = models.ForeignKey(State) - -def _patch_change_callback(sender, instance, **kwargs): - # we only want notification of modified patches - if instance.pk is None: - return - - if instance.project is None or not instance.project.send_notifications: - return - - try: - orig_patch = Patch.objects.get(pk = instance.pk) - except Patch.DoesNotExist: - return - - # If there's no interesting changes, abort without creating the - # notification - if orig_patch.state == instance.state: - return - - notification = None - try: - notification = PatchChangeNotification.objects.get(patch = instance) - except PatchChangeNotification.DoesNotExist: - pass - - if notification is None: - notification = PatchChangeNotification(patch = instance, - orig_state = orig_patch.state) - - elif notification.orig_state == instance.state: - # If we're back at the original state, there is no need to notify - notification.delete() - return - - notification.last_modified = datetime.datetime.now() - notification.save() - -models.signals.pre_save.connect(_patch_change_callback, sender = Patch)