arminta-web/skyonnweb/skyonnadmin/models.py

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}"