models.py

from django.db import models
from django.contrib.auth.models import AbstractUser #right-click to go to user definitions
from django.db.models.signals import post_save,pre_save
# Create your models here

class User(AbstractUser):
	#cellphone_number=models.Ch
	is_organiser=models.BooleanField(default=True)
	is_agent=models.BooleanField(default=False)





class UserProfile(models.Model):
	user=models.OneToOneField(User,on_delete=models.CASCADE)

	def __str__(self):
		return self.user.username


class Agent(models.Model):
	user=models.OneToOneField(User, on_delete=models.CASCADE)
	organisation=models.ForeignKey(UserProfile,on_delete=models.CASCADE)
	def __str__(self):
		return self.user.email



class Lead(models.Model):
	def __str__(self):
		return f"{self.first_name} {self.last_name}"
	
	SOURCE_CHOICES=(
		('YT','Youtube'),
		('Google','Google'),	
		('Newsletter','Newsletter')
		)
	first_name=models.CharField(max_length=20)
	last_name=models.CharField(max_length=20)
	age=models.IntegerField(default=0)
	agent=models.ForeignKey("Agent",null=True,blank=True, on_delete=models.SET_NULL)
	organisation=models.ForeignKey(UserProfile,on_delete=models.CASCADE)
    
    
    
def post_user_created_signal(sender,instance,created,**kwargs):
	if created:
		UserProfile.objects.create(user=instance)
	


post_save.connect(post_user_created_signal,sender=User)

'''
	phoned=models.BooleanField(default=False)
	source=models.CharField(choices=SOURCE_CHOICES)

	profile_picture=models.ImageField(blank=True,null=True)
	special_files=models.FileField(blank=True,null=True)
	'''