Save import remark
This commit is contained in:
		
					parent
					
						
							
								a2635e6fb9
							
						
					
				
			
			
				commit
				
					
						b683a5ac44
					
				
			
		
					 1 changed files with 16 additions and 5 deletions
				
			
		|  | @ -1,3 +1,4 @@ | ||||||
|  | import datetime | ||||||
| import logging | import logging | ||||||
| 
 | 
 | ||||||
| from django.core.management.base import BaseCommand | from django.core.management.base import BaseCommand | ||||||
|  | @ -19,6 +20,10 @@ class Command(BaseCommand): | ||||||
|     def handle(self, *args, **options): |     def handle(self, *args, **options): | ||||||
|         try: |         try: | ||||||
|             for email in options['customer_email']: |             for email in options['customer_email']: | ||||||
|  |                 self.stdout.write( | ||||||
|  |                     self.style.SUCCESS( | ||||||
|  |                         "---------------------------------------------") | ||||||
|  |                 ) | ||||||
|                 stripe_utils = StripeUtils() |                 stripe_utils = StripeUtils() | ||||||
|                 user = CustomUser.objects.get(email=email) |                 user = CustomUser.objects.get(email=email) | ||||||
|                 if hasattr(user, 'stripecustomer'): |                 if hasattr(user, 'stripecustomer'): | ||||||
|  | @ -39,7 +44,9 @@ class Command(BaseCommand): | ||||||
|                     ) |                     ) | ||||||
|                     if all_invoices_response['error'] is not None: |                     if all_invoices_response['error'] is not None: | ||||||
|                         self.stdout.write(self.style.ERROR(all_invoices_response['error'])) |                         self.stdout.write(self.style.ERROR(all_invoices_response['error'])) | ||||||
|                         exit(1) |                         user.import_stripe_bill_remark += "{}: {},".format(datetime.datetime.now(), all_invoices_response['error']) | ||||||
|  |                         user.save() | ||||||
|  |                         continue | ||||||
|                     all_invoices = all_invoices_response['response_object'] |                     all_invoices = all_invoices_response['response_object'] | ||||||
|                     self.stdout.write(self.style.SUCCESS("Obtained {} invoices".format(len(all_invoices) if all_invoices is not None else 0))) |                     self.stdout.write(self.style.SUCCESS("Obtained {} invoices".format(len(all_invoices) if all_invoices is not None else 0))) | ||||||
|                     num_invoice_created = 0 |                     num_invoice_created = 0 | ||||||
|  | @ -54,6 +61,10 @@ class Command(BaseCommand): | ||||||
|                             if MonthlyHostingBill.create(invoice) is not None: |                             if MonthlyHostingBill.create(invoice) is not None: | ||||||
|                                 num_invoice_created += 1 |                                 num_invoice_created += 1 | ||||||
|                             else: |                             else: | ||||||
|  |                                 user.import_stripe_bill_remark += "{}: Import failed - {},".format( | ||||||
|  |                                     datetime.datetime.now(), | ||||||
|  |                                     invoice['invoice_id']) | ||||||
|  |                                 user.save() | ||||||
|                                 logger.error("Did not import invoice for %s" |                                 logger.error("Did not import invoice for %s" | ||||||
|                                              "" % str(invoice)) |                                              "" % str(invoice)) | ||||||
|                     self.stdout.write( |                     self.stdout.write( | ||||||
|  | @ -62,9 +73,9 @@ class Command(BaseCommand): | ||||||
|                 else: |                 else: | ||||||
|                     self.stdout.write(self.style.SUCCESS( |                     self.stdout.write(self.style.SUCCESS( | ||||||
|                         'Customer email %s does not have a stripe customer.' % email)) |                         'Customer email %s does not have a stripe customer.' % email)) | ||||||
|                 self.stdout.write( |                     user.import_stripe_bill_remark += "{}: No stripecustomer,".format( | ||||||
|                     self.style.SUCCESS( |                         datetime.datetime.now() | ||||||
|                         "---------------------------------------------") |  | ||||||
|                     ) |                     ) | ||||||
|  |                     user.save() | ||||||
|         except Exception as e: |         except Exception as e: | ||||||
|             print(" *** Error occurred. Details {}".format(str(e))) |             print(" *** Error occurred. Details {}".format(str(e))) | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue