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)

View File

@@ -1,5 +1,4 @@
from django import template
from Main.models import SetTask, Subscription
register = template.Library()

View File

@@ -28,7 +28,7 @@ class AccountView(BaseView):
return "/account"
def post_set_language(self):
lang_id = self.request.POST["language"]
lang_id = int(self.request.POST["language"])
self.request.user.userinfo.favourite_language_id = lang_id if lang_id != -1 else None
self.request.user.userinfo.save()
return "/account"