Part 4: Weekly Bonus & Rank System

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

class WeeklyBonus(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    total_direct_investment = models.DecimalField(max_digits=12, decimal_places=2)
    bonus_amount = models.DecimalField(max_digits=10, decimal_places=2)
    week_start = models.DateField()
    week_end = models.DateField()
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"{self.user.username} - ${self.bonus_amount} for week {self.week_start}"

class RankAchievement(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    rank_name = models.CharField(max_length=50)
    team_a_volume = models.DecimalField(max_digits=12, decimal_places=2)
    team_b_volume = models.DecimalField(max_digits=12, decimal_places=2)
    all_team_volume = models.DecimalField(max_digits=12, decimal_places=2)
    gift_amount = models.DecimalField(max_digits=10, decimal_places=2)
    achieved_at = models.DateField(auto_now_add=True)

    def __str__(self):
        return f"{self.user.username} - {self.rank_name}"

from accounts.models import Profile
from investment.models import UserInvestment
from .models import WeeklyBonus
from django.utils import timezone
from datetime import timedelta

WEEKLY_BONUSES = [
    (100000, 250),
    (50000, 100),
    (20000, 50),
    (10000, 20),
    (5000, 10),
    (2000, 5),
]

def process_weekly_bonuses():
    today = timezone.now().date()
    start_of_week = today - timedelta(days=today.weekday())
    end_of_week = start_of_week + timedelta(days=6)

    all_profiles = Profile.objects.all()
    for profile in all_profiles:
        user = profile.user
        direct_referrals = user.referrals.all()
        total_direct = sum([
            inv.amount for ref in direct_referrals
            for inv in ref.investment_set.filter(start_date__range=(start_of_week, end_of_week))
        ])
        bonus = 0
        for threshold, reward in WEEKLY_BONUSES:
            if total_direct >= threshold:
                bonus = reward
                break

        if bonus > 0:
            WeeklyBonus.objects.create(
                user=user,
                total_direct_investment=total_direct,
                bonus_amount=bonus,
                week_start=start_of_week,
                week_end=end_of_week
            )

from django.contrib import admin
from unfold.admin import ModelAdmin
from .models import WeeklyBonus, RankAchievement

@admin.register(WeeklyBonus)
class WeeklyBonusAdmin(ModelAdmin):
    list_display = ['user', 'total_direct_investment', 'bonus_amount', 'week_start', 'week_end']
    list_filter = ['week_start', 'bonus_amount']


@admin.register(RankAchievement)
class RankAchievementAdmin(ModelAdmin):
    list_display = ['user', 'rank_name', 'gift_amount', 'achieved_at']
    list_filter = ['rank_name', 'achieved_at']

python manage.py makemigrations bonus
python manage.py migrate