notifications and email

Main/migrations/0038_auto_20220516_1836.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.2.4 on 2022-05-16 15:36
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('Main', '0037_alter_userinfo_telegram_chat_id'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='userinfo',
+            name='notification_email',
+            field=models.BooleanField(default=False),
+        ),
+        migrations.AddField(
+            model_name='userinfo',
+            name='notification_telegram',
+            field=models.BooleanField(default=False),
+        ),
+    ]

Main/models/userinfo.py

@@ -26,6 +26,8 @@ class UserInfo(models.Model):
     notification_solution_result = models.BooleanField(default=False)
     notification_friends = models.BooleanField(default=False)
     notification_messages = models.BooleanField(default=False)
+    notification_telegram = models.BooleanField(default=False)
+    notification_email = models.BooleanField(default=False)
     code = models.IntegerField(null=True, blank=True)
     verified = models.BooleanField(default=False)
     teacher = models.BooleanField(default=False)

Main/views/AccountView.py

@@ -84,6 +84,11 @@ class AccountView(BaseView):
         self.request.user.userinfo.save()
         return "/account"
 
+    def post_add_email(self):
+        self.request.user.email = self.request.POST['email']
+        self.request.user.save()
+        return "/account"
+
     def post_notifications(self):
         for attr in dir(self.request.user.userinfo):
             if attr.startswith("notification"):