push

app/services/health_service.py

@@ -0,0 +1,71 @@
+from datetime import datetime
+from sqlalchemy import text
+from app.extensions import db
+from app.services.company_service import CompanyService
+from app.services.ksef_service import KSeFService
+from app.services.settings_service import SettingsService
+from app.services.ceidg_service import CeidgService
+from app.services.redis_service import RedisService
+
+
+class HealthService:
+    CACHE_TTL = 300
+
+    def cache_key(self, company_id):
+        return f'health.status.company.{company_id or "global"}'
+
+    def get_cached_status(self, company_id=None):
+        return RedisService.get_json(self.cache_key(company_id))
+
+    def warm_cache(self, company_id=None):
+        status = self.get_status(force_refresh=True, company_id=company_id)
+        RedisService.set_json(self.cache_key(company_id), status, ttl=self.CACHE_TTL)
+        return status
+
+    def get_status(self, force_refresh: bool = False, company_id=None):
+        if not force_refresh:
+            cached = self.get_cached_status(company_id)
+            if cached:
+                return cached
+
+        company = CompanyService.get_current_company()
+        if company_id is None and company:
+            company_id = company.id
+        redis_status, redis_details = RedisService.ping()
+
+        status = {
+            'timestamp': datetime.utcnow().isoformat(),
+            'db': 'ok',
+            'smtp': 'configured' if SettingsService.get_effective('mail.server', company_id=company_id) else 'not_configured',
+            'redis': redis_status,
+            'redis_details': redis_details,
+            'ksef': 'unknown',
+            'ceidg': 'unknown',
+            'ksef_message': '',
+            'ceidg_message': '',
+        }
+
+        try:
+            db.session.execute(text('SELECT 1'))
+        except Exception:
+            status['db'] = 'error'
+
+        try:
+            ping = KSeFService(company_id=company_id).ping()
+            status['ksef'] = ping.get('status', 'unknown')
+            status['ksef_message'] = ping.get('message', '')
+        except Exception as exc:
+            status['ksef'] = 'error'
+            status['ksef_message'] = str(exc)
+
+        try:
+            ping = CeidgService().diagnostics()
+            status['ceidg'] = ping.get('status', 'unknown')
+            status['ceidg_message'] = ping.get('message', '')
+        except Exception as exc:
+            status['ceidg'] = 'error'
+            status['ceidg_message'] = str(exc)
+
+        status['critical'] = status['db'] != 'ok' or status['ksef'] not in ['ok', 'mock'] or status['ceidg'] != 'ok'
+        RedisService.set_json(self.cache_key(company_id), status, ttl=self.CACHE_TTL)
+        return status