from django.core.management.base import BaseCommand

from labs.models import User
from labs.billing.user import update_user_credit_summary


class Command(BaseCommand):
    def add_arguments(self, parser):
        parser.add_argument('--user-id', type=int, required=True, dest='user_id')

    def handle(self, *args, **options):
        user = User.objects.get(id=options['user_id'])
        update_user_credit_summary(user)