group

Main/migrations/0017_auto_20220201_0033.py

@@ -0,0 +1,32 @@
+# Generated by Django 3.2.4 on 2022-01-31 21:33
+
+from django.conf import settings
+import django.contrib.postgres.fields
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('Main', '0016_userinfo_teacher'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='group',
+            name='creator',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL),
+        ),
+        migrations.AddField(
+            model_name='group',
+            name='editors',
+            field=django.contrib.postgres.fields.ArrayField(base_field=models.TextField(), default=list, size=None),
+        ),
+        migrations.AddField(
+            model_name='group',
+            name='users',
+            field=models.ManyToManyField(related_name='user_groups', to=settings.AUTH_USER_MODEL),
+        ),
+    ]

Main/models/group.py

@@ -1,3 +1,5 @@
+from django.contrib.auth.models import User
+from django.contrib.postgres.fields import ArrayField
 from django.db import models
 from Main.models.set import Set
 
@@ -5,3 +7,6 @@ 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)

Main/models/userinfo.py

@@ -1,6 +1,6 @@
 from functools import cached_property
 
-from django.contrib.auth.models import User
+from django.contrib.auth.models import User, Group
 from django.db import models
 from django.db.models import Q
 from django.utils import timezone