django-cfg 1.4.20__py3-none-any.whl → 1.4.23__py3-none-any.whl

django_cfg/apps/maintenance/services/maintenance_service.py

@@ -6,7 +6,8 @@ No Workers, no templates, no complexity - just Page Rules!
 
 import logging
 import time
-from typing import Dict, Any
+from typing import Any, Dict
+
 from cloudflare import Cloudflare
 
 from ..models import CloudflareSite, MaintenanceLog
@@ -21,12 +22,12 @@ class MaintenanceService:
     Enable: Create Page Rule redirect to maintenance.reforms.ai
     Disable: Delete Page Rule
     """
-    
+
     def __init__(self, site: CloudflareSite):
         """Initialize service for specific site."""
         self.site = site
         self.client = Cloudflare(api_token=site.api_key.api_token)
-    
+
     def enable_maintenance(self, reason: str = "Scheduled maintenance") -> MaintenanceLog:
         """
         Enable maintenance mode using Page Rule redirect.
@@ -38,15 +39,15 @@ class MaintenanceService:
         """
         start_time = time.time()
         log_entry = MaintenanceLog.log_pending(self.site, MaintenanceLog.Action.ENABLE, reason)
-        
+
         try:
             # 1. Create Page Rule for maintenance redirect
             logger.info(f"Creating page rule for: {self.site.domain} → {self.site.get_maintenance_url()}")
             page_rule_response = self._create_maintenance_page_rule()
-            
+
             # 2. Update site status
             self.site.enable_maintenance()
-            
+
             # 3. Log success
             duration = int(time.time() - start_time)
             log_entry.status = MaintenanceLog.Status.SUCCESS
@@ -55,9 +56,9 @@ class MaintenanceService:
                 'page_rule_create': self._serialize_response(page_rule_response)
             }
             log_entry.save()
-            
+
             return log_entry
-            
+
         except Exception as e:
             # Log failure
             duration = int(time.time() - start_time)
@@ -65,9 +66,9 @@ class MaintenanceService:
             log_entry.error_message = str(e)
             log_entry.duration_seconds = duration
             log_entry.save()
-            
+
             raise e
-    
+
     def disable_maintenance(self) -> MaintenanceLog:
         """
         Disable maintenance mode by removing Page Rule.
@@ -79,15 +80,15 @@ class MaintenanceService:
         """
         start_time = time.time()
         log_entry = MaintenanceLog.log_pending(self.site, MaintenanceLog.Action.DISABLE)
-        
+
         try:
             # 1. Remove Page Rule
             logger.info(f"Removing page rule for: {self.site.domain}")
             page_rule_response = self._delete_maintenance_page_rule()
-            
+
             # 2. Update site status
             self.site.disable_maintenance()
-            
+
             # 3. Log success
             duration = int(time.time() - start_time)
             log_entry.status = MaintenanceLog.Status.SUCCESS
@@ -96,9 +97,9 @@ class MaintenanceService:
                 'page_rule_delete': self._serialize_response(page_rule_response)
             }
             log_entry.save()
-            
+
             return log_entry
-            
+
         except Exception as e:
             # Log failure
             duration = int(time.time() - start_time)
@@ -106,20 +107,20 @@ class MaintenanceService:
             log_entry.error_message = str(e)
             log_entry.duration_seconds = duration
             log_entry.save()
-            
+
             raise e
-    
+
     def get_status(self) -> bool:
         """Get current maintenance status for site."""
         return self.site.maintenance_active
-    
+
     # Private helper methods
-    
+
     def _serialize_response(self, response: Dict[str, Any]) -> Dict[str, Any]:
         """Convert Cloudflare API response to JSON serializable format."""
         import json
         from datetime import datetime
-        
+
         def convert_datetime(obj):
             if isinstance(obj, datetime):
                 return obj.isoformat()
@@ -129,7 +130,7 @@ class MaintenanceService:
                 return [convert_datetime(item) for item in obj]
             else:
                 return obj
-        
+
         try:
             # Convert all datetime objects to ISO strings
             serializable = convert_datetime(response)
@@ -139,32 +140,32 @@ class MaintenanceService:
         except Exception:
             # If serialization fails, return simple success message
             return {"success": True, "serialization_error": True}
-    
+
     @retry_on_failure(max_retries=3, base_delay=1.0)
     def _create_maintenance_page_rule(self) -> Dict[str, Any]:
         """Create Page Rules to redirect all traffic to maintenance page (including subdomains)."""
         maintenance_url = self.site.get_maintenance_url()
         patterns = self.site.get_domain_patterns()
-        
+
         logger.info(f"Creating page rules for {self.site.domain}: {patterns} → {maintenance_url}")
-        
+
         # First, check if conflicting Page Rules already exist and delete them
         try:
             existing_rules = self.client.page_rules.list(zone_id=self.site.zone_id)
-            
+
             # Handle different API response formats
             if hasattr(existing_rules, 'result'):
                 rules = existing_rules.result
             else:
                 rules = existing_rules
-            
+
             # Look for conflicting rules with same patterns
             for rule in rules:
-                if (hasattr(rule, 'targets') and rule.targets and 
-                    len(rule.targets) > 0 and 
-                    hasattr(rule.targets[0], 'constraint') and 
+                if (hasattr(rule, 'targets') and rule.targets and
+                    len(rule.targets) > 0 and
+                    hasattr(rule.targets[0], 'constraint') and
                     hasattr(rule.targets[0].constraint, 'value')):
-                    
+
                     rule_pattern = rule.targets[0].constraint.value
                     if rule_pattern in patterns:
                         logger.info(f"Found conflicting page rule {rule.id} with pattern {rule_pattern}, deleting...")
@@ -173,11 +174,11 @@ class MaintenanceService:
                             pagerule_id=rule.id
                         )
                         logger.info(f"Deleted conflicting page rule {rule.id}")
-                    
+
         except Exception as e:
             logger.warning(f"Error checking existing page rules: {e}")
             # Continue with creation anyway
-        
+
         # Create Page Rules for each pattern
         created_rules = []
         for pattern in patterns:
@@ -207,71 +208,71 @@ class MaintenanceService:
                     "response": response.model_dump()
                 })
                 logger.info(f"Created page rule for pattern {pattern}")
-                
+
             except Exception as e:
                 logger.error(f"Failed to create page rule for pattern {pattern}: {e}")
                 # Continue with other patterns
-        
+
         return {
             "success": True,
             "patterns": patterns,
             "created_rules": created_rules,
             "total_rules": len(created_rules)
         }
-    
+
     @retry_on_failure(max_retries=3, base_delay=1.0)
     def _delete_maintenance_page_rule(self) -> Dict[str, Any]:
         """Delete maintenance Page Rules with retry logic (including subdomains)."""
         # Find the maintenance page rules
         page_rules_response = self.client.page_rules.list(zone_id=self.site.zone_id)
-        
+
         # Handle different API response formats
         if hasattr(page_rules_response, 'result'):
             page_rules = page_rules_response.result
         else:
             page_rules = page_rules_response
-        
+
         patterns = self.site.get_domain_patterns()
         maintenance_url = self.site.get_maintenance_url()
-        
+
         logger.info(f"Looking for page rules to delete: patterns={patterns}, url={maintenance_url}")
         logger.info(f"Found {len(page_rules)} page rules total")
-        
+
         deleted_rules = []
-        
+
         for rule in page_rules:
             logger.info(f"Checking rule {rule.id}: targets={getattr(rule, 'targets', None)}, actions={getattr(rule, 'actions', None)}")
-            
+
             # Check if this rule matches our maintenance patterns
             rule_matches = False
-            
+
             # Check by pattern
-            if (hasattr(rule, 'targets') and rule.targets and 
-                len(rule.targets) > 0 and 
-                hasattr(rule.targets[0], 'constraint') and 
+            if (hasattr(rule, 'targets') and rule.targets and
+                len(rule.targets) > 0 and
+                hasattr(rule.targets[0], 'constraint') and
                 hasattr(rule.targets[0].constraint, 'value')):
-                
+
                 rule_pattern = rule.targets[0].constraint.value
                 if rule_pattern in patterns:
                     rule_matches = True
                     logger.info(f"Rule {rule.id} matches pattern: {rule_pattern}")
-            
+
             # Also check by URL (fallback for older rules)
-            if not rule_matches and (hasattr(rule, 'actions') and rule.actions and 
-                len(rule.actions) > 0 and 
-                hasattr(rule.actions[0], 'id') and 
+            if not rule_matches and (hasattr(rule, 'actions') and rule.actions and
+                len(rule.actions) > 0 and
+                hasattr(rule.actions[0], 'id') and
                 rule.actions[0].id == "forwarding_url"):
-                
+
                 action_value = getattr(rule.actions[0], 'value', {})
                 action_url = getattr(action_value, 'url', '')
-                
+
                 logger.info(f"Found forwarding rule with URL: {action_url}")
-                
-                if ("maintenance.reforms.ai" in action_url or 
+
+                if ("maintenance.reforms.ai" in action_url or
                     "djangocfg.com/maintenance" in action_url):
                     rule_matches = True
                     logger.info(f"Rule {rule.id} matches maintenance URL")
-            
+
             # Delete matching rule
             if rule_matches:
                 try:
@@ -286,10 +287,10 @@ class MaintenanceService:
                         "response": response.model_dump()
                     })
                     logger.info(f"Successfully deleted page rule: {rule.id}")
-                    
+
                 except Exception as e:
                     logger.error(f"Failed to delete page rule {rule.id}: {e}")
-        
+
         if deleted_rules:
             logger.info(f"Deleted {len(deleted_rules)} maintenance page rules for {self.site.domain}")
             return {
@@ -300,7 +301,7 @@ class MaintenanceService:
         else:
             logger.warning(f"No maintenance page rules found for {self.site.domain}")
             return {"success": True, "message": "No page rules to delete"}
-    
+
 
 
 # Convenience functions for easy usage

django_cfg/apps/maintenance/services/scheduled_maintenance_service.py

@@ -5,24 +5,24 @@ Handles automatic execution of scheduled maintenance events.
 """
 
 import logging
-from typing import Dict, List, Any, Optional
 from datetime import datetime, timedelta
-from django.utils import timezone
+from typing import Any, Dict, List
+
 from django.db import transaction
+from django.utils import timezone
 
-from ..models import ScheduledMaintenance, CloudflareSite
-from .bulk_operations_service import bulk_operations
+from ..models import CloudflareSite, ScheduledMaintenance
 
 logger = logging.getLogger(__name__)
 
 
 class ScheduledMaintenanceService:
     """Service for managing scheduled maintenance events."""
-    
+
     def __init__(self):
         """Initialize scheduled maintenance service."""
         pass
-    
+
     def create_scheduled_maintenance(self,
                                    title: str,
                                    scheduled_start: datetime,
@@ -70,13 +70,13 @@ class ScheduledMaintenanceService:
                 notify_before=notify_before,
                 created_by=created_by
             )
-            
+
             # Add sites
             scheduled_maintenance.sites.set(sites)
-            
+
             logger.info(f"Created scheduled maintenance: {scheduled_maintenance}")
             return scheduled_maintenance
-    
+
     def process_due_maintenances(self) -> Dict[str, Any]:
         """
         Process all maintenance events that are due to start.
@@ -85,26 +85,26 @@ class ScheduledMaintenanceService:
             Dict with processing results
         """
         due_maintenances = ScheduledMaintenance.get_due_maintenances()
-        
+
         results = {
             'processed': 0,
             'successful': 0,
             'failed': 0,
             'details': []
         }
-        
+
         for maintenance in due_maintenances:
             if maintenance.auto_enable:
                 try:
                     result = maintenance.start_maintenance()
-                    
+
                     if result['success']:
                         results['successful'] += 1
                         logger.info(f"Started scheduled maintenance: {maintenance.title}")
                     else:
                         results['failed'] += 1
                         logger.error(f"Failed to start maintenance {maintenance.title}: {result.get('error')}")
-                    
+
                     results['details'].append({
                         'maintenance_id': maintenance.id,
                         'title': maintenance.title,
@@ -112,25 +112,25 @@ class ScheduledMaintenanceService:
                         'sites_affected': result.get('sites_affected', 0),
                         'error': result.get('error')
                     })
-                    
+
                 except Exception as e:
                     results['failed'] += 1
                     logger.error(f"Exception starting maintenance {maintenance.title}: {e}")
-                    
+
                     results['details'].append({
                         'maintenance_id': maintenance.id,
                         'title': maintenance.title,
                         'success': False,
                         'error': str(e)
                     })
-                
+
                 results['processed'] += 1
-        
+
         if results['processed'] > 0:
             logger.info(f"Processed {results['processed']} due maintenances: {results['successful']} successful, {results['failed']} failed")
-        
+
         return results
-    
+
     def process_overdue_maintenances(self) -> Dict[str, Any]:
         """
         Process all maintenance events that should have ended.
@@ -139,26 +139,26 @@ class ScheduledMaintenanceService:
             Dict with processing results
         """
         overdue_maintenances = ScheduledMaintenance.get_overdue_maintenances()
-        
+
         results = {
             'processed': 0,
             'successful': 0,
             'failed': 0,
             'details': []
         }
-        
+
         for maintenance in overdue_maintenances:
             if maintenance.auto_disable:
                 try:
                     result = maintenance.complete_maintenance()
-                    
+
                     if result['success']:
                         results['successful'] += 1
                         logger.info(f"Completed overdue maintenance: {maintenance.title}")
                     else:
                         results['failed'] += 1
                         logger.error(f"Failed to complete maintenance {maintenance.title}: {result.get('error')}")
-                    
+
                     results['details'].append({
                         'maintenance_id': maintenance.id,
                         'title': maintenance.title,
@@ -167,25 +167,25 @@ class ScheduledMaintenanceService:
                         'actual_duration': result.get('actual_duration'),
                         'error': result.get('error')
                     })
-                    
+
                 except Exception as e:
                     results['failed'] += 1
                     logger.error(f"Exception completing maintenance {maintenance.title}: {e}")
-                    
+
                     results['details'].append({
                         'maintenance_id': maintenance.id,
                         'title': maintenance.title,
                         'success': False,
                         'error': str(e)
                     })
-                
+
                 results['processed'] += 1
-        
+
         if results['processed'] > 0:
             logger.info(f"Processed {results['processed']} overdue maintenances: {results['successful']} successful, {results['failed']} failed")
-        
+
         return results
-    
+
     def get_upcoming_maintenances(self, hours: int = 24) -> List[Dict[str, Any]]:
         """
         Get upcoming maintenance events.
@@ -197,7 +197,7 @@ class ScheduledMaintenanceService:
             List of upcoming maintenance info
         """
         upcoming = ScheduledMaintenance.get_upcoming_maintenances(hours=hours)
-        
+
         return [
             {
                 'id': maintenance.id,
@@ -212,13 +212,13 @@ class ScheduledMaintenanceService:
             }
             for maintenance in upcoming
         ]
-    
+
     def get_active_maintenances(self) -> List[Dict[str, Any]]:
         """Get currently active maintenance events."""
         active = ScheduledMaintenance.objects.filter(
             status=ScheduledMaintenance.Status.ACTIVE
         )
-        
+
         return [
             {
                 'id': maintenance.id,
@@ -233,7 +233,7 @@ class ScheduledMaintenanceService:
             }
             for maintenance in active
         ]
-    
+
     def get_maintenance_calendar(self, days: int = 30) -> Dict[str, List[Dict[str, Any]]]:
         """
         Get maintenance calendar for specified days.
@@ -245,20 +245,20 @@ class ScheduledMaintenanceService:
             Dict with dates as keys and maintenance events as values
         """
         end_date = timezone.now() + timedelta(days=days)
-        
+
         maintenances = ScheduledMaintenance.objects.filter(
             scheduled_start__gte=timezone.now(),
             scheduled_start__lte=end_date
         ).order_by('scheduled_start')
-        
+
         calendar = {}
-        
+
         for maintenance in maintenances:
             date_key = maintenance.scheduled_start.date().isoformat()
-            
+
             if date_key not in calendar:
                 calendar[date_key] = []
-            
+
             calendar[date_key].append({
                 'id': maintenance.id,
                 'title': maintenance.title,
@@ -268,9 +268,9 @@ class ScheduledMaintenanceService:
                 'priority': maintenance.priority,
                 'status': maintenance.status
             })
-        
+
         return calendar
-    
+
     def bulk_schedule_maintenance(self,
                                  title_template: str,
                                  sites_groups: List[List[CloudflareSite]],
@@ -292,12 +292,12 @@ class ScheduledMaintenanceService:
         """
         if len(sites_groups) != len(start_times):
             raise ValueError("sites_groups and start_times must have same length")
-        
+
         created_maintenances = []
-        
+
         for i, (sites, start_time) in enumerate(zip(sites_groups, start_times)):
             title = title_template.format(index=i+1, start_time=start_time.strftime('%Y-%m-%d %H:%M'))
-            
+
             maintenance = self.create_scheduled_maintenance(
                 title=title,
                 scheduled_start=start_time,
@@ -305,13 +305,13 @@ class ScheduledMaintenanceService:
                 sites=sites,
                 **kwargs
             )
-            
+
             created_maintenances.append(maintenance)
-        
+
         logger.info(f"Bulk created {len(created_maintenances)} scheduled maintenances")
         return created_maintenances
-    
-    def cancel_conflicting_maintenances(self, 
+
+    def cancel_conflicting_maintenances(self,
                                        new_start: datetime,
                                        new_end: datetime,
                                        sites: List[CloudflareSite],
@@ -329,7 +329,7 @@ class ScheduledMaintenanceService:
             List of cancelled maintenance events
         """
         site_ids = [site.id for site in sites]
-        
+
         # Find conflicting maintenances
         conflicting = ScheduledMaintenance.objects.filter(
             status__in=[ScheduledMaintenance.Status.SCHEDULED, ScheduledMaintenance.Status.ACTIVE],
@@ -337,15 +337,15 @@ class ScheduledMaintenanceService:
             scheduled_start__lt=new_end,
             scheduled_end__gt=new_start
         ).distinct()
-        
+
         cancelled = []
-        
+
         for maintenance in conflicting:
             result = maintenance.cancel_maintenance(reason=reason)
             if result['success']:
                 cancelled.append(maintenance)
                 logger.info(f"Cancelled conflicting maintenance: {maintenance.title}")
-        
+
         return cancelled
 
 
@@ -373,7 +373,7 @@ def process_scheduled_maintenances() -> Dict[str, Any]:
     """Process all due and overdue maintenances."""
     due_results = scheduled_maintenance_service.process_due_maintenances()
     overdue_results = scheduled_maintenance_service.process_overdue_maintenances()
-    
+
     return {
         'due_maintenances': due_results,
         'overdue_maintenances': overdue_results,