inpt/ponytracker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add permissions models and settings model

Browse source
This commit is contained in:
Élie Bouttier 2014-08-07 13:51:57 -07:00
parent 98f181ca4f
commit dcadc324db
4 changed files with 197 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
issue/admin.py
View file

@ -6,3 +6,7 @@ admin.site.register(Issue)
admin.site.register(Event)
admin.site.register(Label)
admin.site.register(Milestone)
admin.site.register(Settings)
admin.site.register(Team)
admin.site.register(GlobalPermission)
admin.site.register(ProjectPermission)

106
issue/migrations/0006_auto_20140807_2032.py Normal file
View file

@ -0,0 +1,106 @@
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.db import models, migrations
from django.conf import settings
class Migration(migrations.Migration):
dependencies = [
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
('auth', '0001_initial'),
('contenttypes', '0001_initial'),
('sites', '0001_initial'),
('issue', '0005_project_public'),
]
operations = [
migrations.CreateModel(
name='GlobalPermission',
fields=[
('id', models.AutoField(primary_key=True, auto_created=True, verbose_name='ID', serialize=False)),
('grantee', models.CharField(max_length=50)),
('create_project', models.BooleanField(default=True)),
('modify_project', models.BooleanField(default=False)),
('delete_project', models.BooleanField(default=False)),
('content_type', models.ForeignKey(to='contenttypes.ContentType')),
],
options={
'abstract': False,
},
bases=(models.Model,),
),
migrations.CreateModel(
name='ProjectPermission',
fields=[
('id', models.AutoField(primary_key=True, auto_created=True, verbose_name='ID', serialize=False)),
('grantee', models.CharField(max_length=50)),
('create_issue', models.BooleanField(default=True)),
('modify_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)),
('create_label', models.BooleanField(default=True)),
('modify_label', models.BooleanField(default=False)),
('delete_label', models.BooleanField(default=False)),
('create_milestone', models.BooleanField(default=True)),
('modify_milestone', models.BooleanField(default=False)),
('delete_milestone', models.BooleanField(default=False)),
('content_type', models.ForeignKey(to='contenttypes.ContentType')),
('project', models.ForeignKey(to='issue.Project')),
],
options={
'abstract': False,
},
bases=(models.Model,),
),
migrations.CreateModel(
name='Settings',
fields=[
('id', models.AutoField(primary_key=True, auto_created=True, verbose_name='ID', serialize=False)),
('site', models.OneToOneField(to='sites.Site')),
],
options={
'verbose_name_plural': 'Settings',
},
bases=(models.Model,),
),
migrations.CreateModel(
name='Team',
fields=[
('id', models.AutoField(primary_key=True, auto_created=True, verbose_name='ID', serialize=False)),
('name', models.CharField(max_length=128, unique=True)),
('groups', models.ManyToManyField(to='auth.Group', blank=True, null=True)),
('users', models.ManyToManyField(to=settings.AUTH_USER_MODEL, blank=True, null=True)),
],
options={
},
bases=(models.Model,),
),
migrations.CreateModel(
name='User',
fields=[
],
options={
'proxy': True,
},
bases=('auth.user',),
),
migrations.AlterField(
model_name='event',
name='author',
field=models.ForeignKey(to='issue.User'),
),
migrations.AlterField(
model_name='issue',
name='assignee',
field=models.ForeignKey(to='issue.User', null=True, blank=True),
),
migrations.AlterField(
model_name='issue',
name='author',
field=models.ForeignKey(to='issue.User'),
),
]

85
issue/models.py
View file

@ -1,10 +1,14 @@
from django.db import models
from django.contrib.auth.models import User
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 colorful.fields import RGBColorField
@ -13,6 +17,15 @@ import json
from issue.templatetags.issue_tags import same_label, labeled
class User(auth.models.User):
class Meta:
proxy = True
@property
def teams(self):
return Team.objects.filter(Q(groups__in=self.groups.all()) | Q(users=self))
class Project(models.Model):
url_name_validator = RegexValidator(regex='^[a-z0-9_-]+$',
@ -301,3 +314,73 @@ class Event(models.Model):
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 = models.CharField(max_length=50)
# type of grantee (user, group, team)
content_type = models.ForeignKey(ContentType)
class Meta:
abstract = True
def granted_to(self, user):
if self.content_type == ContentType.objects.get_for_model(User):
return user.username == grantee
elif self.content_type == ContentType.objects.get_for_model(Group):
group = self.content_type.get_object_for_this_type(name=grantee)
return user.groups.filter(name=grantee).exists()
elif self.content_type == ContentType.objects.get_for_model(Team):
team = self.content_type.get_object_for_this_type(name=grantee)
return team in user.teams
else:
return False
class GlobalPermission(PermissionModel):
create_project = models.BooleanField(default=True)
modify_project = models.BooleanField(default=False)
delete_project = models.BooleanField(default=False)
def __str__(self):
return self.grantee + "'s global permissions"
class ProjectPermission(PermissionModel):
project = models.ForeignKey(Project, related_name='permissions')
create_issue = models.BooleanField(default=True)
modify_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)
create_label = models.BooleanField(default=True)
modify_label = models.BooleanField(default=False)
delete_label = models.BooleanField(default=False)
create_milestone = models.BooleanField(default=True)
modify_milestone = models.BooleanField(default=False)
delete_milestone = models.BooleanField(default=False)
def __str__(self):
return self.grantee + "'s permissions on " + self.project.name + " project"

3
ponytracker/settings.py
View file

@ -38,6 +38,7 @@ INSTALLED_APPS = (
'django.contrib.staticfiles',
'django.contrib.humanize',
'django.contrib.sites',
'django_markdown',
'bootstrap3_datetime',
@ -109,3 +110,5 @@ TEMPLATE_CONTEXT_PROCESSORS = (
'django.core.context_processors.request',
'issue.context_processors.projects',
)
SITE_ID = 1