You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
5.5 KiB
124 lines
5.5 KiB
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
|
|
from django.db import models
|
|
|
|
|
|
class CustomUserManager(BaseUserManager):
|
|
def create_user(self, phone, password=None, **extra_fields):
|
|
if not phone:
|
|
raise ValueError('The Phone number must be set')
|
|
user = self.model(phone=phone, **extra_fields)
|
|
user.set_password(password)
|
|
user.save(using=self._db)
|
|
return user
|
|
|
|
def create_superuser(self, phone, password=None, **extra_fields):
|
|
extra_fields.setdefault('is_staff', True)
|
|
extra_fields.setdefault('is_superuser', True)
|
|
|
|
if extra_fields.get('is_staff') is not True:
|
|
raise ValueError('Superuser must have is_staff=True.')
|
|
if extra_fields.get('is_superuser') is not True:
|
|
raise ValueError('Superuser must have is_superuser=True.')
|
|
|
|
return self.create_user(phone, password, **extra_fields)
|
|
|
|
|
|
class skyonnAdmin(AbstractBaseUser):
|
|
ADMIN_ID_PREFIX = 'SKYA' # Prefix for AdminId
|
|
|
|
first_name = models.CharField(max_length=50)
|
|
last_name = models.CharField(max_length=50)
|
|
phone = models.CharField(max_length=15, unique=True)
|
|
phone_number_2 = models.CharField(max_length=15, blank=True, null=True)
|
|
company_email = models.EmailField(null=True) # Additional email field
|
|
personal_email = models.EmailField(null=True) # Additional email field
|
|
address = models.TextField(null=True) # Additional address field
|
|
work_location = models.CharField(max_length=100, null=True) # Additional work location field
|
|
picture = models.ImageField(upload_to='profile_pics/', null=True) # Additional picture field
|
|
role = models.CharField(max_length=20) # Additional role field
|
|
admin_id = models.CharField(max_length=10, unique=True) # AdminId field
|
|
|
|
is_active = models.BooleanField(default=True)
|
|
is_staff = models.BooleanField(default=False)
|
|
|
|
objects = CustomUserManager()
|
|
|
|
USERNAME_FIELD = 'phone'
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.admin_id:
|
|
last_admin = skyonnAdmin.objects.order_by('-id').first()
|
|
if last_admin:
|
|
last_id = last_admin.admin_id[4:] # Extract the serial number part
|
|
new_id = int(last_id) + 1
|
|
else:
|
|
new_id = 1
|
|
self.admin_id = f'{self.ADMIN_ID_PREFIX}{new_id:03d}'
|
|
super().save(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return self.phone
|
|
|
|
|
|
class Contact(models.Model):
|
|
client_id = models.CharField(max_length=20, null=True)
|
|
first_name = models.CharField(max_length=50)
|
|
last_name = models.CharField(max_length=50)
|
|
phone_no1 = models.CharField(max_length=20)
|
|
phone_no2 = models.CharField(max_length=20, blank=True, null=True)
|
|
company_email = models.EmailField()
|
|
designation = models.CharField(max_length=100)
|
|
department = models.CharField(max_length=100)
|
|
|
|
def __str__(self):
|
|
return f"{self.first_name} {self.last_name} - {self.company_email}"
|
|
|
|
|
|
class ClientDetails(models.Model):
|
|
parent_company = models.CharField(max_length=100)
|
|
location = models.CharField(max_length=100)
|
|
department = models.CharField(max_length=100)
|
|
gst_no = models.CharField(max_length=50)
|
|
address = models.CharField(max_length=200)
|
|
client_id = models.CharField(max_length=20, null=True)
|
|
admin_first_name = models.CharField(max_length=50)
|
|
admin_last_name = models.CharField(max_length=50)
|
|
admin_phone_no1 = models.CharField(max_length=20)
|
|
admin_phone_no2 = models.CharField(max_length=20, blank=True, null=True)
|
|
admin_company_email = models.EmailField()
|
|
admin_designation = models.CharField(max_length=100, null=True)
|
|
admin_department = models.CharField(max_length=100, null=True)
|
|
admin_location = models.CharField(max_length=100, null=True)
|
|
admin_contacts = models.ManyToManyField(Contact, related_name='admin_contacts')
|
|
hr_contacts = models.ManyToManyField(Contact, related_name='hr_contacts')
|
|
accounts_contacts = models.ManyToManyField(Contact, related_name='accounts_contacts')
|
|
recruiter_contacts = models.ManyToManyField(Contact, related_name='recruiter_contacts')
|
|
|
|
def __str__(self):
|
|
return f"{self.parent_company} - {self.location}"
|
|
|
|
|
|
class SubcompanyDetails(models.Model):
|
|
parent_company = models.CharField(max_length=100)
|
|
sub_company = models.CharField(max_length=100)
|
|
location = models.CharField(max_length=100)
|
|
department = models.CharField(max_length=100)
|
|
gst_no = models.CharField(max_length=50)
|
|
address = models.CharField(max_length=200)
|
|
client_id = models.CharField(max_length=20, null=True)
|
|
admin_first_name = models.CharField(max_length=50)
|
|
admin_last_name = models.CharField(max_length=50)
|
|
admin_phone_no1 = models.CharField(max_length=20)
|
|
admin_phone_no2 = models.CharField(max_length=20, blank=True, null=True)
|
|
admin_company_email = models.EmailField()
|
|
admin_designation = models.CharField(max_length=100, null=True)
|
|
admin_department = models.CharField(max_length=100, null=True)
|
|
admin_location = models.CharField(max_length=100, null=True)
|
|
admin_contacts = models.ManyToManyField(Contact, related_name='subcompany_admin_contacts')
|
|
hr_contacts = models.ManyToManyField(Contact, related_name='subcompany_hr_contacts')
|
|
accounts_contacts = models.ManyToManyField(Contact, related_name='subcompany_accounts_contacts')
|
|
recruiter_contacts = models.ManyToManyField(Contact, related_name='subcompany_recruiter_contacts')
|
|
|
|
def __str__(self):
|
|
return f"{self.parent_company} - {self.location}"
|