models.py 1.37 KB
Newer Older
Mike Chen committed
1 2
from django.db import models
from django.contrib.auth.models import User
3
import logging
Mike Chen committed
4 5 6


class Role(models.Model):
7
    name = models.CharField(max_length=30, null=False, blank=False)
Mike Chen committed
8
    users = models.ManyToManyField(User, related_name="roles")
9
    course_id = models.CharField(max_length=255, blank=True, db_index=True)
Mike Chen committed
10 11

    def __unicode__(self):
12
        return self.name + " for " + (self.course_id if self.course_id else "all courses")
Mike Chen committed
13

Rocky Duan committed
14 15
    def inherit_permissions(self, role): # TODO the name of this method is a little bit confusing,
                                         # since it's one-off and doesn't handle inheritance later
16 17 18 19 20
        if role.course_id and role.course_id != self.course_id:
            logging.warning("%s cannot inheret permissions from %s due to course_id inconsistency" % 
                            (self, role))
        for per in role.permissions.all():
            self.add_permission(per)
Mike Chen committed
21

22 23
    def add_permission(self, permission):
        self.permissions.add(Permission.objects.get_or_create(name=permission)[0])
Mike Chen committed
24

25 26
    def has_permission(self, permission):
        return self.permissions.filter(name=permission).exists()
Mike Chen committed
27 28 29 30 31 32 33 34


class Permission(models.Model):
    name = models.CharField(max_length=30, null=False, blank=False, primary_key=True)
    roles = models.ManyToManyField(Role, related_name="permissions")

    def __unicode__(self):
        return self.name