7f8982a0ab
- Introduced a new scheduled task for executing automation rules every 5 minutes in the Celery beat schedule. - Updated URL routing to include a new endpoint for automation-related functionalities. - Refactored imports in various modules to align with the new business layer structure, ensuring backward compatibility for billing models, exceptions, and services.
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
"""
|
|
Automation Models
|
|
Phase 2: Automation System
|
|
"""
|
|
from django.db import models
|
|
from django.core.validators import MinValueValidator
|
|
from igny8_core.auth.models import SiteSectorBaseModel, AccountBaseModel
|
|
import json
|
|
|
|
|
|
class AutomationRule(SiteSectorBaseModel):
|
|
"""
|
|
Automation Rule model for defining automated workflows.
|
|
|
|
Rules can be triggered by:
|
|
- schedule: Time-based triggers (cron-like)
|
|
- event: Event-based triggers (content created, keyword added, etc.)
|
|
- manual: Manual execution only
|
|
"""
|
|
|
|
TRIGGER_CHOICES = [
|
|
('schedule', 'Schedule'),
|
|
('event', 'Event'),
|
|
('manual', 'Manual'),
|
|
]
|
|
|
|
STATUS_CHOICES = [
|
|
('active', 'Active'),
|
|
('inactive', 'Inactive'),
|
|
('paused', 'Paused'),
|
|
]
|
|
|
|
name = models.CharField(max_length=255, help_text="Rule name")
|
|
description = models.TextField(blank=True, null=True, help_text="Rule description")
|
|
|
|
# Trigger configuration
|
|
trigger = models.CharField(max_length=50, choices=TRIGGER_CHOICES, default='manual')
|
|
|
|
# Schedule configuration (for schedule triggers)
|
|
# Stored as cron-like string: "0 0 * * *" (daily at midnight)
|
|
schedule = models.CharField(
|
|
max_length=100,
|
|
blank=True,
|
|
null=True,
|
|
help_text="Cron-like schedule string (e.g., '0 0 * * *' for daily at midnight)"
|
|
)
|
|
|
|
# Conditions (JSON field)
|
|
# Format: [{"field": "content.status", "operator": "equals", "value": "draft"}, ...]
|
|
conditions = models.JSONField(
|
|
default=list,
|
|
help_text="List of conditions that must be met for rule to execute"
|
|
)
|
|
|
|
# Actions (JSON field)
|
|
# Format: [{"type": "generate_content", "params": {...}}, ...]
|
|
actions = models.JSONField(
|
|
default=list,
|
|
help_text="List of actions to execute when rule triggers"
|
|
)
|
|
|
|
# Status
|
|
is_active = models.BooleanField(default=True, help_text="Whether rule is active")
|
|
status = models.CharField(max_length=50, choices=STATUS_CHOICES, default='active')
|
|
|
|
# Execution tracking
|
|
last_executed_at = models.DateTimeField(null=True, blank=True)
|
|
execution_count = models.IntegerField(default=0, validators=[MinValueValidator(0)])
|
|
|
|
# Metadata
|
|
metadata = models.JSONField(default=dict, help_text="Additional metadata")
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = 'igny8_automation_rules'
|
|
ordering = ['-created_at']
|
|
verbose_name = 'Automation Rule'
|
|
verbose_name_plural = 'Automation Rules'
|
|
indexes = [
|
|
models.Index(fields=['trigger', 'is_active']),
|
|
models.Index(fields=['status']),
|
|
models.Index(fields=['site', 'sector']),
|
|
models.Index(fields=['trigger', 'is_active', 'status']),
|
|
]
|
|
|
|
def __str__(self):
|
|
return f"{self.name} ({self.get_trigger_display()})"
|
|
|
|
|
|
class ScheduledTask(AccountBaseModel):
|
|
"""
|
|
Scheduled Task model for tracking scheduled automation rule executions.
|
|
"""
|
|
|
|
STATUS_CHOICES = [
|
|
('pending', 'Pending'),
|
|
('running', 'Running'),
|
|
('completed', 'Completed'),
|
|
('failed', 'Failed'),
|
|
('cancelled', 'Cancelled'),
|
|
]
|
|
|
|
automation_rule = models.ForeignKey(
|
|
AutomationRule,
|
|
on_delete=models.CASCADE,
|
|
related_name='scheduled_tasks',
|
|
help_text="The automation rule this task belongs to"
|
|
)
|
|
|
|
scheduled_at = models.DateTimeField(help_text="When the task is scheduled to run")
|
|
executed_at = models.DateTimeField(null=True, blank=True, help_text="When the task was actually executed")
|
|
|
|
status = models.CharField(max_length=50, choices=STATUS_CHOICES, default='pending')
|
|
|
|
# Execution results
|
|
result = models.JSONField(default=dict, help_text="Execution result data")
|
|
error_message = models.TextField(blank=True, null=True, help_text="Error message if execution failed")
|
|
|
|
# Metadata
|
|
metadata = models.JSONField(default=dict, help_text="Additional metadata")
|
|
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = 'igny8_scheduled_tasks'
|
|
ordering = ['-scheduled_at']
|
|
verbose_name = 'Scheduled Task'
|
|
verbose_name_plural = 'Scheduled Tasks'
|
|
indexes = [
|
|
models.Index(fields=['automation_rule', 'status']),
|
|
models.Index(fields=['scheduled_at', 'status']),
|
|
models.Index(fields=['account', 'status']),
|
|
models.Index(fields=['status', 'scheduled_at']),
|
|
]
|
|
|
|
def __str__(self):
|
|
return f"Scheduled task for {self.automation_rule.name} at {self.scheduled_at}"
|
|
|