refator: revert task files to be the same as main

Author: psincraian

server/polar/checkout/tasks.py

@@ -17,10 +17,7 @@ def __init__(self, checkout_id: uuid.UUID) -> None:
         super().__init__(message)
 
 
-@actor(
-    actor_name="checkout.handle_free_success",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="checkout.handle_free_success", priority=TaskPriority.HIGH)
 async def handle_free_success(checkout_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
         repository = CheckoutRepository.from_session(session)

server/polar/eventstream/tasks.py

@@ -3,9 +3,6 @@
 from .service import send_event
 
 
-@actor(
-    actor_name="eventstream.publish",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="eventstream.publish", priority=TaskPriority.HIGH)
 async def eventstream_publish(event: str, channels: list[str]) -> None:
     await send_event(RedisMiddleware.get(), event, channels)

server/polar/integrations/stripe/tasks.py

@@ -77,10 +77,7 @@ async def wrapper(*args: Params.args, **kwargs: Params.kwargs) -> ReturnValue:
 class StripeTaskError(PolarTaskError): ...
 
 
-@actor(
-    actor_name="stripe.webhook.account.updated",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.account.updated", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def account_updated(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -91,10 +88,7 @@ async def account_updated(event_id: uuid.UUID) -> None:
             )
 
 
-@actor(
-    actor_name="stripe.webhook.payment_intent.succeeded",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.payment_intent.succeeded", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def payment_intent_succeeded(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -160,10 +154,7 @@ async def payment_intent_payment_failed(event_id: uuid.UUID) -> None:
                     raise
 
 
-@actor(
-    actor_name="stripe.webhook.setup_intent.succeeded",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.setup_intent.succeeded", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def setup_intent_succeeded(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -182,8 +173,7 @@ async def setup_intent_succeeded(event_id: uuid.UUID) -> None:
 
 
 @actor(
-    actor_name="stripe.webhook.setup_intent.setup_failed",
-    priority=TaskPriority.HIGH,
+    actor_name="stripe.webhook.setup_intent.setup_failed", priority=TaskPriority.HIGH
 )
 @stripe_api_connection_error_retry
 async def setup_intent_setup_failed(event_id: uuid.UUID) -> None:
@@ -201,10 +191,7 @@ async def setup_intent_setup_failed(event_id: uuid.UUID) -> None:
                     raise
 
 
-@actor(
-    actor_name="stripe.webhook.charge.pending",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.charge.pending", priority=TaskPriority.HIGH)
 async def charge_pending(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
         async with external_event_service.handle_stripe(session, event_id) as event:
@@ -224,10 +211,7 @@ async def charge_pending(event_id: uuid.UUID) -> None:
             )
 
 
-@actor(
-    actor_name="stripe.webhook.charge.failed",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.charge.failed", priority=TaskPriority.HIGH)
 async def charge_failed(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
         async with external_event_service.handle_stripe(session, event_id) as event:
@@ -243,10 +227,7 @@ async def charge_failed(event_id: uuid.UUID) -> None:
                     raise
 
 
-@actor(
-    actor_name="stripe.webhook.charge.succeeded",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.charge.succeeded", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def charge_succeeded(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -264,10 +245,7 @@ async def charge_succeeded(event_id: uuid.UUID) -> None:
                     raise
 
 
-@actor(
-    actor_name="stripe.webhook.refund.created",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.refund.created", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def refund_created(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -282,10 +260,7 @@ async def refund_created(event_id: uuid.UUID) -> None:
             await refund_service.create_from_stripe(session, stripe_refund=refund)
 
 
-@actor(
-    actor_name="stripe.webhook.refund.updated",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.refund.updated", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def refund_updated(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -300,10 +275,7 @@ async def refund_updated(event_id: uuid.UUID) -> None:
             await refund_service.upsert_from_stripe(session, stripe_refund=refund)
 
 
-@actor(
-    actor_name="stripe.webhook.refund.failed",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.refund.failed", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def refund_failed(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -318,10 +290,7 @@ async def refund_failed(event_id: uuid.UUID) -> None:
             await refund_service.upsert_from_stripe(session, stripe_refund=refund)
 
 
-@actor(
-    actor_name="stripe.webhook.charge.dispute.closed",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.charge.dispute.closed", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def charge_dispute_closed(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -389,10 +358,7 @@ async def customer_subscription_deleted(event_id: uuid.UUID) -> None:
                     raise
 
 
-@actor(
-    actor_name="stripe.webhook.invoice.created",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.invoice.created", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def invoice_created(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
@@ -414,10 +380,7 @@ async def invoice_created(event_id: uuid.UUID) -> None:
                 return
 
 
-@actor(
-    actor_name="stripe.webhook.invoice.paid",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="stripe.webhook.invoice.paid", priority=TaskPriority.HIGH)
 @stripe_api_connection_error_retry
 async def invoice_paid(event_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:

server/polar/subscription/tasks.py

@@ -86,10 +86,7 @@ async def subscription_update_meters(subscription_id: uuid.UUID) -> None:
         await subscription_service.update_meters(session, subscription)
 
 
-@actor(
-    actor_name="subscription.cancel_customer",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="subscription.cancel_customer", priority=TaskPriority.HIGH)
 async def subscription_cancel_customer(customer_id: uuid.UUID) -> None:
     async with AsyncSessionMaker() as session:
         await subscription_service.cancel_customer(session, customer_id)

server/polar/webhook/tasks.py

@@ -126,10 +126,7 @@ async def _webhook_event_send(
             await session.commit()
 
 
-@actor(
-    actor_name="webhook_event.success",
-    priority=TaskPriority.HIGH,
-)
+@actor(actor_name="webhook_event.success", priority=TaskPriority.HIGH)
 async def webhook_event_success(webhook_event_id: UUID) -> None:
     async with AsyncSessionMaker() as session:
         return await webhook_service.on_event_success(session, webhook_event_id)