Coverage for invoices/resources.py: 30%

1from import_export import resources, fields 

2from payments.models import ManualPayment, PaymentIntent 

3from .models import Invoice 

4 

5 

6class InvoiceResource(resources.ModelResource): 

7 invoice_date = fields.Field() 

8 membership_type = fields.Field() 

9 segment = fields.Field() 

10 amount = fields.Field() 

11 payment_reference = fields.Field() 

12 segment_comment = fields.Field() 

13 product_comment = fields.Field() 

14 product_variation_comment = fields.Field() 

15 

16 class Meta: 

17 model = Invoice 

18 fields = [ 

19 "invoice_date", 

20 "membership", 

21 "membership_type", 

22 "segment", 

23 "use", 

24 "sat_uuid", 

25 "folio_number", 

26 "amount", 

27 "status", 

28 "cancellation_status", 

29 "cancellation_motive", 

30 "payment_reference", 

31 ] 

32 

33 def dehydrate_invoice_date(self, instance): 

34 return instance.created_at.date() 

35 

36 def dehydrate_membership(self, instance): 

37 return instance.membership.name 

38 

39 def dehydrate_membership_type(self, instance): 

40 return instance.membership.type_of.name 

41 

42 def dehydrate_segment(self, instance): 

43 return instance.membership.segment 

44 

45 def dehydrate_amount(self, instance): 

46 return instance.total_amount 

47 

48 def dehydrate_status(self, instance): 

49 return instance.get_status_display() 

50 

51 def dehydrate_cancellation_status(self, instance): 

52 return instance.get_cancellation_status_display() 

53 

54 def dehydrate_cancellation_motive(self, instance): 

55 return instance.get_cancellation_motive_display() 

56 

57 def dehydrate_payment_reference(self, instance): 

58 product_charges = instance.product_charges.all() 

59 

60 payment_references = [] 

61 

62 for product_charge in product_charges: 

63 payment_allocations = product_charge.payment_allocations.all() 

64 for payment_allocation in payment_allocations: 

65 payment = payment_allocation.payment 

66 if isinstance(payment, ManualPayment): 

67 payment_references.append(payment.reference) 

68 elif isinstance(payment, PaymentIntent): 

69 payment_references.append(payment.stripe_pi_id) 

70 

71 return ", ".join(payment_reference for payment_reference in payment_references) 

72 

73 def dehydrate_segment_comment(self, instance): 

74 product_charges = instance.product_charges.all() 

75 

76 segment_comments = [] 

77 

78 for product_charge in product_charges: 

79 product = product_charge.product 

80 for segment in product.segments.all(): 

81 if segment.comment: 

82 segment_comments.append(segment.comment) 

83 

84 return ", ".join(segment_comment for segment_comment in segment_comments) 

85 

86 def dehydrate_product_comment(self, instance): 

87 product_charges = instance.product_charges.all() 

88 

89 product_comments = [] 

90 

91 for product_charge in product_charges: 

92 product = product_charge.product 

93 if product.comment: 

94 product_comments.append(product.comment) 

95 

96 return ", ".join(product_comment for product_comment in product_comments) 

97 

98 def dehydrate_product_variation_comment(self, instance): 

99 product_charges = instance.product_charges.all() 

100 

101 product_variation_comments = [] 

102 

103 for product_charge in product_charges: 

104 product_purchase = product_charge.purchase 

105 for variation in product_purchase.price_breakdown.get("variations", []): 

106 if variation.get("comment", None): 

107 product_variation_comments.append(variation.get("comment")) 

108 

109 return ", ".join(product_variation_comment for product_variation_comment in product_variation_comments)