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

django_cfg/apps/maintenance/models/maintenance_log.py

@@ -5,6 +5,7 @@ Simple log of all maintenance operations.
 """
 
 from django.db import models
+
 from .cloudflare_site import CloudflareSite
 
 
@@ -14,18 +15,18 @@ class MaintenanceLog(models.Model):
     
     Tracks enable/disable operations with success/failure status.
     """
-    
+
     class Action(models.TextChoices):
         ENABLE = "enable", "Enable Maintenance"
         DISABLE = "disable", "Disable Maintenance"
         ERROR = "error", "Error"
         SYNC = "sync", "Sync from Cloudflare"
-    
+
     class Status(models.TextChoices):
         SUCCESS = "success", "Success"
         FAILED = "failed", "Failed"
         PENDING = "pending", "Pending"
-    
+
     # Related site
     site = models.ForeignKey(
         CloudflareSite,
@@ -33,7 +34,7 @@ class MaintenanceLog(models.Model):
         related_name='logs',
         help_text="Site this log entry belongs to"
     )
-    
+
     # Operation details
     action = models.CharField(
         max_length=20,
@@ -45,7 +46,7 @@ class MaintenanceLog(models.Model):
         choices=Status.choices,
         help_text="Result of the operation"
     )
-    
+
     # Additional info
     reason = models.TextField(
         blank=True,
@@ -60,7 +61,7 @@ class MaintenanceLog(models.Model):
         blank=True,
         help_text="Full Cloudflare API response for debugging"
     )
-    
+
     # Timing
     created_at = models.DateTimeField(auto_now_add=True)
     duration_seconds = models.IntegerField(
@@ -68,11 +69,11 @@ class MaintenanceLog(models.Model):
         blank=True,
         help_text="How long the operation took"
     )
-    
+
     # Custom manager
     from ..managers.maintenance_log_manager import MaintenanceLogManager
     objects = MaintenanceLogManager()
-    
+
     class Meta:
         ordering = ['-created_at']
         verbose_name = "Maintenance Log"
@@ -82,18 +83,18 @@ class MaintenanceLog(models.Model):
             models.Index(fields=['action', '-created_at']),
             models.Index(fields=['status', '-created_at']),
         ]
-    
+
     def __str__(self) -> str:
         status_emoji = {
             self.Status.SUCCESS: "✅",
             self.Status.FAILED: "❌",
             self.Status.PENDING: "⏳"
         }.get(self.status, "❓")
-        
+
         return f"{status_emoji} {self.get_action_display()} - {self.site.domain}"
-    
+
     @classmethod
-    def log_success(cls, site: CloudflareSite, action: str, reason: str = "", 
+    def log_success(cls, site: CloudflareSite, action: str, reason: str = "",
                    duration_seconds: int = None, cloudflare_response: dict = None) -> 'MaintenanceLog':
         """Log successful operation."""
         return cls.objects.create(
@@ -104,7 +105,7 @@ class MaintenanceLog(models.Model):
             duration_seconds=duration_seconds,
             cloudflare_response=cloudflare_response
         )
-    
+
     @classmethod
     def log_failure(cls, site: CloudflareSite, action: str, error_message: str,
                    reason: str = "", cloudflare_response: dict = None) -> 'MaintenanceLog':
@@ -117,7 +118,7 @@ class MaintenanceLog(models.Model):
             error_message=error_message,
             cloudflare_response=cloudflare_response
         )
-    
+
     @classmethod
     def log_pending(cls, site: CloudflareSite, action: str, reason: str = "") -> 'MaintenanceLog':
         """Log pending operation."""

django_cfg/apps/maintenance/models/scheduled_maintenance.py

