c7a5d97632d54f3631f5f0dd872c9af90c00c473
[cas.git] / src / services / models.py
1 import secrets
2 from django.conf import settings
3 from django.contrib.auth.models import User
4 from django.db import models
5 from django.utils.translation import ugettext_lazy as _
6
7
8 class Service(models.Model):
9     name = models.CharField(_('name'), max_length=255)
10     url = models.URLField(_('URL'), blank=True)
11     key = models.CharField(_('key'), max_length=255, blank=True)
12     uses_ssh = models.BooleanField(_('uses SSH'), default=False)
13     groups = models.ManyToManyField('auth.Group', verbose_name=_('groups'), blank=True)
14     users = models.ManyToManyField(settings.AUTH_USER_MODEL, verbose_name=_('users'), blank=True)
15
16     class Meta:
17         ordering = ('name', )
18         verbose_name = _('service')
19         verbose_name_plural = _('services')
20
21     def __str__(self):
22         return self.name
23
24     def save(self, *args, **kwargs):
25         if not self.key:
26             self.key = secrets.token_urlsafe()
27         return super().save(*args, **kwargs)
28
29     def all_users(self):
30         return User.objects.filter(
31                 models.Q(service=self) |
32                 models.Q(groups__service=self)
33             ).distinct()
34
35     @classmethod
36     def for_user(cls, user):
37         return cls.objects.filter(models.Q(users=user) | models.Q(groups__user=user))
38
39
40 class Hook(models.Model):
41     service = models.ForeignKey(Service, on_delete=models.CASCADE)
42     email = models.CharField(max_length=255)