Add management command
This commit is contained in:
parent
0e84081880
commit
8dc00c9dd9
1 changed files with 28 additions and 0 deletions
28
hosting/management/commands/fetch_stripe_bills.py
Normal file
28
hosting/management/commands/fetch_stripe_bills.py
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
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)))
|
Loading…
Reference in a new issue