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

django_cfg/apps/payments/models/managers/payment_managers.py

@@ -4,12 +4,14 @@ Payment managers for the Universal Payment System v2.0.
 Optimized querysets and managers for payment operations.
 """
 
-from typing import Optional, Dict, Any
-from decimal import Decimal
 from datetime import datetime
+from decimal import Decimal
+from typing import Any, Dict, Optional
+
 from django.db import models
 from django.utils import timezone
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
+
 from django_cfg.modules.django_logging import get_logger
 
 logger = get_logger("payment_managers")
@@ -26,18 +28,18 @@ class PaymentStatusUpdateFields(BaseModel):
         extra="forbid",
         str_strip_whitespace=True
     )
-    
+
     # Transaction related fields
     transaction_hash: Optional[str] = Field(None, min_length=1, max_length=200, description="Blockchain transaction hash")
     confirmations_count: Optional[int] = Field(None, ge=0, description="Number of blockchain confirmations")
-    
-    # Amount related fields  
+
+    # Amount related fields
     actual_amount_usd: Optional[Decimal] = Field(None, gt=0, description="Actual amount received in USD")
     fee_amount_usd: Optional[Decimal] = Field(None, ge=0, description="Fee amount in USD")
-    
+
     # Provider data
     provider_data: Optional[Dict[str, Any]] = Field(None, description="Provider-specific data")
-    
+
     # Completion timestamp (auto-set by manager for completed status)
     completed_at: Optional[datetime] = Field(None, description="Completion timestamp")
 
@@ -48,7 +50,7 @@ class PaymentQuerySet(models.QuerySet):
     
     Provides efficient queries with proper indexing and select_related optimization.
     """
-    
+
     def optimized(self):
         """
         Prevent N+1 queries with select_related and prefetch_related.
@@ -56,26 +58,26 @@ class PaymentQuerySet(models.QuerySet):
         Use this for admin interfaces and API responses.
         """
         return self.select_related(
-            'user', 
-            'currency', 
+            'user',
+            'currency',
             'network'
         ).prefetch_related(
             'user__payment_balance',
             'user__payment_transactions'
         )
-    
+
     def by_status(self, status):
         """Filter by payment status with index optimization."""
         return self.filter(status=status)
-    
+
     def by_provider(self, provider):
         """Filter by payment provider."""
         return self.filter(provider=provider)
-    
+
     def by_user(self, user):
         """Filter by user with proper indexing."""
         return self.filter(user=user)
-    
+
     def by_amount_range(self, min_amount=None, max_amount=None):
         """
         Filter by USD amount range.
@@ -90,36 +92,36 @@ class PaymentQuerySet(models.QuerySet):
         if max_amount is not None:
             queryset = queryset.filter(amount_usd__lte=max_amount)
         return queryset
-    
+
     def by_currency(self, currency_code):
         """Filter by currency code."""
         return self.filter(currency__code=currency_code)
-    
+
     def by_network(self, network_code):
         """Filter by blockchain network."""
         return self.filter(network__code=network_code)
-    
+
     # Status-based filters
     def completed(self):
         """Get completed payments."""
         return self.filter(status='completed')
-    
+
     def pending(self):
         """Get pending payments."""
         return self.filter(status='pending')
-    
+
     def failed(self):
         """Get failed payments (failed, expired, cancelled)."""
         return self.filter(status__in=['failed', 'expired', 'cancelled'])
-    
+
     def confirming(self):
         """Get payments awaiting confirmation."""
         return self.filter(status__in=['confirming', 'confirmed'])
-    
+
     def active(self):
         """Get active payments (not failed or completed)."""
         return self.filter(status__in=['pending', 'confirming', 'confirmed'])
-    
+
     # Time-based filters
     def recent(self, hours=24):
         """
@@ -130,22 +132,22 @@ class PaymentQuerySet(models.QuerySet):
         """
         since = timezone.now() - timezone.timedelta(hours=hours)
         return self.filter(created_at__gte=since)
-    
+
     def today(self):
         """Get payments created today."""
         today = timezone.now().date()
         return self.filter(created_at__date=today)
-    
+
     def this_week(self):
         """Get payments from this week."""
         week_start = timezone.now().date() - timezone.timedelta(days=timezone.now().weekday())
         return self.filter(created_at__date__gte=week_start)
-    
+
     def this_month(self):
         """Get payments from this month."""
         month_start = timezone.now().replace(day=1).date()
         return self.filter(created_at__date__gte=month_start)
-    
+
     def expired(self):
         """Get expired payments."""
         now = timezone.now()
@@ -153,7 +155,7 @@ class PaymentQuerySet(models.QuerySet):
             expires_at__lte=now,
             status__in=['pending', 'confirming']
         )
-    
+
     def expiring_soon(self, hours=2):
         """
         Get payments expiring in the next N hours.
@@ -167,59 +169,59 @@ class PaymentQuerySet(models.QuerySet):
             expires_at__gt=timezone.now(),
             status__in=['pending', 'confirming']
         )
-    
+
     # Provider-specific filters
     def nowpayments(self):
         """Get NowPayments payments."""
         return self.filter(provider='nowpayments')
-    
+
     def crypto_payments(self):
         """Get cryptocurrency payments."""
         return self.filter(currency__currency_type='crypto')
-    
+
     def fiat_payments(self):
         """Get fiat currency payments."""
         return self.filter(currency__currency_type='fiat')
-    
+
     # Aggregation methods
     def total_amount(self):
         """Get total USD amount for queryset."""
         result = self.aggregate(total=models.Sum('amount_usd'))
         return result['total'] or 0.0
-    
+
     def average_amount(self):
         """Get average USD amount for queryset."""
         result = self.aggregate(avg=models.Avg('amount_usd'))
         return result['avg'] or 0.0
-    
+
     def count_by_status(self):
         """Get count of payments grouped by status."""
         return self.values('status').annotate(count=models.Count('id')).order_by('status')
-    
+
     def count_by_provider(self):
         """Get count of payments grouped by provider."""
         return self.values('provider').annotate(count=models.Count('id')).order_by('provider')
-    
+
     def count_by_currency(self):
         """Get count of payments grouped by currency."""
         return self.values('currency__code').annotate(count=models.Count('id')).order_by('currency__code')
-    
+
     # Advanced queries
     def with_transactions(self):
         """Include related transaction data."""
         return self.prefetch_related('user__payment_transactions')
-    
+
     def with_balance_info(self):
         """Include user balance information."""
         return self.select_related('user__payment_balance')
-    
+
     def requiring_confirmation(self):
         """Get payments that need blockchain confirmation."""
         return self.filter(
             status__in=['confirming', 'confirmed'],
             transaction_hash__isnull=False
         )
-    
+
     def large_amounts(self, threshold=1000.0):
         """
         Get payments above threshold amount.
@@ -228,7 +230,7 @@ class PaymentQuerySet(models.QuerySet):
             threshold: USD amount threshold (default: $1000)
         """
         return self.filter(amount_usd__gte=threshold)
-    
+
     def small_amounts(self, threshold=10.0):
         """
         Get payments below threshold amount.
@@ -245,80 +247,80 @@ class PaymentManager(models.Manager):
     
     Provides high-level methods for common payment operations.
     """
-    
+
     def get_queryset(self):
         """Return optimized queryset by default."""
         return PaymentQuerySet(self.model, using=self._db)
-    
+
     def optimized(self):
         """Get optimized queryset for admin/API use."""
         return self.get_queryset().optimized()
-    
+
     # Status-based methods
     def by_status(self, status):
         """Get payments by status."""
         return self.get_queryset().by_status(status)
-    
+
     def completed(self):
         """Get completed payments."""
         return self.get_queryset().completed()
-    
+
     def pending(self):
         """Get pending payments."""
         return self.get_queryset().pending()
-    
+
     def failed(self):
         """Get failed payments."""
         return self.get_queryset().failed()
-    
+
     def active(self):
         """Get active payments."""
         return self.get_queryset().active()
-    
+
     # User-based methods
     def by_user(self, user):
         """Get payments by user."""
         return self.get_queryset().by_user(user)
-    
+
     # Provider-based methods
     def by_provider(self, provider):
         """Get payments by provider."""
         return self.get_queryset().by_provider(provider)
-    
+
     def nowpayments(self):
         """Get NowPayments payments."""
         return self.get_queryset().nowpayments()
-    
+
     # Time-based methods
     def recent(self, hours=24):
         """Get recent payments."""
         return self.get_queryset().recent(hours)
-    
+
     def today(self):
         """Get today's payments."""
         return self.get_queryset().today()
-    
+
     def this_week(self):
         """Get this week's payments."""
         return self.get_queryset().this_week()
-    
+
     def this_month(self):
         """Get this month's payments."""
         return self.get_queryset().this_month()
-    
+
     # Maintenance methods
     def expired(self):
         """Get expired payments."""
         return self.get_queryset().expired()
-    
+
     def expiring_soon(self, hours=2):
         """Get payments expiring soon."""
         return self.get_queryset().expiring_soon(hours)
-    
+
     def requiring_confirmation(self):
         """Get payments needing confirmation."""
         return self.get_queryset().requiring_confirmation()
-    
+
     # Statistics methods
     def get_stats(self, days=30):
         """
@@ -332,7 +334,7 @@ class PaymentManager(models.Manager):
         """
         since = timezone.now() - timezone.timedelta(days=days)
         queryset = self.filter(created_at__gte=since)
-        
+
         stats = {
             'total_payments': queryset.count(),
             'total_amount_usd': queryset.total_amount(),
@@ -344,15 +346,15 @@ class PaymentManager(models.Manager):
             'by_provider': list(queryset.count_by_provider()),
             'by_currency': list(queryset.count_by_currency()),
         }
-        
+
         logger.info(f"Generated payment stats for {days} days", extra={
             'days': days,
             'total_payments': stats['total_payments'],
             'total_amount': stats['total_amount_usd']
         })
-        
+
         return stats
-    
+
     def cleanup_expired(self, dry_run=True):
         """
         Mark expired payments as failed.
@@ -365,13 +367,13 @@ class PaymentManager(models.Manager):
         """
         expired_payments = self.expired()
         count = expired_payments.count()
-        
+
         if not dry_run and count > 0:
             expired_payments.update(status='expired')
             logger.info(f"Marked {count} payments as expired")
-        
+
         return count
-    
+
     def get_user_payment_summary(self, user):
         """
         Get payment summary for a specific user.
@@ -383,7 +385,7 @@ class PaymentManager(models.Manager):
             dict: User payment summary
         """
         user_payments = self.filter(user=user)
-        
+
         summary = {
             'total_payments': user_payments.count(),
             'total_amount_usd': user_payments.total_amount(),
@@ -393,9 +395,9 @@ class PaymentManager(models.Manager):
             'last_payment_at': user_payments.first().created_at if user_payments.exists() else None,
             'average_amount_usd': user_payments.average_amount(),
         }
-        
+
         return summary
-    
+
     # Business logic methods
     def mark_payment_completed(self, payment_id, actual_amount_usd=None, transaction_hash=None):
         """
@@ -414,41 +416,41 @@ class PaymentManager(models.Manager):
                 payment = self.get(id=payment_id)
             else:
                 payment = payment_id
-            
+
             # Validate payment can be completed
-            if not payment.status in ['pending', 'confirming', 'confirmed']:
+            if payment.status not in ['pending', 'confirming', 'confirmed']:
                 logger.warning(f"Cannot complete payment in status {payment.status}", extra={
                     'payment_id': str(payment.id),
                     'current_status': payment.status
                 })
                 return False
-            
+
             # Update payment using centralized status update method
             extra_fields = PaymentStatusUpdateFields(
                 actual_amount_usd=actual_amount_usd,
                 transaction_hash=transaction_hash
             )
-            
+
             success = self.update_payment_status(payment, 'completed', extra_fields)
             if not success:
                 return False
-            
-            logger.info(f"Payment marked as completed", extra={
+
+            logger.info("Payment marked as completed", extra={
                 'payment_id': str(payment.id),
                 'user_id': payment.user.id,
                 'amount_usd': payment.amount_usd,
                 'actual_amount_usd': actual_amount_usd,
                 'transaction_hash': transaction_hash
             })
-            
+
             return True
-            
+
         except Exception as e:
             logger.error(f"Failed to mark payment as completed: {e}", extra={
                 'payment_id': str(payment_id) if hasattr(payment_id, 'id') else payment_id
             })
             return False
-    
+
     def mark_payment_failed(self, payment_id, reason=None, error_code=None):
         """
         Mark payment as failed (business logic in manager).
@@ -466,7 +468,7 @@ class PaymentManager(models.Manager):
                 payment = self.get(id=payment_id)
             else:
                 payment = payment_id
-            
+
             # Prepare error info
             provider_data = payment.provider_data.copy() if payment.provider_data else {}
             if reason or error_code:
@@ -477,31 +479,31 @@ class PaymentManager(models.Manager):
                 if error_code:
                     provider_data['error_info']['code'] = error_code
                 provider_data['error_info']['failed_at'] = timezone.now().isoformat()
-            
+
             # Update payment using centralized status update method
             extra_fields = PaymentStatusUpdateFields(
                 provider_data=provider_data if provider_data != payment.provider_data else None
             )
-            
+
             success = self.update_payment_status(payment, 'failed', extra_fields)
             if not success:
                 return False
-            
-            logger.warning(f"Payment marked as failed", extra={
+
+            logger.warning("Payment marked as failed", extra={
                 'payment_id': str(payment.id),
                 'user_id': payment.user.id,
                 'reason': reason,
                 'error_code': error_code
             })
-            
+
             return True
-            
+
         except Exception as e:
             logger.error(f"Failed to mark payment as failed: {e}", extra={
                 'payment_id': str(payment_id) if hasattr(payment_id, 'id') else payment_id
             })
             return False
-    
+
     def cancel_payment(self, payment_id, reason=None):
         """
         Cancel payment (business logic in manager).
@@ -518,15 +520,15 @@ class PaymentManager(models.Manager):
                 payment = self.get(id=payment_id)
             else:
                 payment = payment_id
-            
+
             # Validate payment can be cancelled
-            if not payment.status in ['pending', 'confirming']:
+            if payment.status not in ['pending', 'confirming']:
                 logger.warning(f"Cannot cancel payment in status {payment.status}", extra={
                     'payment_id': str(payment.id),
                     'current_status': payment.status
                 })
                 return False
-            
+
             # Prepare cancellation info
             provider_data = payment.provider_data.copy() if payment.provider_data else {}
             if reason:
@@ -534,34 +536,34 @@ class PaymentManager(models.Manager):
                     provider_data['cancellation_info'] = {}
                 provider_data['cancellation_info']['reason'] = reason
                 provider_data['cancellation_info']['cancelled_at'] = timezone.now().isoformat()
-            
+
             # Update payment using centralized status update method
             extra_fields = PaymentStatusUpdateFields(
                 provider_data=provider_data if provider_data != payment.provider_data else None
             )
-            
+
             success = self.update_payment_status(payment, 'cancelled', extra_fields)
             if not success:
                 return False
-            
-            logger.info(f"Payment cancelled", extra={
+
+            logger.info("Payment cancelled", extra={
                 'payment_id': str(payment.id),
                 'user_id': payment.user.id,
                 'reason': reason
             })
-            
+
             return True
-            
+
         except Exception as e:
             logger.error(f"Failed to cancel payment: {e}", extra={
                 'payment_id': str(payment_id) if hasattr(payment_id, 'id') else payment_id
             })
             return False
-    
+
     def update_payment_status(
-        self, 
-        payment, 
-        new_status: str, 
+        self,
+        payment,
+        new_status: str,
         extra_fields: Optional[PaymentStatusUpdateFields] = None
     ) -> bool:
         """
@@ -577,25 +579,25 @@ class PaymentManager(models.Manager):
         """
         try:
             old_status = payment.status
-            
+
             # Only update if status actually changed
             if old_status != new_status:
                 payment.status = new_status
                 payment.status_changed_at = timezone.now()
-                
+
                 # Set completed_at if status changed to completed
                 if new_status == 'completed' and not payment.completed_at:
                     payment.completed_at = timezone.now()
-                
+
                 # Update fields list for save()
                 update_fields = ['status', 'status_changed_at', 'updated_at']
-                
+
                 # Apply extra fields if provided
                 if extra_fields:
                     # Validate extra fields
                     if isinstance(extra_fields, dict):
                         extra_fields = PaymentStatusUpdateFields(**extra_fields)
-                    
+
                     # Apply non-None fields
                     for field_name, field_value in extra_fields.model_dump(exclude_none=True).items():
                         if hasattr(payment, field_name):
@@ -606,28 +608,28 @@ class PaymentManager(models.Manager):
                                 'payment_id': str(payment.id),
                                 'field_name': field_name
                             })
-                
+
                 payment.save(update_fields=update_fields)
-                
-                logger.info(f"Payment status updated", extra={
+
+                logger.info("Payment status updated", extra={
                     'payment_id': str(payment.id),
                     'old_status': old_status,
                     'new_status': new_status,
                     'updated_fields': update_fields
                 })
-                
+
                 return True
             else:
-                logger.debug(f"Payment status unchanged", extra={
+                logger.debug("Payment status unchanged", extra={
                     'payment_id': str(payment.id),
                     'status': new_status
                 })
                 return False
-                
+
         except Exception as e:
             logger.error(f"Failed to update payment status: {e}", extra={
                 'payment_id': str(payment.id),
                 'old_status': old_status if 'old_status' in locals() else 'unknown',
                 'new_status': new_status
             })
-            return False
+            return False