Add subcription feature 9
This commit is contained in:
@@ -247,6 +247,50 @@ class SubscriptionRepository(TenantBaseRepository):
|
||||
error=str(e))
|
||||
raise DatabaseError(f"Failed to update plan: {str(e)}")
|
||||
|
||||
async def update_subscription_status(
|
||||
self,
|
||||
subscription_id: str,
|
||||
status: str,
|
||||
additional_data: Dict[str, Any] = None
|
||||
) -> Optional[Subscription]:
|
||||
"""Update subscription status with optional additional data"""
|
||||
try:
|
||||
# Get subscription to find tenant_id for cache invalidation
|
||||
subscription = await self.get_by_id(subscription_id)
|
||||
if not subscription:
|
||||
raise ValidationError(f"Subscription {subscription_id} not found")
|
||||
|
||||
update_data = {
|
||||
"status": status,
|
||||
"updated_at": datetime.utcnow()
|
||||
}
|
||||
|
||||
# Merge additional data if provided
|
||||
if additional_data:
|
||||
update_data.update(additional_data)
|
||||
|
||||
updated_subscription = await self.update(subscription_id, update_data)
|
||||
|
||||
# Invalidate cache
|
||||
if subscription.tenant_id:
|
||||
await self._invalidate_cache(str(subscription.tenant_id))
|
||||
|
||||
logger.info("Subscription status updated",
|
||||
subscription_id=subscription_id,
|
||||
new_status=status,
|
||||
additional_data=additional_data)
|
||||
|
||||
return updated_subscription
|
||||
|
||||
except ValidationError:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error("Failed to update subscription status",
|
||||
subscription_id=subscription_id,
|
||||
status=status,
|
||||
error=str(e))
|
||||
raise DatabaseError(f"Failed to update subscription status: {str(e)}")
|
||||
|
||||
async def cancel_subscription(
|
||||
self,
|
||||
subscription_id: str,
|
||||
|
||||
Reference in New Issue
Block a user