jumpserver/apps/perms/models.py

from __future__ import unicode_literals, absolute_import

from django.db import models
from django.utils.translation import ugettext_lazy as _
from django.utils import timezone

from users.models import User, UserGroup
from assets.models import Asset, AssetGroup, SystemUser
from common.utils import date_expired_default


class PermUserAsset(models.Model):
    ACTION_CHOICE = (
        ('1', 'Allow'),
        ('0', 'Deny'),
    )

    user = models.ForeignKey(User, on_delete=models.CASCADE)
    action = models.CharField(choices=ACTION_CHOICE, max_length=8, default='1')
    assets = models.ManyToManyField(Asset, blank=True)
    asset_groups = models.ManyToManyField(AssetGroup,  blank=True)
    system_users = models.ManyToManyField(SystemUser,  blank=True)
    date_expired = models.DateTimeField(default=date_expired_default, verbose_name=_('Date expired'))
    created_by = models.CharField(max_length=128, blank=True)
    date_created = models.DateTimeField(auto_now=True)
    comment = models.TextField(verbose_name=_('Comment'))

    def __unicode__(self):
        return '%(id)s: %(user)s %(action)s' % {
            'id': self.id,
            'user': self.user.username,
            'action': self.action,
        }

    @property
    def is_expired(self):
        if self.date_expired > timezone.now():
            return False
        else:
            return True

    class Meta:
        db_table = 'perm_user_asset'


class PermUserGroupAsset(models.Model):
    ACTION_CHOICES = (
        ('0', 'Deny'),
        ('1', 'Allow'),
    )

    user_group = models.ForeignKey(User,  on_delete=models.CASCADE)
    action = models.CharField(choices=ACTION_CHOICES, max_length=8, default='1')
    assets = models.ManyToManyField(Asset, blank=True)
    asset_groups = models.ManyToManyField(AssetGroup, blank=True)
    system_users = models.ManyToManyField(SystemUser, blank=True)
    date_expired = models.DateTimeField(default=date_expired_default, verbose_name=_('Date expired'))
    created_by = models.CharField(max_length=128)
    date_created = models.DateTimeField(auto_now=True)
    comment = models.TextField(verbose_name=_('Comment'))

    def __unicode__(self):
        return '%(id)s: %(user)s %(action)s' % {
            'id': self.id,
            'user': self.user_group.name,
            'action': self.action,
        }

    @property
    def is_expired(self):
        if self.date_expired > timezone.now():
            return False
        else:
            return True

    class Meta:
        db_table = 'perm_user_group_asset'