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.
78 lines
3.1 KiB
78 lines
3.1 KiB
8 months ago
|
from django.db import models
|
||
|
|
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
|
||
|
|
from django.contrib.auth.models import User
|
||
|
|
|
||
|
|
|
||
|
|
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)
|
||
|
|
return self.create_user(phone, password, **extra_fields)
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUser(AbstractBaseUser):
|
||
|
|
first_name = models.CharField(max_length=50)
|
||
|
|
last_name = models.CharField(max_length=50)
|
||
|
|
phone = models.CharField(max_length=15, unique=True)
|
||
|
|
phone2 = models.CharField(max_length=15, blank=True, null=True)
|
||
|
|
company_email = models.EmailField(null=True)
|
||
|
|
personal_email = models.EmailField(null=True)
|
||
|
|
address = models.TextField(null=True)
|
||
|
|
work_location = models.CharField(max_length=100, null=True)
|
||
|
|
picture = models.ImageField(upload_to='profile_pics/', null=True)
|
||
|
|
role = models.CharField(max_length=20)
|
||
|
|
user_id = models.CharField(max_length=10, unique=True)
|
||
|
|
is_active = models.BooleanField(default=True)
|
||
|
|
is_staff = models.BooleanField(default=False)
|
||
|
|
manager_assigned = models.CharField(max_length=100, null=True)
|
||
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
||
|
|
updated_at = models.DateTimeField(auto_now=True)
|
||
|
|
|
||
|
|
objects = CustomUserManager()
|
||
|
|
|
||
|
|
USERNAME_FIELD = 'phone'
|
||
|
|
|
||
|
|
|
||
|
|
class JobPosting(models.Model):
|
||
|
|
client = models.CharField(max_length=100)
|
||
|
|
spoc = models.CharField(max_length=100, blank=True, null=True)
|
||
|
|
start_date = models.DateField()
|
||
|
|
budget_max = models.DecimalField(max_digits=10, decimal_places=2, null=True)
|
||
|
|
budget_min = models.DecimalField(max_digits=10, decimal_places=2, null=True)
|
||
|
|
job_description = models.TextField()
|
||
|
|
special_instructions = models.CharField(max_length=200, blank=True, null=True)
|
||
|
|
job_id = models.CharField(max_length=20)
|
||
|
|
spoc_2 = models.CharField(max_length=100, blank=True, null=True)
|
||
|
|
close_date = models.DateField()
|
||
|
|
location = models.CharField(max_length=100)
|
||
|
|
no_of_posting = models.IntegerField()
|
||
|
|
job_type = models.CharField(max_length=20)
|
||
|
|
header = models.CharField(max_length=200)
|
||
|
|
experience_in_years = models.IntegerField()
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.job_id
|
||
|
|
|
||
|
|
|
||
|
|
class Message(models.Model):
|
||
|
|
sender = models.ForeignKey(User, related_name='sent_messages', on_delete=models.CASCADE)
|
||
|
|
recipient = models.ForeignKey(User, related_name='received_messages', on_delete=models.CASCADE)
|
||
|
|
subject = models.CharField(max_length=255)
|
||
|
|
body = models.TextField()
|
||
|
|
is_read = models.BooleanField(default=False)
|
||
|
|
sent_at = models.DateTimeField(auto_now_add=True)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return f'Message from {self.sender} to {self.recipient} - {self.subject}'
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
ordering = ['-sent_at']
|