push

app/scheduler.py

@@ -0,0 +1,62 @@
+from datetime import datetime, timedelta
+from apscheduler.schedulers.background import BackgroundScheduler
+from flask import current_app
+
+scheduler = BackgroundScheduler(job_defaults={'coalesce': True, 'max_instances': 1})
+
+
+def init_scheduler(app):
+    if scheduler.running:
+        return
+
+    def sync_job():
+        with app.app_context():
+            from app.models.company import Company
+            from app.models.sync_log import SyncLog
+            from app.services.sync_service import SyncService
+            current_app.logger.info('Scheduled sync checker started')
+            companies = Company.query.filter_by(is_active=True, sync_enabled=True).all()
+            for company in companies:
+                last_log = SyncLog.query.filter_by(company_id=company.id, status='finished').order_by(SyncLog.finished_at.desc()).first()
+                due = not last_log or not last_log.finished_at or (datetime.utcnow() - last_log.finished_at) >= timedelta(minutes=company.sync_interval_minutes)
+                if due:
+                    SyncService(company).run_scheduled_sync()
+
+    def refresh_dashboard_cache_job():
+        with app.app_context():
+            from app.models.company import Company
+            from app.models.invoice import Invoice
+            from app.services.health_service import HealthService
+            from app.services.redis_service import RedisService
+            from sqlalchemy import extract
+            from datetime import date
+            from decimal import Decimal
+
+            today = date.today()
+            for company in Company.query.filter_by(is_active=True).all():
+                base = Invoice.query.filter_by(company_id=company.id)
+                month_invoices = base.filter(
+                    extract('month', Invoice.issue_date) == today.month,
+                    extract('year', Invoice.issue_date) == today.year,
+                ).order_by(Invoice.issue_date.desc(), Invoice.id.desc()).all()
+                recent_ids = [invoice.id for invoice in base.order_by(Invoice.created_at.desc(), Invoice.id.desc()).limit(200).all()]
+                totals = {
+                    'net': str(sum(Decimal(invoice.net_amount) for invoice in month_invoices)),
+                    'vat': str(sum(Decimal(invoice.vat_amount) for invoice in month_invoices)),
+                    'gross': str(sum(Decimal(invoice.gross_amount) for invoice in month_invoices)),
+                }
+                RedisService.set_json(
+                    f'dashboard.summary.company.{company.id}',
+                    {
+                        'month_invoice_ids': [invoice.id for invoice in month_invoices],
+                        'unread': base.filter_by(is_unread=True).count(),
+                        'totals': totals,
+                        'recent_invoice_ids': recent_ids,
+                    },
+                    ttl=300,
+                )
+                HealthService().warm_cache(company.id)
+
+    scheduler.add_job(sync_job, 'interval', minutes=1, id='ksef_sync', replace_existing=True)
+    scheduler.add_job(refresh_dashboard_cache_job, 'interval', minutes=5, id='dashboard_cache_refresh', replace_existing=True)
+    scheduler.start()