Part 3: Referral & Affiliate System

from django.db import models
from django.contrib.auth.models import User

class ReferralIncome(models.Model):
    referrer = models.ForeignKey(User, on_delete=models.CASCADE, related_name='referral_earnings')
    referred_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='referred_investments')
    level = models.PositiveIntegerField()
    amount = models.DecimalField(max_digits=10, decimal_places=2)
    created_at = models.DateField(auto_now_add=True)

    def __str__(self):
        return f"{self.referrer.username} earned ${self.amount} from {self.referred_user.username} (L{self.level})"

from decimal import Decimal
from django.contrib.auth.models import User
from accounts.models import Profile
from .models import ReferralIncome

REFERRAL_PERCENTAGES = {
    1: 10,
    2: 5,
    3: 3,
    4: 2,
    5: 1,
    6: 0.5,
    7: 0.5,
    8: 0.5,
    9: 0.5,
    10: 0.5,
}

def distribute_referral_income(base_user, base_amount):
    current_user = base_user
    for level in range(1, 11):
        try:
            profile = current_user.profile
            sponsor = profile.sponsor
        except:
            break

        if not sponsor:
            break

        percent = Decimal(REFERRAL_PERCENTAGES.get(level, 0))
        income = (base_amount * percent) / Decimal(100)

        ReferralIncome.objects.create(
            referrer=sponsor,
            referred_user=base_user,
            level=level,
            amount=income
        )

        current_user = sponsor

from referral.utils import distribute_referral_income

def generate_daily_roi():
    today = timezone.now().date()
    investments = UserInvestment.objects.filter(active=True)

    for inv in investments:
        roi_amount = (inv.amount * inv.package.roi_percent) / Decimal(100)

        ROITransaction.objects.create(
            user=inv.user,
            investment=inv,
            amount=roi_amount,
            created_at=today
        )

        # 💸 Trigger referral income
        distribute_referral_income(inv.user, roi_amount)

from django.contrib import admin
from unfold.admin import ModelAdmin
from .models import ReferralIncome

@admin.register(ReferralIncome)
class ReferralIncomeAdmin(ModelAdmin):
    list_display = ['referrer', 'referred_user', 'level', 'amount', 'created_at']
    list_filter = ['level', 'created_at']
    search_fields = ['referrer__username', 'referred_user__username']

python manage.py makemigrations referral
python manage.py migrate