@@ -242,7 +242,6 @@ async def __aexit__(
242242 charging_manager_ctx .set (None )
243243 self .active = False
244244
245- @_ensure_context
246245 @_ensure_context
247246 async def charge (self , event_name : str , count : int = 1 ) -> ChargeResult :
248247 def calculate_chargeable () -> dict [str , int | None ]:
@@ -337,15 +336,13 @@ def calculate_chargeable() -> dict[str, int | None]:
337336 chargeable_within_limit = calculate_chargeable (),
338337 )
339338
340- @_ensure_context
341339 @_ensure_context
342340 def calculate_total_charged_amount (self ) -> Decimal :
343341 return sum (
344342 (item .total_charged_amount for item in self ._charging_state .values ()),
345343 start = Decimal (),
346344 )
347345
348- @_ensure_context
349346 @_ensure_context
350347 def calculate_max_event_charge_count_within_limit (self , event_name : str ) -> int | None :
351348 price = self ._get_event_price (event_name )
@@ -356,7 +353,6 @@ def calculate_max_event_charge_count_within_limit(self, event_name: str) -> int
356353 result = (self ._max_total_charge_usd - self .calculate_total_charged_amount ()) / price
357354 return max (0 , math .floor (result )) if result .is_finite () else None
358355
359- @_ensure_context
360356 @_ensure_context
361357 def get_pricing_info (self ) -> ActorPricingInfo :
362358 return ActorPricingInfo (
@@ -370,13 +366,11 @@ def get_pricing_info(self) -> ActorPricingInfo:
370366 },
371367 )
372368
373- @_ensure_context
374369 @_ensure_context
375370 def get_charged_event_count (self , event_name : str ) -> int :
376371 item = self ._charging_state .get (event_name )
377372 return item .charge_count if item is not None else 0
378373
379- @_ensure_context
380374 @_ensure_context
381375 def get_max_total_charge_usd (self ) -> Decimal :
382376 return self ._max_total_charge_usd
0 commit comments