models.py

import uuid

import django.contrib.auth.models
import django.db.models


class CompanyManager(django.contrib.auth.models.BaseUserManager):
    def create_company(self, email, name, password=None, **extra_fields):
        if not email:
            raise ValueError('The Email must be set')

        email = self.normalize_email(email)
        company = self.model(
            email=email,
            name=name,
            **extra_fields,
        )
        company.set_password(password)
        company.save(using=self._db)
        return company


class Company(django.contrib.auth.models.AbstractBaseUser):
    id = django.db.models.UUIDField(
        'UUID',
        primary_key=True,
        default=uuid.uuid4,
        editable=False,
    )
    email = django.db.models.EmailField(
        unique=True,
        max_length=120,
    )
    name = django.db.models.CharField(max_length=50)

    token_version = django.db.models.IntegerField(default=0)
    created_at = django.db.models.DateTimeField(auto_now_add=True)
    is_active = django.db.models.BooleanField(default=True)

    objects = CompanyManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['name']

    def __str__(self):
        return self.name


class Promo(django.db.models.Model):
    MODE_COMMON = 'COMMON'
    MODE_UNIQUE = 'UNIQUE'
    MODE_CHOICES = [
        (MODE_COMMON, 'Common'),
        (MODE_UNIQUE, 'Unique'),
    ]

    id = django.db.models.UUIDField(
        'UUID',
        primary_key=True,
        default=uuid.uuid4,
        editable=False,
    )

    company = django.db.models.ForeignKey(
        Company,
        on_delete=django.db.models.CASCADE,
        null=True,
        blank=True,
    )
    description = django.db.models.CharField(max_length=300)
    image_url = django.db.models.URLField(
        max_length=350,
        blank=True,
        null=True,
    )
    target = django.db.models.JSONField(default=dict)
    max_count = django.db.models.IntegerField()
    active_from = django.db.models.DateField(null=True, blank=True)
    active_until = django.db.models.DateField(null=True, blank=True)
    mode = django.db.models.CharField(max_length=10, choices=MODE_CHOICES)
    promo_common = django.db.models.CharField(
        max_length=30,
        blank=True,
        null=True,
    )
    active = django.db.models.BooleanField(default=True)

    created_at = django.db.models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f'Promo {self.id} ({self.mode})'


class PromoCode(django.db.models.Model):
    promo = django.db.models.ForeignKey(
        Promo,
        on_delete=django.db.models.CASCADE,
        related_name='unique_codes',
        to_field='id',
    )
    code = django.db.models.CharField(max_length=30)
    is_used = django.db.models.BooleanField(default=False)
    used_at = django.db.models.DateTimeField(null=True, blank=True)

    class Meta:
        unique_together = ('promo', 'code')

    def __str__(self):
        return self.code