The new updates of partners and user

Made-with: Cursor

accounts/models.py

@@ -2,18 +2,24 @@ 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)