61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
from django.contrib.auth.models import AbstractUser
|
|
from django.db import models
|
|
|
|
from accounts.manager import UserManager
|
|
from partner.models import Partner
|
|
ROLE_CHOICES = (
|
|
('admin', 'Admin'),
|
|
('manager', 'Manager'),
|
|
('staff', 'Staff'),
|
|
('customer', 'Customer'),
|
|
('partner', 'Partner'),
|
|
('partner_manager', 'Partner Manager'),
|
|
('partner_staff', 'Partner Staff'),
|
|
('partner_customer', 'Partner Customer'),
|
|
)
|
|
|
|
class User(AbstractUser):
|
|
phone_number = models.CharField(max_length=15, blank=True, null=True)
|
|
role = models.CharField(max_length=20, choices=ROLE_CHOICES, default='Staff')
|
|
|
|
partner = models.ForeignKey(Partner, on_delete=models.CASCADE, blank=True, null=True)
|
|
|
|
is_staff = models.BooleanField(default=False)
|
|
is_customer = models.BooleanField(default=False)
|
|
is_user = models.BooleanField(default=False)
|
|
|
|
# Location fields
|
|
pincode = models.CharField(max_length=10, blank=True, null=True)
|
|
district = models.CharField(max_length=100, blank=True, null=True)
|
|
state = models.CharField(max_length=100, blank=True, null=True)
|
|
country = models.CharField(max_length=100, blank=True, null=True)
|
|
place = models.CharField(max_length=200, blank=True, null=True)
|
|
|
|
# Location fields
|
|
latitude = models.DecimalField(max_digits=9, decimal_places=6, blank=True, null=True)
|
|
longitude = models.DecimalField(max_digits=9, decimal_places=6, blank=True, null=True)
|
|
|
|
profile_picture = models.ImageField(upload_to='profile_pictures/', blank=True, null=True, default='default.png')
|
|
|
|
allowed_modules = models.TextField(
|
|
blank=True, null=True,
|
|
help_text="Comma-separated module slugs this user can access"
|
|
)
|
|
|
|
ALL_MODULES = ["dashboard", "partners", "events", "ad-control", "users", "reviews", "contributions", "financials", "settings"]
|
|
|
|
def get_allowed_modules(self):
|
|
ALL = ["dashboard", "partners", "events", "ad-control", "users", "reviews", "contributions", "financials", "settings"]
|
|
if self.is_superuser or self.role == "admin":
|
|
return ALL
|
|
if self.allowed_modules:
|
|
return [m.strip() for m in self.allowed_modules.split(",") if m.strip()]
|
|
if self.role == "manager":
|
|
return ALL
|
|
return []
|
|
|
|
objects = UserManager()
|
|
|
|
def __str__(self):
|
|
return self.username
|