Automation Part 1

backend/igny8_core/business/automation/models.py

@@ -0,0 +1,104 @@
+"""
+Automation Models
+Tracks automation runs and configuration
+"""
+from django.db import models
+from django.utils import timezone
+from igny8_core.modules.system.models import Account, Site
+
+
+class AutomationConfig(models.Model):
+    """Per-site automation configuration"""
+    
+    FREQUENCY_CHOICES = [
+        ('daily', 'Daily'),
+        ('weekly', 'Weekly'),
+        ('monthly', 'Monthly'),
+    ]
+    
+    account = models.ForeignKey(Account, on_delete=models.CASCADE, related_name='automation_configs')
+    site = models.OneToOneField(Site, on_delete=models.CASCADE, related_name='automation_config')
+    
+    is_enabled = models.BooleanField(default=False, help_text="Whether scheduled automation is active")
+    frequency = models.CharField(max_length=20, choices=FREQUENCY_CHOICES, default='daily')
+    scheduled_time = models.TimeField(default='02:00', help_text="Time to run (e.g., 02:00)")
+    
+    # Batch sizes per stage
+    stage_1_batch_size = models.IntegerField(default=20, help_text="Keywords per batch")
+    stage_2_batch_size = models.IntegerField(default=1, help_text="Clusters at a time")
+    stage_3_batch_size = models.IntegerField(default=20, help_text="Ideas per batch")
+    stage_4_batch_size = models.IntegerField(default=1, help_text="Tasks - sequential")
+    stage_5_batch_size = models.IntegerField(default=1, help_text="Content at a time")
+    stage_6_batch_size = models.IntegerField(default=1, help_text="Images - sequential")
+    
+    last_run_at = models.DateTimeField(null=True, blank=True)
+    next_run_at = models.DateTimeField(null=True, blank=True, help_text="Calculated based on frequency")
+    
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    
+    class Meta:
+        db_table = 'igny8_automation_configs'
+        verbose_name = 'Automation Config'
+        verbose_name_plural = 'Automation Configs'
+        indexes = [
+            models.Index(fields=['is_enabled', 'next_run_at']),
+            models.Index(fields=['account', 'site']),
+        ]
+    
+    def __str__(self):
+        return f"Automation Config: {self.site.domain} ({self.frequency})"
+
+
+class AutomationRun(models.Model):
+    """Tracks each automation execution"""
+    
+    TRIGGER_TYPE_CHOICES = [
+        ('manual', 'Manual'),
+        ('scheduled', 'Scheduled'),
+    ]
+    
+    STATUS_CHOICES = [
+        ('running', 'Running'),
+        ('paused', 'Paused'),
+        ('completed', 'Completed'),
+        ('failed', 'Failed'),
+    ]
+    
+    run_id = models.CharField(max_length=100, unique=True, db_index=True, help_text="Format: run_20251203_140523_manual")
+    account = models.ForeignKey(Account, on_delete=models.CASCADE, related_name='automation_runs')
+    site = models.ForeignKey(Site, on_delete=models.CASCADE, related_name='automation_runs')
+    
+    trigger_type = models.CharField(max_length=20, choices=TRIGGER_TYPE_CHOICES)
+    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='running', db_index=True)
+    current_stage = models.IntegerField(default=1, help_text="Current stage number (1-7)")
+    
+    started_at = models.DateTimeField(auto_now_add=True, db_index=True)
+    completed_at = models.DateTimeField(null=True, blank=True)
+    
+    total_credits_used = models.IntegerField(default=0)
+    
+    # JSON results per stage
+    stage_1_result = models.JSONField(null=True, blank=True, help_text="{keywords_processed, clusters_created, batches}")
+    stage_2_result = models.JSONField(null=True, blank=True, help_text="{clusters_processed, ideas_created}")
+    stage_3_result = models.JSONField(null=True, blank=True, help_text="{ideas_processed, tasks_created}")
+    stage_4_result = models.JSONField(null=True, blank=True, help_text="{tasks_processed, content_created, total_words}")
+    stage_5_result = models.JSONField(null=True, blank=True, help_text="{content_processed, prompts_created}")
+    stage_6_result = models.JSONField(null=True, blank=True, help_text="{images_processed, images_generated}")
+    stage_7_result = models.JSONField(null=True, blank=True, help_text="{ready_for_review}")
+    
+    error_message = models.TextField(null=True, blank=True)
+    
+    class Meta:
+        db_table = 'igny8_automation_runs'
+        verbose_name = 'Automation Run'
+        verbose_name_plural = 'Automation Runs'
+        ordering = ['-started_at']
+        indexes = [
+            models.Index(fields=['site', '-started_at']),
+            models.Index(fields=['status', '-started_at']),
+            models.Index(fields=['account', '-started_at']),
+        ]
+    
+    def __str__(self):
+        return f"{self.run_id} - {self.site.domain} ({self.status})"