40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import secrets
|
|
import string
|
|
|
|
from django.db import models
|
|
|
|
|
|
def generate_referral_code() -> str:
|
|
"""کد ۸ کاراکتری حروفبزرگ/عدد، مثل SARA10 در سند نمونه."""
|
|
alphabet = string.ascii_uppercase + string.digits
|
|
return ''.join(secrets.choice(alphabet) for _ in range(8))
|
|
|
|
|
|
class AffiliateAccount(models.Model):
|
|
user_id = models.IntegerField(unique=True, db_index=True) # ارجاع سبک به accounts.User
|
|
referral_code = models.CharField(max_length=20, unique=True, default=generate_referral_code)
|
|
commission_rate = models.FloatField(default=0.1)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"affiliate-user{self.user_id}-{self.referral_code}"
|
|
|
|
|
|
class Conversion(models.Model):
|
|
STATUS_CHOICES = [('pending', 'در انتظار'), ('paid', 'پرداختشده')]
|
|
|
|
affiliate_account = models.ForeignKey(
|
|
AffiliateAccount, on_delete=models.CASCADE, related_name='conversions'
|
|
)
|
|
referred_user_id = models.IntegerField(unique=True, db_index=True) # ارجاع سبک به accounts.User
|
|
order_id = models.IntegerField(null=True, blank=True) # ارجاع سبک به payment.Order
|
|
status = models.CharField(max_length=10, choices=STATUS_CHOICES, default='pending')
|
|
commission_amount = models.PositiveIntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
ordering = ['-created_at']
|
|
|
|
def __str__(self):
|
|
return f"conversion-{self.pk}-referred{self.referred_user_id}"
|