server/transaction: fix following migration to product

polarsource · May 14, 2024 · 2ad3fbd · 2ad3fbd
1 parent b60ed00
commit 2ad3fbd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/server/polar/transaction/schemas.py b/server/polar/transaction/schemas.py
@@ -59,15 +59,12 @@ class TransactionIssueReward(TimestampedSchema):
     share_thousands: int
 
 
-class TransactionSubscriptionTier(TimestampedSchema):
+class TransactionProduct(TimestampedSchema):
     id: UUID4
     type: SubscriptionTierType
     name: str
     organization_id: UUID4 | None = None
-    repository_id: UUID4 | None = None
-
     organization: TransactionOrganization | None = None
-    repository: TransactionRepository | None = None
 
 
 class TransactionSubscriptionPrice(TimestampedSchema):
@@ -81,7 +78,7 @@ class TransactionSubscriptionPrice(TimestampedSchema):
 class TransactionSubscription(TimestampedSchema):
     id: UUID4
     status: SubscriptionStatus
-    subscription_tier: TransactionSubscriptionTier
+    product: TransactionProduct
 
 
 class TransactionEmbedded(TimestampedSchema):

diff --git a/server/polar/transaction/service/dispute.py b/server/polar/transaction/service/dispute.py
@@ -69,7 +69,7 @@ async def create_dispute(
             pledge_id=payment_transaction.pledge_id,
             issue_reward_id=payment_transaction.issue_reward_id,
             subscription_id=payment_transaction.subscription_id,
-            subscription_tier_price_id=payment_transaction.product_price_id,
+            product_price_id=payment_transaction.product_price_id,
         )
 
         # Compute and link fees
@@ -137,7 +137,7 @@ async def create_dispute_reversal(
             pledge_id=payment_transaction.pledge_id,
             issue_reward_id=payment_transaction.issue_reward_id,
             subscription_id=payment_transaction.subscription_id,
-            subscription_tier_price_id=payment_transaction.product_price_id,
+            product_price_id=payment_transaction.product_price_id,
         )
 
         # Compute and link fees

diff --git a/server/polar/transaction/service/refund.py b/server/polar/transaction/service/refund.py
@@ -78,7 +78,7 @@ async def create_refunds(
                 pledge_id=payment_transaction.pledge_id,
                 issue_reward_id=payment_transaction.issue_reward_id,
                 subscription_id=payment_transaction.subscription_id,
-                subscription_tier_price_id=payment_transaction.product_price_id,
+                product_price_id=payment_transaction.product_price_id,
             )
 
             # Compute and link fees