172 lines
4.6 KiB
Python
172 lines
4.6 KiB
Python
from django.contrib.auth import get_user_model
|
||
from django.db import models
|
||
from django.utils.html import format_html
|
||
from django.core.validators import RegexValidator
|
||
from simple_history.models import HistoricalRecords
|
||
from common.models import TagModel, BaseModel
|
||
from common.consts import UserRoles
|
||
from locations.models import Affairs, Broker, County
|
||
|
||
# Create your models here.
|
||
|
||
class Role(TagModel):
|
||
|
||
class Meta:
|
||
verbose_name = "نقش"
|
||
verbose_name_plural = "نقش‌ها"
|
||
|
||
|
||
class Profile(BaseModel):
|
||
user = models.OneToOneField(
|
||
get_user_model(),
|
||
null=True,
|
||
on_delete=models.CASCADE,
|
||
verbose_name="کاربر",
|
||
related_name="profile"
|
||
)
|
||
national_code = models.CharField(
|
||
max_length=10,
|
||
null=True,
|
||
verbose_name="کد ملی",
|
||
blank=True,
|
||
validators=[
|
||
RegexValidator(
|
||
regex=r'^\d+$',
|
||
message='کد ملی باید فقط شامل اعداد باشد.',
|
||
code='invalid_national_code'
|
||
)
|
||
]
|
||
)
|
||
address = models.TextField(
|
||
null=True,
|
||
verbose_name="آدرس",
|
||
blank=True
|
||
)
|
||
card_number = models.CharField(
|
||
max_length=16,
|
||
null=True,
|
||
verbose_name="شماره کارت",
|
||
blank=True,
|
||
validators=[
|
||
RegexValidator(
|
||
regex=r'^\d+$',
|
||
message='شماره کارت باید فقط شامل اعداد باشد.',
|
||
code='invalid_card_number'
|
||
)
|
||
]
|
||
)
|
||
account_number = models.CharField(
|
||
max_length=20,
|
||
null=True,
|
||
verbose_name="شماره حساب",
|
||
blank=True,
|
||
validators=[
|
||
RegexValidator(
|
||
regex=r'^\d+$',
|
||
message='شماره حساب باید فقط شامل اعداد باشد.',
|
||
code='invalid_account_number'
|
||
)
|
||
]
|
||
)
|
||
phone_number_1 = models.CharField(
|
||
max_length=11,
|
||
null=True,
|
||
verbose_name="شماره تماس ۱",
|
||
blank=True
|
||
)
|
||
phone_number_2 = models.CharField(
|
||
max_length=11,
|
||
null=True,
|
||
verbose_name="شماره تماس ۲",
|
||
blank=True
|
||
)
|
||
|
||
pic = models.ImageField(
|
||
upload_to="profile_images",
|
||
verbose_name="تصویر",
|
||
default="../static/sample_images/profile.jpg"
|
||
)
|
||
roles = models.ManyToManyField(
|
||
Role,
|
||
verbose_name="نقش‌ها",
|
||
blank=True,
|
||
related_name="profiles"
|
||
)
|
||
is_completed = models.BooleanField(
|
||
default=False,
|
||
verbose_name="پروفایل تکمیل شده"
|
||
)
|
||
owner = models.ForeignKey(
|
||
get_user_model(),
|
||
null=True,
|
||
on_delete=models.SET_NULL,
|
||
verbose_name="ایجاد کننده",
|
||
blank=True,
|
||
related_name="owned_profiles"
|
||
)
|
||
affairs = models.ForeignKey(
|
||
Affairs,
|
||
on_delete=models.SET_NULL,
|
||
verbose_name="امور",
|
||
null=True,
|
||
blank=True
|
||
)
|
||
county = models.ForeignKey(
|
||
County,
|
||
on_delete=models.SET_NULL,
|
||
verbose_name="شهرستان",
|
||
null=True,
|
||
blank=True
|
||
)
|
||
broker = models.ForeignKey(
|
||
Broker,
|
||
on_delete=models.SET_NULL,
|
||
verbose_name="کارگزار",
|
||
null=True,
|
||
blank=True
|
||
)
|
||
|
||
history = HistoricalRecords()
|
||
|
||
class Meta:
|
||
verbose_name = "پروفایل"
|
||
verbose_name_plural = "پروفایل‌ها"
|
||
|
||
def __str__(self):
|
||
return str(self.user)
|
||
|
||
|
||
def first_name(self):
|
||
return self.user.first_name
|
||
|
||
first_name.short_description = "نام"
|
||
|
||
def last_name(self):
|
||
return self.user.last_name
|
||
|
||
last_name.short_description = "نام خانوادگی"
|
||
|
||
def fullname(self):
|
||
return self.user.get_full_name()
|
||
|
||
fullname.short_description = "نام و نام خانوادگی"
|
||
|
||
def roles_str(self):
|
||
return ', '.join([role.name for role in self.roles.all()])
|
||
|
||
roles_str.short_description = "نقش‌ها"
|
||
|
||
def has_role(self, role: UserRoles):
|
||
return self.roles.filter(slug=role.value).exists()
|
||
|
||
def has_any_of(self, roles: list[UserRoles]):
|
||
allowed_roles = [role.value for role in roles]
|
||
return len(self.roles.filter(slug__in=allowed_roles)) > 0
|
||
|
||
def has_none_of(self, roles: list[UserRoles]):
|
||
return not self.has_any_of(roles)
|
||
|
||
def pic_tag(self):
|
||
return format_html(f"<img style='width:30px;' src='{self.pic.url}'>")
|
||
|
||
pic_tag.short_description = "تصویر"
|