# ops/automation/framework.py
import functools
import logging
from typing import Callable, TypeVar
from datetime import datetime

logger = logging.getLogger(__name__)
T = TypeVar('T')

def automation_task(
    name: str,
    idempotent: bool = True,
    notify_on_failure: bool = True,
    dry_run_supported: bool = True
):
    """
    Decorator for automation tasks.
    Adds: logging, dry-run support, failure notification, audit trail.
    """
    def decorator(func: Callable[..., T]) -> Callable[..., T]:
        @functools.wraps(func)
        def wrapper(*args, dry_run: bool = False, **kwargs) -> T:
            start = datetime.utcnow()
            logger.info(f"Starting automation task: {name}",
                       extra={"dry_run": dry_run, "task": name})

            if dry_run and not dry_run_supported:
                logger.info(f"Dry run not supported for {name}, skipping")
                return None

            try:
                if dry_run:
                    logger.info(f"DRY RUN: Would execute {name}")
                    return None

                result = func(*args, **kwargs)

                duration = (datetime.utcnow() - start).total_seconds()
                logger.info(f"Automation task completed: {name}",
                           extra={"duration_seconds": duration, "task": name})

                # Audit trail
                record_automation_run(
                    task=name,
                    status="success",
                    duration=duration,
                    args=str(args)
                )

                return result

            except Exception as e:
                logger.error(f"Automation task failed: {name}",
                           exc_info=True,
                           extra={"task": name, "error": str(e)})

                if notify_on_failure:
                    notify_slack(
                        channel="#ops-automation-failures",
                        message=f"Automation task '{name}' failed: {e}"
                    )
                raise

        return wrapper
    return decorator


# Usage
@automation_task(name="customer-onboarding", idempotent=True)
def onboard_customer(customer_id: str, plan: str, domain: str):
    """
    Fully automated customer onboarding.
    Was: 45 minutes manual → Now: 2 minutes automated.
    """
    provision_kubernetes_namespace(customer_id)
    provision_database(customer_id, plan)
    configure_environment(customer_id, plan, domain)
    configure_nginx(domain, customer_id)
    setup_monitoring_dashboards(customer_id)
    send_welcome_email(customer_id)

    logger.info("Customer onboarded successfully",
               extra={"customer_id": customer_id, "plan": plan})
Toil Reduction — אוטומציה וצמצום עבודה ידנית

Toil Reduction — אוטומציה וצמצום עבודה ידנית

Toil Reduction — אוטומציה וצמצום עבודה ידנית
