from django.contrib.auth.models import PermissionsMixin from django.contrib.auth.base_user import AbstractBaseUser, BaseUserManager from django.db import models, transaction class Edition(models.Model): year = models.PositiveIntegerField('Rok', unique=True) roman = models.CharField('Nr. edycji', max_length=25) active = models.BooleanField('Aktualna', default=False) submissions = models.BooleanField('Możliwość zgłoszenia drużyny', default=False) scores_available = models.BooleanField('Wyniki dostępne dla wszystkich', default=False) scores_eliminations = models.BooleanField('Wyniki z eliminacji dostępne dla nauczycieli', default=False) entry_threshold = models.IntegerField('Próg punkowy - wejście do finału', default=0) award_threshold = models.IntegerField('Próg punktowy - wyróżnienie', default=0) laureate_threshold = models.IntegerField('Próg punktowy - tytuł laureata', default=0) scores = models.FileField('Wyniki do wczytania', blank=True, upload_to='wyniki', help_text='Uwaga! Wyniki muszą być zawarte w pliku .csv. Kolumny po kolei to odpowiednio: identyfikator ucznia, wynik z eliminacji, wynik z finału.') first_test = models.FileField('Zadania eliminacyjne', blank=True, upload_to='eliminacje') second_test = models.FileField('Zadania finałowe', blank=True, upload_to='finaly') def __str__(self): return f'Edycja {self.roman}' @staticmethod def current(): return Edition.objects.filter(active=True).first() def save(self, *args, **kwargs): if self.active is True: Edition.objects.exclude(id=self.id).update(active=False) super(Edition, self).save(*args, **kwargs) class Meta: verbose_name = 'Edycja' verbose_name_plural = 'Edycje' class Student(models.Model): name = models.CharField('Imię', max_length=50) surname = models.CharField('Nazwisko', max_length=50) grade = models.PositiveIntegerField('Klasa', choices=map(lambda x: (x, x), range(1, 9))) score_first = models.IntegerField('Wynik z eliminacji', default=0) score_second = models.IntegerField('Wynik z finału', default=0) school_name = models.CharField('Nazwa szkoły', max_length=100, blank=True) school_town = models.CharField('Miejscowość', max_length=100, blank=True) school_address = models.CharField('Adres', max_length=100, blank=True) identifier = models.CharField('Identyfikator', max_length=100, blank=True, editable=True, help_text='UWAGA - tej wartości należy nie zmieniać') def __str__(self): return f'{self.surname} {self.name}' class Meta: verbose_name = 'Uczeń' verbose_name_plural = 'Uczniowie' class Announcement(models.Model): title = models.CharField('Tytuł', max_length=250) content = models.TextField('Treść') created_at = models.DateTimeField(auto_now_add=True) def __str__(self): return self.title class Meta: verbose_name = 'Ogłoszenie' verbose_name_plural = 'Ogłoszenia' ordering = ['-created_at'] class UserManager(BaseUserManager): use_in_migrations = True def _create_user(self, email, password, **extra_fields): if not email: raise ValueError('The given email must be set') email = self.normalize_email(email) user = self.model(email=email, **extra_fields) user.set_password(password) user.save(using=self._db) return user def create_user(self, email, password=None, **extra_fields): extra_fields.setdefault('is_superuser', False) return self._create_user(email, password, **extra_fields) def create_superuser(self, email, password, **extra_fields): extra_fields.setdefault('is_superuser', True) extra_fields.setdefault('is_staff', True) extra_fields.setdefault('request_notifications', True) extra_fields.setdefault('email_notifications', True) if extra_fields.get('is_superuser') is not True: raise ValueError('Superuser must have is_superuser=True.') return self._create_user(email, password, **extra_fields) class School(AbstractBaseUser, PermissionsMixin): email = models.EmailField('Adres e-mail', unique=True) name = models.CharField('Nazwa szkoły', max_length=100) town = models.CharField('Miejscowość', max_length=100) address = models.CharField('Adres', max_length=100) phone = models.PositiveIntegerField('Numer telefonu', null=True) is_staff = models.BooleanField('Staff status', default=False) is_active = models.BooleanField('Active status', default=True) request_notifications = models.BooleanField('Powiadomienia przy wysłaniu zgłoszenia', default=False) email_notifications = models.BooleanField('Powiadomienia przy wysłaniu wiadomości email', default=False) objects = UserManager() USERNAME_FIELD = 'email' REQUIRED_FIELDS = [] def __str__(self): return self.name or self.email def get_full_name(self): return self.email def get_short_name(self): return self.email def updateTeam(self): with transaction.atomic(): identifier = f'{Edition.current().year}-{self.id}-' Student.objects.filter(identifier__startswith=identifier).update( school_name=self.name, school_town=self.town, school_address=self.address ) class Meta: verbose_name = 'Szkoła' verbose_name_plural = 'Szkoły'