from django.core.management.base import BaseCommand
from accounts.models import CustomUser, FriendCode

class Command(BaseCommand):
    help = "Seed default friend codes for TestUsers after friend codes have been loaded."

    def handle(self, *args, **options):
        users_updated = 0
        for user in CustomUser.objects.all():
            # Automatically select the earliest friend code added for the user:
            default_code = FriendCode.objects.filter(user=user).order_by('created_at').first()
            if default_code:
                user.default_friend_code = default_code
                user.save(update_fields=["default_friend_code"])
                self.stdout.write(f"Set default friend code for user {user.username} to {default_code.friend_code}.")
                users_updated += 1
            else:
                self.stdout.write(f"No friend code found for user {user.username}.")
        self.stdout.write(self.style.SUCCESS(f"Seeded default friend codes for {users_updated} user(s)."))