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}"