mirror of
https://github.com/more-tech4-magnum-opus/backend.git
synced 2024-09-21 10:08:47 +03:00
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
from django.contrib.auth.models import AbstractUser
|
|
from django.core.validators import MinValueValidator
|
|
from django.db import models
|
|
|
|
from faker import Faker
|
|
|
|
|
|
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)
|
|
type = models.CharField(
|
|
max_length=6, choices=WorkerType.choices, default=WorkerType.WORKER
|
|
)
|
|
clan = models.ForeignKey(
|
|
"users.Clan",
|
|
related_name="users",
|
|
on_delete=models.SET_NULL,
|
|
null=True,
|
|
blank=True,
|
|
)
|
|
command = models.ForeignKey(
|
|
"users.Command", related_name="workers", on_delete=models.CASCADE
|
|
)
|
|
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)
|
|
wallet_public_key = models.CharField(max_length=96, unique=True)
|
|
money = models.IntegerField(default=0, validators=[MinValueValidator(0)])
|
|
|
|
telegram = models.CharField(max_length=100, unique=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
|
|
|
|
@property
|
|
def department(self):
|
|
return self.command.stream.department.name
|
|
|
|
@property
|
|
def clan_name(self):
|
|
return self.clan.name
|
|
|
|
class Meta:
|
|
ordering = ["-id"]
|
|
|
|
|
|
class Department(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Stream(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
department = models.ForeignKey(
|
|
Department, related_name="streams", on_delete=models.CASCADE
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Command(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
stream = models.ForeignKey(
|
|
Stream, related_name="commands", on_delete=models.CASCADE
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
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)
|