Handle invoice for no stripe error

This commit is contained in:
PCoder 2020-02-25 11:13:26 +05:30
parent d0455fbafb
commit e25ed77538

View file

@ -1297,9 +1297,12 @@ class InvoiceListView(LoginRequiredMixin, TemplateView):
except CustomUser.DoesNotExist as dne: except CustomUser.DoesNotExist as dne:
logger.debug("User does not exist") logger.debug("User does not exist")
cu = self.request.user cu = self.request.user
invs = stripe.Invoice.list(customer=cu.stripecustomer.stripe_id, if cu.stripecustomer:
count=100) invs = stripe.Invoice.list(customer=cu.stripecustomer.stripe_id,
paginator = Paginator(invs.data, 10) count=100)
else:
invs = []
paginator = Paginator(invs.data if len(invs) > 0 else [], 10)
try: try:
invs_page = paginator.page(page) invs_page = paginator.page(page)
except PageNotAnInteger: except PageNotAnInteger:
@ -1308,11 +1311,14 @@ class InvoiceListView(LoginRequiredMixin, TemplateView):
invs_page = paginator.page(paginator.num_pages) invs_page = paginator.page(paginator.num_pages)
else: else:
try: try:
invs = stripe.Invoice.list( if self.request.user.stripecustomer:
customer=self.request.user.stripecustomer.stripe_id, invs = stripe.Invoice.list(
count=100 customer=self.request.user.stripecustomer.stripe_id,
) count=100
paginator = Paginator(invs.data, 10) )
else:
invs = []
paginator = Paginator(invs.data if len(invs) > 0 else [], 10)
try: try:
invs_page = paginator.page(page) invs_page = paginator.page(page)
except PageNotAnInteger: except PageNotAnInteger: