django-cfg 1.4.21__py3-none-any.whl → 1.4.24__py3-none-any.whl

django_cfg/apps/payments/services/core/subscription_service.py

@@ -4,18 +4,21 @@ Subscription service for the Universal Payment System v2.0.
 Handles subscription management and access control.
 """
 
-from typing import Optional, Dict, Any, List
+from datetime import timedelta
+from typing import List
+
 from django.contrib.auth.models import User
 from django.db import models
 from django.utils import timezone
-from datetime import timedelta
 
-from .base import BaseService
+from ...models import EndpointGroup, Subscription, Tariff
 from ..types import (
-    SubscriptionCreateRequest, SubscriptionResult, SubscriptionData,
-    ServiceOperationResult
+    ServiceOperationResult,
+    SubscriptionCreateRequest,
+    SubscriptionData,
+    SubscriptionResult,
 )
-from ...models import Subscription, EndpointGroup, Tariff
+from .base import BaseService
 
 
 class SubscriptionService(BaseService):
@@ -24,7 +27,7 @@ class SubscriptionService(BaseService):
     
     Handles subscription operations using Pydantic validation and Django ORM managers.
     """
-    
+
     def create_subscription(self, request: SubscriptionCreateRequest) -> SubscriptionResult:
         """
         Create new subscription for user.
@@ -39,13 +42,13 @@ class SubscriptionService(BaseService):
             # Validate request
             if isinstance(request, dict):
                 request = SubscriptionCreateRequest(**request)
-            
+
             self.logger.info("Creating subscription", extra={
                 'user_id': request.user_id,
                 'tier': request.tier,
                 'duration_days': request.duration_days
             })
-            
+
             # Get user
             try:
                 user = User.objects.get(id=request.user_id)
@@ -55,7 +58,7 @@ class SubscriptionService(BaseService):
                     message=f"User {request.user_id} not found",
                     error_code="user_not_found"
                 )
-            
+
             # Get tariff for tier
             try:
                 tariff = Tariff.objects.get(tier=request.tier, is_active=True)
@@ -65,21 +68,21 @@ class SubscriptionService(BaseService):
                     message=f"Tariff for tier {request.tier} not found",
                     error_code="tariff_not_found"
                 )
-            
+
             # Cancel existing active subscriptions
             existing_active = Subscription.objects.filter(
                 user=user,
                 status=Subscription.SubscriptionStatus.ACTIVE
             )
-            
+
             def create_subscription_transaction():
                 # Cancel existing subscriptions
                 for sub in existing_active:
                     sub.cancel("Replaced by new subscription")
-                
+
                 # Create new subscription
                 expires_at = timezone.now() + timedelta(days=request.duration_days)
-                
+
                 subscription = Subscription.objects.create(
                     user=user,
                     tier=request.tier,
@@ -90,7 +93,7 @@ class SubscriptionService(BaseService):
                     auto_renew=request.auto_renew,
                     expires_at=expires_at
                 )
-                
+
                 # Add endpoint groups
                 if request.endpoint_groups:
                     endpoint_groups = EndpointGroup.objects.filter(
@@ -102,14 +105,14 @@ class SubscriptionService(BaseService):
                     # Add default endpoint groups for tier
                     default_groups = self._get_default_endpoint_groups(request.tier)
                     subscription.endpoint_groups.set(default_groups)
-                
+
                 return subscription
-            
+
             subscription = self._execute_with_transaction(create_subscription_transaction)
-            
+
             # Convert to response data
             subscription_data = SubscriptionData.model_validate(subscription)
-            
+
             self._log_operation(
                 "create_subscription",
                 True,
@@ -117,7 +120,7 @@ class SubscriptionService(BaseService):
                 user_id=request.user_id,
                 tier=request.tier
             )
-            
+
             return SubscriptionResult(
                 success=True,
                 message="Subscription created successfully",
@@ -128,13 +131,13 @@ class SubscriptionService(BaseService):
                 expires_at=subscription.expires_at,
                 data={'subscription': subscription_data.model_dump()}
             )
-            
+
         except Exception as e:
             return SubscriptionResult(**self._handle_exception(
                 "create_subscription", e,
                 user_id=request.user_id if hasattr(request, 'user_id') else None
             ).model_dump())
-    
+
     def get_user_subscription(self, user_id: int) -> SubscriptionResult:
         """
         Get active subscription for user.
@@ -147,7 +150,7 @@ class SubscriptionService(BaseService):
         """
         try:
             self.logger.debug("Getting user subscription", extra={'user_id': user_id})
-            
+
             # Check user exists
             try:
                 user = User.objects.get(id=user_id)
@@ -157,20 +160,20 @@ class SubscriptionService(BaseService):
                     message=f"User {user_id} not found",
                     error_code="user_not_found"
                 )
-            
+
             # Get active subscription
             subscription = Subscription.objects.get_active_for_user(user)
-            
+
             if not subscription:
                 # Create free subscription if none exists
                 subscription = Subscription.objects.create_free_subscription(user)
-            
+
             # Convert to response data
             subscription_data = SubscriptionData.model_validate(subscription)
-            
+
             # Calculate requests remaining
             requests_remaining = self._calculate_requests_remaining(subscription)
-            
+
             return SubscriptionResult(
                 success=True,
                 message="Subscription retrieved successfully",
@@ -182,13 +185,13 @@ class SubscriptionService(BaseService):
                 requests_remaining=requests_remaining,
                 data={'subscription': subscription_data.model_dump()}
             )
-            
+
         except Exception as e:
             return SubscriptionResult(**self._handle_exception(
                 "get_user_subscription", e,
                 user_id=user_id
             ).model_dump())
-    
+
     def check_access(self, user_id: int, endpoint_group: str) -> ServiceOperationResult:
         """
         Check if user has access to endpoint group.
@@ -205,7 +208,7 @@ class SubscriptionService(BaseService):
                 'user_id': user_id,
                 'endpoint_group': endpoint_group
             })
-            
+
             # Get user subscription
             subscription_result = self.get_user_subscription(user_id)
             if not subscription_result.success:
@@ -213,30 +216,30 @@ class SubscriptionService(BaseService):
                     subscription_result.message,
                     subscription_result.error_code
                 )
-            
+
             subscription = Subscription.objects.get(id=subscription_result.subscription_id)
-            
+
             # Check if subscription is active and not expired
             if not subscription.is_active():
                 return self._create_error_result(
                     "Subscription is not active",
                     "subscription_inactive"
                 )
-            
+
             # Check endpoint group access
             has_access = subscription.has_access_to_endpoint_group(endpoint_group)
-            
+
             if not has_access:
                 return self._create_error_result(
                     f"Access denied to endpoint group: {endpoint_group}",
                     "access_denied"
                 )
-            
+
             # Check rate limits
             rate_limit_result = self._check_rate_limits(subscription)
             if not rate_limit_result.success:
                 return rate_limit_result
-            
+
             return self._create_success_result(
                 "Access granted",
                 {
@@ -247,14 +250,14 @@ class SubscriptionService(BaseService):
                     'requests_remaining': self._calculate_requests_remaining(subscription)
                 }
             )
-            
+
         except Exception as e:
             return self._handle_exception(
                 "check_access", e,
                 user_id=user_id,
                 endpoint_group=endpoint_group
             )
-    
+
     def increment_usage(self, user_id: int) -> ServiceOperationResult:
         """
         Increment subscription usage counter.
@@ -273,12 +276,12 @@ class SubscriptionService(BaseService):
                     subscription_result.message,
                     subscription_result.error_code
                 )
-            
+
             subscription = Subscription.objects.get(id=subscription_result.subscription_id)
-            
+
             # Increment usage using manager
             success = subscription.increment_usage()
-            
+
             if success:
                 return self._create_success_result(
                     "Usage incremented successfully",
@@ -294,13 +297,13 @@ class SubscriptionService(BaseService):
                     "Failed to increment usage",
                     "increment_failed"
                 )
-                
+
         except Exception as e:
             return self._handle_exception(
                 "increment_usage", e,
                 user_id=user_id
             )
-    
+
     def renew_subscription(self, subscription_id: str, duration_days: int = 30) -> SubscriptionResult:
         """
         Renew existing subscription.
@@ -317,7 +320,7 @@ class SubscriptionService(BaseService):
                 'subscription_id': subscription_id,
                 'duration_days': duration_days
             })
-            
+
             # Get subscription
             try:
                 subscription = Subscription.objects.get(id=subscription_id)
@@ -327,14 +330,14 @@ class SubscriptionService(BaseService):
                     message=f"Subscription {subscription_id} not found",
                     error_code="subscription_not_found"
                 )
-            
+
             # Renew using manager
             success = subscription.renew(duration_days)
-            
+
             if success:
                 subscription.refresh_from_db()
                 subscription_data = SubscriptionData.model_validate(subscription)
-                
+
                 self._log_operation(
                     "renew_subscription",
                     True,
@@ -342,7 +345,7 @@ class SubscriptionService(BaseService):
                     duration_days=duration_days,
                     new_expires_at=subscription.expires_at.isoformat()
                 )
-                
+
                 return SubscriptionResult(
                     success=True,
                     message="Subscription renewed successfully",
@@ -359,13 +362,13 @@ class SubscriptionService(BaseService):
                     message="Failed to renew subscription",
                     error_code="renewal_failed"
                 )
-                
+
         except Exception as e:
             return SubscriptionResult(**self._handle_exception(
                 "renew_subscription", e,
                 subscription_id=subscription_id
             ).model_dump())
-    
+
     def cancel_subscription(self, subscription_id: str, reason: str = None) -> SubscriptionResult:
         """
         Cancel subscription.
@@ -382,7 +385,7 @@ class SubscriptionService(BaseService):
                 'subscription_id': subscription_id,
                 'reason': reason
             })
-            
+
             # Get subscription
             try:
                 subscription = Subscription.objects.get(id=subscription_id)
@@ -392,21 +395,21 @@ class SubscriptionService(BaseService):
                     message=f"Subscription {subscription_id} not found",
                     error_code="subscription_not_found"
                 )
-            
+
             # Cancel using manager
             success = subscription.cancel(reason)
-            
+
             if success:
                 subscription.refresh_from_db()
                 subscription_data = SubscriptionData.model_validate(subscription)
-                
+
                 self._log_operation(
                     "cancel_subscription",
                     True,
                     subscription_id=subscription_id,
                     reason=reason
                 )
-                
+
                 return SubscriptionResult(
                     success=True,
                     message="Subscription cancelled successfully",
@@ -422,13 +425,13 @@ class SubscriptionService(BaseService):
                     message="Failed to cancel subscription",
                     error_code="cancellation_failed"
                 )
-                
+
         except Exception as e:
             return SubscriptionResult(**self._handle_exception(
                 "cancel_subscription", e,
                 subscription_id=subscription_id
             ).model_dump())
-    
+
     def get_subscription_stats(self, days: int = 30) -> ServiceOperationResult:
         """
         Get subscription statistics.
@@ -441,7 +444,7 @@ class SubscriptionService(BaseService):
         """
         try:
             since = timezone.now() - timedelta(days=days)
-            
+
             # Overall stats
             overall_stats = Subscription.objects.aggregate(
                 total_subscriptions=models.Count('id'),
@@ -458,7 +461,7 @@ class SubscriptionService(BaseService):
                     filter=models.Q(status=Subscription.SubscriptionStatus.CANCELLED)
                 )
             )
-            
+
             # Tier breakdown
             tier_breakdown = Subscription.objects.values('tier').annotate(
                 count=models.Count('id'),
@@ -467,7 +470,7 @@ class SubscriptionService(BaseService):
                     filter=models.Q(status=Subscription.SubscriptionStatus.ACTIVE)
                 )
             ).order_by('-count')
-            
+
             # Recent activity
             recent_stats = Subscription.objects.filter(
                 created_at__gte=since
@@ -475,7 +478,7 @@ class SubscriptionService(BaseService):
                 new_subscriptions=models.Count('id'),
                 total_requests=models.Sum('total_requests')
             )
-            
+
             stats = {
                 'period_days': days,
                 'overall_stats': overall_stats,
@@ -483,15 +486,15 @@ class SubscriptionService(BaseService):
                 'recent_stats': recent_stats,
                 'generated_at': timezone.now().isoformat()
             }
-            
+
             return self._create_success_result(
                 f"Subscription statistics for {days} days",
                 stats
             )
-            
+
         except Exception as e:
             return self._handle_exception("get_subscription_stats", e)
-    
+
     def _get_default_endpoint_groups(self, tier: str) -> List[EndpointGroup]:
         """Get default endpoint groups for subscription tier."""
         tier_groups = {
@@ -500,18 +503,18 @@ class SubscriptionService(BaseService):
             'pro': ['payments', 'balance', 'subscriptions', 'analytics'],
             'enterprise': ['payments', 'balance', 'subscriptions', 'analytics', 'admin']
         }
-        
+
         group_codes = tier_groups.get(tier, ['payments'])
         return EndpointGroup.objects.filter(
             code__in=group_codes,
             is_enabled=True
         )
-    
+
     def _calculate_requests_remaining(self, subscription: Subscription) -> int:
         """Calculate remaining requests for today."""
         # Simple calculation - in production this would check daily usage
         return max(0, subscription.requests_per_day - subscription.total_requests)
-    
+
     def _check_rate_limits(self, subscription: Subscription) -> ServiceOperationResult:
         """Check if subscription has exceeded rate limits."""
         # Simplified rate limit check
@@ -520,9 +523,9 @@ class SubscriptionService(BaseService):
                 "Daily request limit exceeded",
                 "rate_limit_exceeded"
             )
-        
+
         return self._create_success_result("Rate limits OK")
-    
+
     def health_check(self) -> ServiceOperationResult:
         """Perform subscription service health check."""
         try:
@@ -531,24 +534,24 @@ class SubscriptionService(BaseService):
             active_count = Subscription.objects.filter(
                 status=Subscription.SubscriptionStatus.ACTIVE
             ).count()
-            
+
             # Check for expired subscriptions that need cleanup
             expired_count = Subscription.objects.filter(
                 status=Subscription.SubscriptionStatus.ACTIVE,
                 expires_at__lt=timezone.now()
             ).count()
-            
+
             stats = {
                 'total_subscriptions': subscription_count,
                 'active_subscriptions': active_count,
                 'expired_needing_cleanup': expired_count,
                 'service_name': 'SubscriptionService'
             }
-            
+
             return self._create_success_result(
                 "SubscriptionService is healthy",
                 stats
             )
-            
+
         except Exception as e:
             return self._handle_exception("health_check", e)

django_cfg/apps/payments/services/core/utils/data_converter.py

@@ -5,6 +5,7 @@ Converts between Django models and Pydantic data models.
 """
 
 from typing import TYPE_CHECKING
+
 from ...types import PaymentData
 
 if TYPE_CHECKING:

django_cfg/apps/payments/services/core/utils/statistics_calculator.py

@@ -5,6 +5,7 @@ Calculates various payment statistics and metrics.
 """
 
 from datetime import timedelta
+
 from django.db import models
 from django.utils import timezone