@@ -4,49 +4,50 @@ Scheduled maintenance model.
 Allows planning and automatic execution of maintenance windows.
 """
 
+from datetime import timedelta
+from typing import Any, Dict, Optional
+
+from django.core.exceptions import ValidationError
 from django.db import models
 from django.utils import timezone
-from django.core.exceptions import ValidationError
-from datetime import datetime, timedelta
-from typing import Optional, Dict, Any
 
 
 class ScheduledMaintenance(models.Model):
     """Scheduled maintenance events for sites."""
-    
+
     class Status(models.TextChoices):
         SCHEDULED = "scheduled", "Scheduled"
         ACTIVE = "active", "Active"
         COMPLETED = "completed", "Completed"
         CANCELLED = "cancelled", "Cancelled"
         FAILED = "failed", "Failed"
-    
+
     class Priority(models.TextChoices):
         LOW = "low", "Low"
         NORMAL = "normal", "Normal"
         HIGH = "high", "High"
         CRITICAL = "critical", "Critical"
-    
+
     # Basic Information
     title = models.CharField(max_length=200, help_text="Maintenance event title")
     description = models.TextField(blank=True, help_text="Detailed description of maintenance")
-    
+
     # Scheduling
     scheduled_start = models.DateTimeField(help_text="When maintenance should start")
     estimated_duration = models.DurationField(help_text="Expected duration of maintenance")
     scheduled_end = models.DateTimeField(editable=False, help_text="Auto-calculated end time")
-    
+
     # Target Sites
     sites = models.ManyToManyField(
-        'CloudflareSite', 
+        'CloudflareSite',
         related_name='scheduled_maintenances',
         help_text="Sites affected by this maintenance"
     )
-    
+
     # Execution Details
     status = models.CharField(
-        max_length=20, 
-        choices=Status.choices, 
+        max_length=20,
+        choices=Status.choices,
         default=Status.SCHEDULED
     )
     priority = models.CharField(
@@ -54,10 +55,10 @@ class ScheduledMaintenance(models.Model):
         choices=Priority.choices,
         default=Priority.NORMAL
     )
-    
+
     actual_start = models.DateTimeField(null=True, blank=True)
     actual_end = models.DateTimeField(null=True, blank=True)
-    
+
     # Configuration
     maintenance_message = models.TextField(
         blank=True,
@@ -74,17 +75,17 @@ class ScheduledMaintenance(models.Model):
         ],
         help_text="Maintenance page template"
     )
-    
+
     # Automation Settings
     auto_enable = models.BooleanField(
-        default=True, 
+        default=True,
         help_text="Automatically enable maintenance at scheduled time"
     )
     auto_disable = models.BooleanField(
-        default=True, 
+        default=True,
         help_text="Automatically disable maintenance after duration"
     )
-    
+
     # Notifications
     notify_before = models.DurationField(
         default=timedelta(hours=1),
@@ -92,19 +93,19 @@ class ScheduledMaintenance(models.Model):
     )
     notify_on_start = models.BooleanField(default=True)
     notify_on_complete = models.BooleanField(default=True)
-    
+
     # Metadata
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
     created_by = models.CharField(max_length=100, blank=True, help_text="Who created this maintenance")
-    
+
     # Execution Results
     execution_log = models.JSONField(
         default=dict,
         blank=True,
         help_text="Log of execution steps and results"
     )
-    
+
     class Meta:
         ordering = ['scheduled_start']
         verbose_name = "Scheduled Maintenance"
@@ -114,22 +115,22 @@ class ScheduledMaintenance(models.Model):
             models.Index(fields=['scheduled_start']),
             models.Index(fields=['status']),
         ]
-    
+
     def save(self, *args, **kwargs):
         """Auto-calculate scheduled_end."""
         if self.scheduled_start and self.estimated_duration:
             self.scheduled_end = self.scheduled_start + self.estimated_duration
         super().save(*args, **kwargs)
-    
+
     def clean(self):
         """Validate the scheduled maintenance."""
         if self.scheduled_start and self.scheduled_start <= timezone.now():
             if self.status == self.Status.SCHEDULED:
                 raise ValidationError("Scheduled start time must be in the future")
-        
+
         if self.estimated_duration and self.estimated_duration <= timedelta(0):
             raise ValidationError("Estimated duration must be positive")
-    
+
     def __str__(self):
         status_emoji = {
             self.Status.SCHEDULED: "📅",
@@ -138,14 +139,14 @@ class ScheduledMaintenance(models.Model):
             self.Status.CANCELLED: "❌",
             self.Status.FAILED: "💥"
         }.get(self.status, "❓")
-        
+
         return f"{status_emoji} {self.title} - {self.scheduled_start.strftime('%Y-%m-%d %H:%M')}"
-    
+
     @property
     def affected_sites_count(self) -> int:
         """Count total affected sites."""
         return self.sites.count()
-    
+
     @property
     def is_due(self) -> bool:
         """Check if maintenance is due to start."""
@@ -153,7 +154,7 @@ class ScheduledMaintenance(models.Model):
             self.status == self.Status.SCHEDULED and
             timezone.now() >= self.scheduled_start
         )
-    
+
     @property
     def is_overdue(self) -> bool:
         """Check if maintenance should have ended."""
@@ -161,28 +162,28 @@ class ScheduledMaintenance(models.Model):
             self.status == self.Status.ACTIVE and
             timezone.now() >= self.scheduled_end
         )
-    
+
     @property
     def time_until_start(self) -> Optional[timedelta]:
         """Time until maintenance starts."""
         if self.status == self.Status.SCHEDULED:
             return self.scheduled_start - timezone.now()
         return None
-    
+
     @property
     def time_until_end(self) -> Optional[timedelta]:
         """Time until maintenance ends."""
         if self.status == self.Status.ACTIVE:
             return self.scheduled_end - timezone.now()
         return None
-    
+
     @property
     def actual_duration(self) -> Optional[timedelta]:
         """Actual duration of maintenance."""
         if self.actual_start and self.actual_end:
             return self.actual_end - self.actual_start
         return None
-    
+
     def start_maintenance(self) -> Dict[str, Any]:
         """Start the scheduled maintenance."""
         if self.status != self.Status.SCHEDULED:
@@ -190,21 +191,21 @@ class ScheduledMaintenance(models.Model):
                 'success': False,
                 'error': f'Cannot start maintenance in {self.status} status'
             }
-        
+
         from ..services.bulk_operations_service import bulk_operations
-        
+
         # Update status
         self.status = self.Status.ACTIVE
         self.actual_start = timezone.now()
         self.save()
-        
+
         # Enable maintenance for all sites
         sites_queryset = bulk_operations.sites(self.sites.all())
         result = sites_queryset.enable_maintenance(
             reason=self.maintenance_message or f"Scheduled maintenance: {self.title}",
             template=self.template
         )
-        
+
         # Log execution
         self.execution_log['start'] = {
             'timestamp': timezone.now().isoformat(),
@@ -212,14 +213,14 @@ class ScheduledMaintenance(models.Model):
             'sites_affected': result['total']
         }
         self.save()
-        
+
         return {
             'success': True,
             'sites_affected': result['total'],
             'successful': len(result['successful']),
             'failed': len(result['failed'])
         }
-    
+
     def complete_maintenance(self) -> Dict[str, Any]:
         """Complete the scheduled maintenance."""
         if self.status != self.Status.ACTIVE:
@@ -227,17 +228,17 @@ class ScheduledMaintenance(models.Model):
                 'success': False,
                 'error': f'Cannot complete maintenance in {self.status} status'
             }
-        
+
         from ..services.bulk_operations_service import bulk_operations
-        
+
         # Disable maintenance for all sites
         sites_queryset = bulk_operations.sites(self.sites.all())
         result = sites_queryset.disable_maintenance()
-        
+
         # Update status
         self.status = self.Status.COMPLETED
         self.actual_end = timezone.now()
-        
+
         # Log execution
         self.execution_log['complete'] = {
             'timestamp': timezone.now().isoformat(),
@@ -245,7 +246,7 @@ class ScheduledMaintenance(models.Model):
             'sites_affected': result['total']
         }
         self.save()
-        
+
         return {
             'success': True,
             'sites_affected': result['total'],
@@ -253,7 +254,7 @@ class ScheduledMaintenance(models.Model):
             'failed': len(result['failed']),
             'actual_duration': self.actual_duration.total_seconds() if self.actual_duration else None
         }
-    
+
     def cancel_maintenance(self, reason: str = "") -> Dict[str, Any]:
         """Cancel the scheduled maintenance."""
         if self.status in [self.Status.COMPLETED, self.Status.CANCELLED]:
@@ -261,26 +262,26 @@ class ScheduledMaintenance(models.Model):
                 'success': False,
                 'error': f'Cannot cancel maintenance in {self.status} status'
             }
-        
+
         # If active, disable maintenance first
         if self.status == self.Status.ACTIVE:
             self.complete_maintenance()
-        
+
         # Update status
         self.status = self.Status.CANCELLED
-        
+
         # Log cancellation
         self.execution_log['cancelled'] = {
             'timestamp': timezone.now().isoformat(),
             'reason': reason
         }
         self.save()
-        
+
         return {
             'success': True,
             'reason': reason
         }
-    
+
     @classmethod
     def get_due_maintenances(cls):
         """Get maintenances that are due to start."""
@@ -288,7 +289,7 @@ class ScheduledMaintenance(models.Model):
             status=cls.Status.SCHEDULED,
             scheduled_start__lte=timezone.now()
         )
-    
+
     @classmethod
     def get_overdue_maintenances(cls):
         """Get active maintenances that should have ended."""
@@ -296,7 +297,7 @@ class ScheduledMaintenance(models.Model):
             status=cls.Status.ACTIVE,
             scheduled_end__lte=timezone.now()
         )
-    
+
     @classmethod
     def get_upcoming_maintenances(cls, hours: int = 24):
         """Get maintenances starting within specified hours."""

django_cfg/apps/maintenance/services/__init__.py

@@ -11,7 +11,7 @@ _LAZY_IMPORTS = {
     'SiteSyncService': ('django_cfg.apps.maintenance.services.site_sync_service', 'SiteSyncService'),
     'BulkOperationsService': ('django_cfg.apps.maintenance.services.bulk_operations_service', 'BulkOperationsService'),
     'ScheduledMaintenanceService': ('django_cfg.apps.maintenance.services.scheduled_maintenance_service', 'ScheduledMaintenanceService'),
-    
+
     # Service functions
     'enable_maintenance_for_domain': ('django_cfg.apps.maintenance.services.maintenance_service', 'enable_maintenance_for_domain'),
     'disable_maintenance_for_domain': ('django_cfg.apps.maintenance.services.maintenance_service', 'disable_maintenance_for_domain'),
@@ -31,11 +31,11 @@ def __getattr__(name: str):
     """Lazy import mechanism to avoid Django initialization issues."""
     if name in _LAZY_IMPORTS:
         module_path, attr_name = _LAZY_IMPORTS[name]
-        
+
         import importlib
         module = importlib.import_module(module_path)
         return getattr(module, attr_name)
-    
+
     raise AttributeError(f"module '{__name__}' has no attribute '{name}'")