dynamicweb/hosting/management/commands/fetch_stripe_bills.py

29 lines
1.1 KiB
Python
Raw Normal View History

2019-04-02 07:18:46 +00:00
from django.core.management.base import BaseCommand
from hosting.models import UserCardDetail
from membership.models import CustomUser
from utils.stripe_utils import StripeUtils
class Command(BaseCommand):
help = '''Fetches invoices from Stripe and creates bills for a given
customer in the MonthlyHostingBill model'''
def add_arguments(self, parser):
parser.add_argument('customer_email', nargs='+', type=str)
def handle(self, *args, **options):
try:
for email in options['customer_email']:
stripe_utils = StripeUtils()
user = CustomUser.objects.get(email=email)
if hasattr(user, 'stripecustomer'):
self.stdout.write(self.style.SUCCESS('Found %s. Fetching bills for him.' % email))
stripe_utils.get_all_invoices(
user.stripecustomer.stripe_id
)
else:
self.stdout.write(self.style.SUCCESS('Customer email %s does not have a stripe customer.' % email))
except Exception as e:
print(" *** Error occurred. Details {}".format(str(e)))