backend/app/users/models.py

from django.contrib.auth.models import AbstractUser
from django.core.validators import MinValueValidator
from django.db import models
from faker import Faker


class Clan(models.Model):
    name = models.CharField(max_length=100, null=True)

    def save(self, *args, **kwargs):
        name = Faker().name()
        self.name = name  + "'s clan"
        super(Clan, self).save(*args, **kwargs)


class User(AbstractUser):
    class WorkerType(models.TextChoices):
        WORKER = "WORKER", "worker"
        HR = "HR", "human resources"
        ADMIN = "ADMIN", "administrator"

    first_name = None
    last_name = None

    # image = models.ImageField(upload_to=user_file_upload_mixin, blank=True)
    # image_cropped = models.ImageField(upload_to="cropped/", blank=True)

    about = models.TextField(blank=True)
    name = models.CharField(max_length=120, null=True)
    type = models.CharField(
        max_length=6, choices=WorkerType.choices, default=WorkerType.WORKER
    )
    salary = models.IntegerField(default=0)
    clan = models.ForeignKey(Clan, on_delete=models.CASCADE, null=True)
    salary = models.IntegerField(default=0, validators=[MinValueValidator(0)])
    respect = models.IntegerField(default=0, validators=[MinValueValidator(0)])
    wallet_private_key = models.CharField(max_length=96, unique=True, null=True)
    wallet_public_key = models.CharField(max_length=96, unique=True, null=True)
    telegram = models.CharField(max_length=100, unique=True, null=True)

    def __str__(self):
        return self.username

    @property
    def is_manager(self):
        return self.type in [self.WorkerType.HR, self.WorkerType.ADMIN]

    @property
    def is_admin(self):
        return self.type == self.WorkerType.ADMIN

    class Meta:
        ordering = ["-id"]