458 lines
13 KiB
Python
458 lines
13 KiB
Python
from django.db import models
|
|
from django.db.models import Q
|
|
from django.contrib import auth
|
|
from django.contrib.auth.models import Group
|
|
from django.core.validators import RegexValidator
|
|
from django.core.exceptions import ValidationError
|
|
from django.utils.safestring import mark_safe
|
|
from django.utils.html import escape
|
|
from django.core.urlresolvers import reverse
|
|
from django.contrib.sites.models import Site
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.utils.encoding import python_2_unicode_compatible
|
|
from django import VERSION
|
|
|
|
from colorful.fields import RGBColorField
|
|
|
|
import json
|
|
|
|
from issue.templatetags.issue_tags import same_label, labeled
|
|
|
|
|
|
class User(auth.models.User):
|
|
|
|
class Meta:
|
|
proxy = True
|
|
|
|
@property
|
|
def teams(self):
|
|
query = Q(groups__in=self.groups.all()) | Q(users=self)
|
|
return Team.objects.filter(query)
|
|
|
|
|
|
class Project(models.Model):
|
|
|
|
url_name_validator = RegexValidator(regex='^[a-z0-9_-]+$',
|
|
message="Please enter only lowercase characters, number, "
|
|
"underscores or hyphens.")
|
|
|
|
name = models.CharField(primary_key=True, blank=False, max_length=32,
|
|
verbose_name="Short name (used in URL, definitive)",
|
|
validators=[url_name_validator])
|
|
|
|
display_name = models.CharField(max_length=32, unique=True,
|
|
verbose_name="Project name")
|
|
|
|
description = models.TextField(blank=True, default="",
|
|
verbose_name="Description")
|
|
|
|
public = models.BooleanField(default=True,
|
|
verbose_name="Do unregistered users have read access "
|
|
"to this project?")
|
|
|
|
def grant_user(self, user):
|
|
perm = ProjectPermission(project=self,
|
|
manage_project_permission=True,
|
|
grantee_type=PermissionModel.GRANTEE_USER,
|
|
grantee_name=user.username)
|
|
perm.save()
|
|
|
|
def __str__(self):
|
|
|
|
return self.display_name
|
|
|
|
|
|
class Label(models.Model):
|
|
|
|
project = models.ForeignKey(Project, related_name='labels')
|
|
|
|
name = models.CharField(max_length=32)
|
|
|
|
deleted = models.BooleanField(default=False)
|
|
|
|
color = RGBColorField(default='#000000',
|
|
verbose_name="Background color")
|
|
|
|
inverted = models.BooleanField(default=True,
|
|
verbose_name="Inverse text color")
|
|
|
|
@property
|
|
def quotted_name(self):
|
|
if ' ' in self.name:
|
|
name = '"' + escape(self.name) + '"'
|
|
else:
|
|
name = escape(self.name)
|
|
return mark_safe(name)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Milestone(models.Model):
|
|
|
|
name_validator = RegexValidator(regex='^[a-z0-9_.-]+$',
|
|
message="Please enter only lowercase characters, number, "
|
|
"dot, underscores or hyphens.")
|
|
|
|
project = models.ForeignKey(Project, related_name='milestones')
|
|
|
|
name = models.CharField(max_length=32, validators=[name_validator])
|
|
|
|
class Meta:
|
|
unique_together = ['project', 'name']
|
|
|
|
due_date = models.DateTimeField(blank=True, null=True)
|
|
|
|
closed = models.BooleanField(default=False)
|
|
|
|
def closed_issues(self):
|
|
|
|
return self.issues.filter(closed=True).count()
|
|
|
|
def total_issues(self):
|
|
|
|
return self.issues.count()
|
|
|
|
def progress(self):
|
|
|
|
closed = self.closed_issues()
|
|
total = self.total_issues()
|
|
|
|
if total:
|
|
return int(100 * closed / total)
|
|
else:
|
|
return 0
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Issue(models.Model):
|
|
|
|
global_id = models.AutoField(primary_key=True)
|
|
|
|
project = models.ForeignKey(Project, related_name='issues')
|
|
id = models.IntegerField(editable=False)
|
|
|
|
class Meta:
|
|
unique_together = ['project', 'id']
|
|
|
|
title = models.CharField(max_length=128)
|
|
|
|
author = models.ForeignKey(User, related_name='+')
|
|
|
|
opened_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
closed = models.BooleanField(default=False)
|
|
|
|
labels = models.ManyToManyField(Label, blank=True, null=True,
|
|
related_name='issues')
|
|
|
|
milestone = models.ForeignKey(Milestone, blank=True, null=True,
|
|
related_name='issues')
|
|
|
|
assignee = models.ForeignKey(User, blank=True, null=True, related_name='+')
|
|
|
|
@staticmethod
|
|
def next_id(project):
|
|
|
|
last_issue = project.issues.last()
|
|
if last_issue:
|
|
return last_issue.id + 1
|
|
else:
|
|
return 1
|
|
|
|
def comments(self):
|
|
|
|
comments = self.events.filter(issue=self, code=Event.COMMENT)
|
|
|
|
return comments
|
|
|
|
def getdesc(self):
|
|
desc = self.events.filter(issue=self, code=Event.DESCRIBE)
|
|
if desc.exists():
|
|
return desc.first().additionnal_section
|
|
else:
|
|
return None
|
|
|
|
def setdesc(self, value):
|
|
desc = self.events.filter(issue=self, code=Event.DESCRIBE)
|
|
if desc.exists():
|
|
desc = desc.first()
|
|
desc.additionnal_section = value
|
|
desc.save()
|
|
else:
|
|
desc = Event(issue=self, author=self.author, code=Event.DESCRIBE,
|
|
additionnal_section=value)
|
|
desc.save()
|
|
|
|
def deldesc(self):
|
|
desc = self.events.filter(issue=self, code=Event.DESCRIBE)
|
|
if desc.exists():
|
|
desc.first().delete()
|
|
description = property(getdesc, setdesc, deldesc)
|
|
|
|
def add_label(self, author, label, commit=True):
|
|
if self.labels.filter(pk=label.pk).exists():
|
|
return
|
|
self.labels.add(label)
|
|
if commit:
|
|
self.save()
|
|
event = Event(issue=self, author=author,
|
|
code=Event.ADD_LABEL, args={'label': label.id})
|
|
event.save()
|
|
|
|
def remove_label(self, author, label, commit=True):
|
|
self.labels.remove(label)
|
|
if commit:
|
|
self.save()
|
|
event = Event(issue=self, author=author,
|
|
code=Event.DEL_LABEL, args={'label': label.id})
|
|
event.save()
|
|
|
|
def add_milestone(self, author, milestone, commit=True):
|
|
if self.milestone == milestone:
|
|
return
|
|
if self.milestone:
|
|
event = Event(issue=self, author=author,
|
|
code=Event.CHANGE_MILESTONE,
|
|
args={'old_milestone': self.milestone.name,
|
|
'new_milestone': milestone.name})
|
|
event.save()
|
|
else:
|
|
event = Event(issue=self, author=author,
|
|
code=Event.SET_MILESTONE,
|
|
args={'milestone': milestone.name})
|
|
event.save()
|
|
self.milestone = milestone
|
|
if commit:
|
|
self.save()
|
|
|
|
def remove_milestone(self, author, milestone, commit=True):
|
|
self.milestone = None
|
|
if commit:
|
|
self.save()
|
|
event = Event(issue=self, author=author,
|
|
code=Event.UNSET_MILESTONE,
|
|
args={'milestone': milestone.name})
|
|
event.save()
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Event(models.Model):
|
|
|
|
UNKNOW = 0
|
|
CLOSE = 1
|
|
REOPEN = 2
|
|
RENAME = 3
|
|
ADD_LABEL = 4
|
|
DEL_LABEL = 5
|
|
SET_MILESTONE = 6
|
|
CHANGE_MILESTONE = 7
|
|
UNSET_MILESTONE = 8
|
|
REFERENCE = 9
|
|
COMMENT = 10
|
|
DESCRIBE = 11
|
|
ASSIGN = 12
|
|
UNASSIGN = 13
|
|
|
|
issue = models.ForeignKey(Issue, related_name="%(class)ss")
|
|
|
|
date = models.DateTimeField(auto_now_add=True)
|
|
|
|
author = models.ForeignKey(User)
|
|
|
|
code = models.IntegerField(default=UNKNOW)
|
|
|
|
_args = models.CharField(max_length=1024, blank=True, default="{}")
|
|
|
|
def getargs(self):
|
|
return json.loads(self._args)
|
|
|
|
def setargs(self, args):
|
|
self._args = json.dumps(args)
|
|
|
|
def delargs(self):
|
|
self._args = "{}"
|
|
args = property(getargs, setargs, delargs)
|
|
|
|
additionnal_section = models.TextField(blank=True, default="")
|
|
|
|
def editable(self):
|
|
|
|
return self.code == Event.COMMENT or self.code == Event.DESCRIBE
|
|
|
|
def glyphicon(self):
|
|
|
|
if self.code == Event.COMMENT:
|
|
return "comment"
|
|
elif self.code == Event.DESCRIBE:
|
|
return "edit"
|
|
elif self.code == Event.CLOSE:
|
|
return "ban-circle"
|
|
elif self.code == Event.REOPEN:
|
|
return "refresh"
|
|
elif self.code == Event.RENAME:
|
|
return "transfer"
|
|
elif self.code == Event.ADD_LABEL \
|
|
or self.code == Event.DEL_LABEL:
|
|
return "tag"
|
|
elif self.code == Event.SET_MILESTONE \
|
|
or self.code == Event.CHANGE_MILESTONE \
|
|
or self.code == Event.UNSET_MILESTONE:
|
|
return "road"
|
|
elif self.code == Event.REFERENCE:
|
|
return "transfer"
|
|
elif self.code == Event.ASSIGN \
|
|
or self.code == Event.UNASSIGN:
|
|
return "user"
|
|
else:
|
|
return "cog"
|
|
|
|
def __str__(self):
|
|
|
|
args = self.args
|
|
|
|
if self.code == Event.COMMENT or self.code == Event.DESCRIBE:
|
|
description = "commented"
|
|
elif self.code == Event.CLOSE:
|
|
description = "closed this issue"
|
|
elif self.code == Event.REOPEN:
|
|
description = "reopened this issue"
|
|
elif self.code == Event.RENAME:
|
|
description = "changed the title from <mark>{old_title}</mark> " \
|
|
"to <mark>{new_title}</mark>"
|
|
elif self.code == Event.ADD_LABEL or self.code == Event.DEL_LABEL:
|
|
label = Label.objects.get(id=args['label'])
|
|
if self.code == Event.ADD_LABEL:
|
|
action = 'added'
|
|
else:
|
|
action = 'removed'
|
|
description = '%s the <a href="%s">%s</a> label' \
|
|
% (action, same_label(label), labeled(label))
|
|
elif self.code == Event.SET_MILESTONE:
|
|
description = "added this to the {milestone} milestone"
|
|
elif self.code == Event.CHANGE_MILESTONE:
|
|
description = "moved this from the {old_milestone} milestone " \
|
|
"to the {new_milestone} milestone"
|
|
elif self.code == Event.UNSET_MILESTONE:
|
|
description = "deleted this from the {milestone} milestone"
|
|
elif self.code == Event.REFERENCE:
|
|
description = "referenced this issue"
|
|
else:
|
|
return None
|
|
|
|
# Escape args
|
|
safe_args = {k: escape(v) for k, v in args.items()}
|
|
|
|
return mark_safe(description.format(**safe_args))
|
|
|
|
|
|
class Settings(models.Model):
|
|
|
|
site = models.OneToOneField(Site)
|
|
|
|
class Meta:
|
|
verbose_name_plural = 'Settings'
|
|
|
|
|
|
class Team(models.Model):
|
|
|
|
name = models.CharField(max_length=128, unique=True)
|
|
|
|
users = models.ManyToManyField(auth.models.User, blank=True, null=True,
|
|
related_name='teams')
|
|
groups = models.ManyToManyField(auth.models.Group, blank=True, null=True,
|
|
related_name='teams')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class PermissionModel(models.Model):
|
|
|
|
GRANTEE_USER = 0
|
|
GRANTEE_GROUP = 1
|
|
GRANTEE_TEAM = 2
|
|
GRANTEE_TYPE = (
|
|
(GRANTEE_USER, 'User'),
|
|
(GRANTEE_GROUP, 'Group'),
|
|
(GRANTEE_TEAM, 'Team'),
|
|
)
|
|
|
|
grantee_type = models.IntegerField(choices=GRANTEE_TYPE,
|
|
default=GRANTEE_USER, verbose_name="Type")
|
|
grantee_name = models.CharField(max_length=50,
|
|
verbose_name="Name")
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
def granted_to(self, user):
|
|
if not user.is_authenticated():
|
|
return False
|
|
if self.grantee_type == self.GRANTEE_USER:
|
|
return user.username == self.grantee_name
|
|
elif self.grantee_type == self.GRANTEE_GROUP:
|
|
return user.groups.filter(name=self.grantee_name).exists()
|
|
elif self.grantee_type == self.GRANTEE_TEAM:
|
|
return Team.objects.filter(name=self.grantee_name) \
|
|
.filter(Q(groups__in=user.groups.all()) | Q(users=user)) \
|
|
.exists()
|
|
else:
|
|
return False
|
|
|
|
def type(self):
|
|
return dict(self.GRANTEE_TYPE)[self.grantee_type]
|
|
|
|
def name(self):
|
|
return self.grantee_name
|
|
|
|
def __str__(self):
|
|
return self.grantee_name
|
|
|
|
|
|
class GlobalPermission(PermissionModel):
|
|
|
|
create_project = models.BooleanField(default=True)
|
|
modify_project = models.BooleanField(default=False)
|
|
delete_project = models.BooleanField(default=False)
|
|
|
|
add_team = models.BooleanField(default=True)
|
|
manage_team = models.BooleanField(default=False)
|
|
|
|
manage_global_permission = models.BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
return self.grantee_name + "'s global permissions"
|
|
|
|
|
|
class ProjectPermission(PermissionModel):
|
|
|
|
project = models.ForeignKey(Project, editable=False,
|
|
related_name='permissions')
|
|
|
|
manage_project_permission = models.BooleanField(default=False)
|
|
|
|
create_issue = models.BooleanField(default=True)
|
|
modify_issue = models.BooleanField(default=False)
|
|
manage_issue = models.BooleanField(default=False)
|
|
delete_issue = models.BooleanField(default=False)
|
|
|
|
create_comment = models.BooleanField(default=True)
|
|
modify_comment = models.BooleanField(default=False)
|
|
delete_comment = models.BooleanField(default=False)
|
|
|
|
manage_tags = models.BooleanField(default=False)
|
|
delete_tags = models.BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
return self.grantee_name + "'s permissions on " \
|
|
+ self.project.name + " project"
|
|
|
|
|
|
if VERSION < (1, 7):
|
|
import issue.signals
|