chore: more formatting

Author: calvin-codecov

services/billing.py

@@ -167,23 +167,40 @@ def delete_subscription(self, owner: Owner):
 
         # we give an auto-refund grace period of 24 hours for a monthly subscription or 72 hours for a yearly subscription
         # current_subscription_timestamp = subscription["current_period_start"]
-        current_subscription_datetime = datetime.fromtimestamp(subscription["current_period_start"])
+        current_subscription_datetime = datetime.fromtimestamp(
+            subscription["current_period_start"]
+        )
         differenceFromNow = datetime.now() - current_subscription_datetime
 
-        subscription_plan_interval = subscription.plan.interval if subscription.plan is not None else None
-        within_refund_grace_period = (subscription_plan_interval == "month" and differenceFromNow.days < 1) or (subscription_plan_interval == "year" and differenceFromNow.days < 3)
+        subscription_plan_interval = (
+            subscription.plan.interval if subscription.plan is not None else None
+        )
+        within_refund_grace_period = (
+            subscription_plan_interval == "month" and differenceFromNow.days < 1
+        ) or (subscription_plan_interval == "year" and differenceFromNow.days < 3)
         if within_refund_grace_period:
             stripe.Subscription.cancel(owner.stripe_subscription_id)
 
-            invoices_list = stripe.Invoice.list(subscription=owner.stripe_subscription_id, status="paid")
+            invoices_list = stripe.Invoice.list(
+                subscription=owner.stripe_subscription_id, status="paid"
+            )
             created_refund = False
             # there could be multiple invoices that need to be refunded such as if the user increased seats within the grace period
             for invoice in invoices_list["data"]:
-                start_of_last_period = current_subscription_datetime - relativedelta(months=1) if subscription_plan_interval == "month" else current_subscription_datetime - relativedelta(years=1)
+                start_of_last_period = (
+                    current_subscription_datetime - relativedelta(months=1)
+                    if subscription_plan_interval == "month"
+                    else current_subscription_datetime - relativedelta(years=1)
+                )
 
                 # refund if the invoice has a charge, it has been fully paid, the creation time was before the start of the current subscription's start and the creation time was after the start of the last period
                 invoice_created_datetime = datetime.fromtimestamp(invoice["created"])
-                if invoice["charge"] is not None and invoice["amount_remaining"] == 0 and invoice_created_datetime < current_subscription_datetime and invoice_created_datetime >= start_of_last_period:
+                if (
+                    invoice["charge"] is not None
+                    and invoice["amount_remaining"] == 0
+                    and invoice_created_datetime < current_subscription_datetime
+                    and invoice_created_datetime >= start_of_last_period
+                ):
                     stripe.Refund.create(invoice["charge"])
                     created_refund = True
             if created_refund == True:

services/tests/test_billing.py

@@ -129,23 +129,34 @@
     }
 ]
 
+
 class MockSubscriptionPlan(object):
     def __init__(self, params):
         self.id = params["new_plan"]
-        self.interval = 'year'
+        self.interval = "year"
+
 
 class MockSubscription(object):
     def __init__(self, subscription_params):
         self.schedule = subscription_params["schedule_id"]
         self.current_period_start = subscription_params["start_date"]
         self.current_period_end = subscription_params["end_date"]
-        self.plan = MockSubscriptionPlan(subscription_params["plan"]) if subscription_params.get("plan") is not None else None
+        self.plan = (
+            MockSubscriptionPlan(subscription_params["plan"])
+            if subscription_params.get("plan") is not None
+            else None
+        )
         self.items = {
             "data": [
                 {
                     "quantity": subscription_params["quantity"],
                     "id": subscription_params["id"],
-                    "plan": {"id": subscription_params["name"], "interval": subscription_params.get("plan", {}).get("interval", "month")},
+                    "plan": {
+                        "id": subscription_params["name"],
+                        "interval": subscription_params.get("plan", {}).get(
+                            "interval", "month"
+                        ),
+                    },
                 }
             ]
         }
@@ -163,6 +174,7 @@ def __init__(self, subscription_params):
     def __getitem__(self, key):
         return getattr(self, key)
 
+
 class StripeServiceTests(TestCase):
     def setUp(self):
         self.user = OwnerFactory()
