From 7b4f57e0726d8314e31126326df5faf567817ad0 Mon Sep 17 00:00:00 2001
From: Egor Matveev <emmtvv@icloud.com>
Date: Sat, 22 Jan 2022 22:04:48 +0300
Subject: [PATCH] teacher

---
 Main/models/userinfo.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/Main/models/userinfo.py b/Main/models/userinfo.py
index b238208..9891724 100644
--- a/Main/models/userinfo.py
+++ b/Main/models/userinfo.py
@@ -7,9 +7,6 @@ from django.utils import timezone
 
 from Main.models.friendship import Friendship
 from Main.models.set import Set
-from Main.models.group import Group
-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
@@ -29,6 +26,7 @@ class UserInfo(models.Model):
     notification_messages = models.BooleanField(default=False)
     code = models.IntegerField(null=True)
     verified = models.BooleanField(default=False)
+    teacher = models.BooleanField(default=False)
 
     @property
     def tasks_solved(self):
@@ -69,8 +67,7 @@ class UserInfo(models.Model):
 
     @property
     def can_create(self):
-        # todo:
-        return self.user.is_superuser
+        return self.user.is_superuser or self.teacher
 
     @property
     def has_profile_pic(self):