from celery.schedules import crontab

beat_schedule = {
    "cfo-sync-1c": {
        "task": "cfo.tasks.sync_1c_data",
        "schedule": crontab(minute="*/30"),
        "options": {"queue": "default"}
    },
    "cfo-cleanup": {
        "task": "cfo.tasks.cleanup_old_reports",
        "schedule": crontab(hour=3, minute=0),
        "options": {"queue": "default"}
    },
    "cfo-office-heartbeat": {
        "task": "cfo.tasks.office_heartbeat",
        "schedule": 60.0,
        "options": {"queue": "default"}
    },
}

from celery import shared_task
from app.utils.office_reporter import OfficeReporter

reporter = OfficeReporter(
    agent_id="cfo_report",
    department="cfo",
    name="Отчёт P&L",
    salary_equivalent=80000,
    fte_coefficient=1.0
)

@shared_task(name='cfo.tasks.sync_1c_data')
def sync_1c_data():
    """Синхронизация данных из 1С."""
    
    reporter.report_working("Синхронизация с 1С")
    
    try:
        # ... логика синхронизации ...
        records_synced = fetch_1c_data()
        
        reporter.report_idle(metrics={
            "reports_generated_today": get_daily_reports_count(),
            "last_sync": datetime.utcnow().isoformat()
        })
        
        return {"success": True, "synced": records_synced}
        
    except Exception as e:
        reporter.report_error(f"1C sync: {e}")
        raise

@shared_task(
    name='cfo.tasks.generate_pnl_report',
    bind=True,
    time_limit=600
)
def generate_pnl_report(self, period: str, brand: str):
    """Генерация отчёта P&L."""
    
    reporter.report_working(f"Генерация P&L за {period}")
    
    try:
        # ... логика генерации ...
        report = build_pnl_report(period, brand)
        
        reporter.report_idle(metrics={
            "reports_generated_today": get_daily_reports_count()
        })
        
        return {"success": True, "report_id": report.id}
        
    except Exception as e:
        reporter.report_error(f"P&L generation: {e}")
        raise

# app/tasks/cfo/office.py

from app.utils.office_reporter import OfficeReporter

OFFICE_REPORTER = OfficeReporter(
    agent_id="cfo_report",
    department="cfo",
    name="Отчёт P&L",
    salary_equivalent=80000,
    fte_coefficient=1.0
)

@shared_task(name='cfo.tasks.office_heartbeat')
def office_heartbeat():
    OFFICE_REPORTER.heartbeat()
    return {"success": True}