@@ -317,7 +329,11 @@ def test_delete_subscription_without_schedule_modifies_subscription_to_delete_at
     @patch("services.billing.stripe.Subscription.retrieve")
     @patch("services.billing.stripe.SubscriptionSchedule.release")
     def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscription_at_end_of_billing_cycle_if_valid_plan(
-        self, schedule_release_mock, retrieve_subscription_mock, modify_mock, create_refund_mock
+        self,
+        schedule_release_mock,
+        retrieve_subscription_mock,
+        modify_mock,
+        create_refund_mock,
     ):
         plan = PlanName.CODECOV_PRO_YEARLY.value
         stripe_subscription_id = "sub_1K77Y5GlVGuVgOrkJrLjRnne"
@@ -361,7 +377,14 @@ def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscri
     @patch("services.billing.stripe.Subscription.retrieve")
     @patch("services.billing.stripe.SubscriptionSchedule.release")
     def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscription_with_grace_month_refund_if_valid_plan(
-        self, schedule_release_mock, retrieve_subscription_mock, cancel_sub_mock, list_invoice_mock, create_refund_mock, modify_customer_mock, modify_sub_mock
+        self,
+        schedule_release_mock,
+        retrieve_subscription_mock,
+        cancel_sub_mock,
+        list_invoice_mock,
+        create_refund_mock,
+        modify_customer_mock,
+        modify_sub_mock,
     ):
         with open("./services/tests/samples/stripe_invoice.json") as f:
             stripe_invoice_response = json.load(f)
@@ -387,16 +410,20 @@ def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscri
                 "new_quantity": 7,
                 "subscription_id": "sub_123",
                 "interval": "month",
-            }
+            },
         }
 
         retrieve_subscription_mock.return_value = MockSubscription(subscription_params)
         self.stripe.delete_subscription(owner)
         schedule_release_mock.assert_called_once_with(stripe_schedule_id)
         cancel_sub_mock.assert_called_once_with(stripe_subscription_id)
-        list_invoice_mock.assert_called_once_with(subscription=stripe_subscription_id, status="paid")
+        list_invoice_mock.assert_called_once_with(
+            subscription=stripe_subscription_id, status="paid"
+        )
         self.assertEqual(create_refund_mock.call_count, 2)
-        modify_customer_mock.assert_called_once_with(owner.stripe_customer_id, balance=0)
+        modify_customer_mock.assert_called_once_with(
+            owner.stripe_customer_id, balance=0
+        )
         modify_sub_mock.assert_not_called()
 
         owner.refresh_from_db()
@@ -414,7 +441,14 @@ def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscri
     @patch("services.billing.stripe.Subscription.retrieve")
     @patch("services.billing.stripe.SubscriptionSchedule.release")
     def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscription_with_grace_year_refund_if_valid_plan(
-        self, schedule_release_mock, retrieve_subscription_mock, cancel_sub_mock, list_invoice_mock, create_refund_mock, modify_customer_mock, modify_sub_mock
+        self,
+        schedule_release_mock,
+        retrieve_subscription_mock,
+        cancel_sub_mock,
+        list_invoice_mock,
+        create_refund_mock,
+        modify_customer_mock,
+        modify_sub_mock,
     ):
         with open("./services/tests/samples/stripe_invoice.json") as f:
             stripe_invoice_response = json.load(f)
@@ -440,16 +474,20 @@ def test_delete_subscription_with_schedule_releases_schedule_and_cancels_subscri
                 "new_quantity": 7,
                 "subscription_id": "sub_123",
                 "interval": "year",
-            }
+            },
         }
 
         retrieve_subscription_mock.return_value = MockSubscription(subscription_params)
         self.stripe.delete_subscription(owner)
         schedule_release_mock.assert_called_once_with(stripe_schedule_id)
         cancel_sub_mock.assert_called_once_with(stripe_subscription_id)
-        list_invoice_mock.assert_called_once_with(subscription=stripe_subscription_id, status="paid")
+        list_invoice_mock.assert_called_once_with(
+            subscription=stripe_subscription_id, status="paid"
+        )
         self.assertEqual(create_refund_mock.call_count, 2)
-        modify_customer_mock.assert_called_once_with(owner.stripe_customer_id, balance=0)
+        modify_customer_mock.assert_called_once_with(
+            owner.stripe_customer_id, balance=0
+        )
         modify_sub_mock.assert_not_called()
 
         owner.refresh_from_db()