initial

web/admin.py

@@ -0,0 +1,9 @@
+from django.contrib import admin
+
+# Register your models here.
+from .models import *
+
+
+admin.site.register(CustomUser)
+admin.site.register(Project)
+admin.site.register(Membership)

web/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class WebConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'web'

web/migrations/0001_initial.py

@@ -0,0 +1,63 @@
+# Generated by Django 4.1.7 on 2023-09-23 09:51
+
+from django.conf import settings
+import django.contrib.auth.models
+import django.contrib.auth.validators
+from django.db import migrations, models
+import django.db.models.deletion
+import django.utils.timezone
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('auth', '0012_alter_user_first_name_max_length'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CustomUser',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('password', models.CharField(max_length=128, verbose_name='password')),
+                ('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
+                ('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
+                ('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
+                ('first_name', models.CharField(blank=True, max_length=150, verbose_name='first name')),
+                ('last_name', models.CharField(blank=True, max_length=150, verbose_name='last name')),
+                ('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
+                ('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
+                ('is_active', models.BooleanField(default=True, help_text='Designates whether this user should be treated as active. Unselect this instead of deleting accounts.', verbose_name='active')),
+                ('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
+                ('vk_id', models.IntegerField(null=True)),
+                ('yandex_id', models.BigIntegerField(null=True)),
+                ('groups', models.ManyToManyField(blank=True, help_text='The groups this user belongs to. A user will get all permissions granted to each of their groups.', related_name='user_set', related_query_name='user', to='auth.group', verbose_name='groups')),
+                ('user_permissions', models.ManyToManyField(blank=True, help_text='Specific permissions for this user.', related_name='user_set', related_query_name='user', to='auth.permission', verbose_name='user permissions')),
+            ],
+            options={
+                'verbose_name': 'user',
+                'verbose_name_plural': 'users',
+                'abstract': False,
+            },
+            managers=[
+                ('objects', django.contrib.auth.models.UserManager()),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Project',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.TextField()),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Membership',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('project', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='web.project')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

web/migrations/0002_customuser_only_one_project.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.1.7 on 2023-09-23 11:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('web', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='customuser',
+            name='only_one_project',
+            field=models.BooleanField(default=True),
+        ),
+    ]

web/migrations/0003_customuser_selected_project.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.1.7 on 2023-09-23 11:34
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('web', '0002_customuser_only_one_project'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='customuser',
+            name='selected_project',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='web.project'),
+        ),
+    ]

web/models/__init__.py

@@ -0,0 +1,3 @@
+from .custom_user import CustomUser
+from .project import Project
+from .membership import Membership

web/models/custom_user.py

@@ -0,0 +1,19 @@
+from django.contrib.auth.models import AbstractUser
+from django.db import models
+
+
+class CustomUser(AbstractUser):
+    vk_id = models.IntegerField(null=True, blank=True)
+    yandex_id = models.BigIntegerField(null=True, blank=True)
+    only_one_project = models.BooleanField(default=True)
+    selected_project = models.ForeignKey('Project', on_delete=models.CASCADE, null=True)
+
+    @property
+    def profile_photo_url(self):
+        return f"/profile_photo?id={self.id}"
+
+    @property
+    def projects(self):
+        from web.models import Membership
+        for membership in Membership.objects.filter(user=self):
+            yield membership.project

web/models/membership.py

@@ -0,0 +1,9 @@
+from django.db import models
+
+from web.models.custom_user import CustomUser
+from web.models.project import Project
+
+
+class Membership(models.Model):
+    user = models.ForeignKey(CustomUser, on_delete=models.CASCADE)
+    project = models.ForeignKey(Project, on_delete=models.CASCADE)

web/models/project.py

@@ -0,0 +1,5 @@
+from django.db import models
+
+
+class Project(models.Model):
+    name = models.TextField()

web/templatetags/filters.py

@@ -0,0 +1,4 @@
+from django import template
+
+register = template.Library()
+

web/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

web/urls.py

@@ -0,0 +1,13 @@
+from django.contrib import admin
+from django.urls import path
+
+from .views import *
+
+urlpatterns = [
+    path(*WelcomeView.as_path()),
+    path(*ProjectsView.as_path()),
+    path(*SelectProject.as_path()),
+    path(*ProfileView.as_path()),
+    path(*ProfilePhoto.as_path()),
+    path(*LogoutView.as_path())
+]

web/views/__init__.py

@@ -0,0 +1,6 @@
+from .welcome import WelcomeView
+from .projects import ProjectsView
+from .select_project import SelectProject
+from .profile import ProfileView
+from .profile_photo import ProfilePhoto
+from .logout import LogoutView

web/views/logout.py

@@ -0,0 +1,12 @@
+from django.contrib.auth import logout
+
+from BaseLib.BaseView import BaseView
+
+
+class LogoutView(BaseView):
+    required_login = True
+    endpoint = 'logout'
+
+    def get(self):
+        logout(self.request)
+        return '/welcome'

web/views/profile.py

@@ -0,0 +1,24 @@
+from django.contrib.auth import login
+
+from BaseLib.BaseView import BaseView
+from BaseLib.minio import minio_client
+
+
+class ProfileView(BaseView):
+    required_login = True
+    endpoint = 'profile'
+    view_file = 'profile.html'
+
+    def get(self):
+        self.context['owner'] = not hasattr(self, 'customUser') or self.customUser == self.request.user
+        self.context['account'] = getattr(self, 'customUser', None) or self.request.user
+
+    def post_upload_photo(self):
+        minio_client.put_object(self.request.FILES["file"].read(), f"profile_photos/{self.request.user.id}")
+        return self
+
+    def post_change_password(self):
+        self.request.user.set_password(self.request.POST['password'])
+        self.request.user.save()
+        login(self.request, self.request.user)
+        return self

web/views/profile_photo.py

@@ -0,0 +1,17 @@
+from django.http import HttpResponse
+
+from BaseLib.BaseView import BaseView
+from BaseLib.minio import minio_client
+
+
+class ProfilePhoto(BaseView):
+    required_login = True
+    endpoint = "profile_photo"
+
+    def get(self):
+        profile_photo_bytes = minio_client.get_object(f"profile_photos/{self.request.GET['id']}")
+        if profile_photo_bytes == b"":
+            profile_photo_bytes = open("content/empty_profile_photo.jpg", "rb").read()
+        return HttpResponse(
+            profile_photo_bytes, content_type="image/jpg"
+        )

web/views/projects.py

@@ -0,0 +1,7 @@
+from BaseLib.BaseView import BaseView
+
+
+class ProjectsView(BaseView):
+    required_login = True
+    endpoint = ''
+    view_file = 'projects.html'

web/views/select_project.py

@@ -0,0 +1,11 @@
+from BaseLib.BaseView import BaseView
+
+
+class SelectProject(BaseView):
+    required_login = True
+    endpoint = 'select_project'
+
+    def get(self):
+        self.request.user.selected_project = self.project
+        self.request.user.save()
+        return '/'

web/views/welcome.py

@@ -0,0 +1,18 @@
+from django.contrib.auth import authenticate, login
+
+from BaseLib.BaseView import BaseView
+
+
+class WelcomeView(BaseView):
+    required_login = False
+    endpoint = 'welcome'
+    view_file = 'welcome.html'
+
+    def post(self):
+        username = self.request.POST['username']
+        password = self.request.POST['password']
+        user = authenticate(username=username, password=password)
+        if user is None:
+            return self
+        login(self.request, user)
+        return '/'