This commit is contained in:
Egor Matveev
2021-11-20 15:24:37 +03:00
parent b4ca90188b
commit 9a20711820
6 changed files with 24 additions and 9 deletions

View File

@@ -7,6 +7,7 @@ from Main.models.settask import SetTask
from Main.models.subscription import Subscription
from Main.models.task import Task
from Sprint.settings import CONSTS
from SprintLib.language import languages
class UserInfo(models.Model):
@@ -21,6 +22,16 @@ class UserInfo(models.Model):
telegram_chat_id = models.TextField(default="")
notification_solution_result = models.BooleanField(default=False)
@property
def has_favourite_language(self):
return self.favourite_language_id is not None
@property
def favourite_language(self):
if not self.has_favourite_language:
return None
return languages[self.favourite_language_id]
def _append_task(self, task, tasks):
if task.creator == self.user or task.public or self.user.is_superuser:
tasks.append(task)