sprint/Main/models/group.py
2022-02-03 00:20:07 +03:00

24 lines
796 B
Python

from django.contrib.auth.models import User
from django.contrib.postgres.fields import ArrayField
from django.db import models
from django.db.models import Q
from django.utils import timezone
from Main.models.set import Set
class Group(models.Model):
name = models.TextField()
sets = models.ManyToManyField(Set)
creator = models.ForeignKey(User, null=True, on_delete=models.SET_NULL)
users = models.ManyToManyField(User, related_name="user_groups")
editors = ArrayField(models.TextField(), default=list)
@property
def available_sets(self):
return self.sets.filter(
Q(opened=True)
& (Q(start_time__isnull=True) | Q(start_time__lte=timezone.now()))
& (Q(end_time__isnull=True) | Q(end_time__lte=timezone.now()))
)