mirror of
https://github.com/kjanat/livegraphs-django.git
synced 2026-01-16 16:12:10 +01:00
80 lines
3.3 KiB
Python
80 lines
3.3 KiB
Python
# dashboard/signals.py
|
|
|
|
import logging
|
|
|
|
from dashboard.models import ChatSession as DashboardChatSession
|
|
from dashboard.models import DataSource
|
|
from data_integration.models import ChatSession as ExternalChatSession
|
|
from django.db.models.signals import post_save
|
|
from django.dispatch import receiver
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@receiver(post_save, sender=ExternalChatSession)
|
|
def sync_external_session_to_dashboard(
|
|
sender, # noqa: ARG001
|
|
instance,
|
|
created,
|
|
**kwargs, # noqa: ARG001
|
|
):
|
|
"""
|
|
Signal handler to sync external chat sessions to dashboard chat sessions
|
|
whenever an external session is created or updated.
|
|
|
|
Args:
|
|
sender: The model class that sent the signal (unused but required by Django's signal interface)
|
|
instance: The ExternalChatSession instance that was saved
|
|
created: Boolean indicating if this is a new instance
|
|
**kwargs: Additional keyword arguments (unused but required by Django's signal interface)
|
|
"""
|
|
# Find all dashboard data sources that are linked to this external data source
|
|
# Since ExternalChatSession doesn't have a direct link to ExternalDataSource,
|
|
# we need to sync to all dashboard data sources with external sources
|
|
data_sources = DataSource.objects.exclude(external_source=None)
|
|
|
|
if not data_sources.exists():
|
|
logger.warning(f"No dashboard data sources with external sources found for session {instance.session_id}")
|
|
return
|
|
|
|
for data_source in data_sources:
|
|
try:
|
|
# Create or update dashboard chat session
|
|
dashboard_session, created = DashboardChatSession.objects.update_or_create(
|
|
data_source=data_source,
|
|
session_id=instance.session_id,
|
|
defaults={
|
|
"start_time": instance.start_time,
|
|
"end_time": instance.end_time,
|
|
"ip_address": instance.ip_address,
|
|
"country": instance.country or "",
|
|
"language": instance.language or "",
|
|
"messages_sent": instance.messages_sent or 0,
|
|
"sentiment": instance.sentiment or "",
|
|
"escalated": instance.escalated or False,
|
|
"forwarded_hr": instance.forwarded_hr or False,
|
|
"full_transcript": instance.full_transcript_url or "",
|
|
"avg_response_time": instance.avg_response_time,
|
|
"tokens": instance.tokens or 0,
|
|
"tokens_eur": instance.tokens_eur,
|
|
"category": instance.category or "",
|
|
"initial_msg": instance.initial_msg or "",
|
|
"user_rating": (str(instance.user_rating) if instance.user_rating is not None else ""),
|
|
},
|
|
)
|
|
|
|
if created:
|
|
logger.info(
|
|
f"Created dashboard session: {dashboard_session.session_id} for data source {data_source.name}"
|
|
)
|
|
else:
|
|
logger.info(
|
|
f"Updated dashboard session: {dashboard_session.session_id} for data source {data_source.name}"
|
|
)
|
|
|
|
except Exception as e:
|
|
logger.error(
|
|
f"Error syncing session {instance.session_id} to data source {data_source.name}: {e}",
|
|
exc_info=True,
|
|
)
|