regluit/payment/management/commands/grant_user_credit.py

16 lines
574 B
Python
Raw Permalink Normal View History

from django.core.management.base import BaseCommand
from django.contrib.auth.models import User
class Command(BaseCommand):
help = "grant credit to a user"
args = "<username> <amount> <action>"
def handle(self, username, amount, action="credit", *args, **kwargs):
if action=="debit":
amount=-int(amount)
else:
amount= int(amount)
user = User.objects.get(username=username)
user.credit.add_to_balance(amount)
print "%s now has a balance of %s gift credits" % (username, user.credit.balance)