107 lines
4.8 KiB
Python
107 lines
4.8 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎虎
|
|||
|
|
@file： application_stats.py
|
|||
|
|
@date：2025/6/9 20:34
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
import datetime
|
|||
|
|
import os
|
|||
|
|
from typing import Dict, List
|
|||
|
|
|
|||
|
|
from django.db import models
|
|||
|
|
from django.db.models import QuerySet
|
|||
|
|
from django.utils.translation import gettext_lazy as _
|
|||
|
|
from rest_framework import serializers
|
|||
|
|
|
|||
|
|
from application.models import ApplicationChatUserStats
|
|||
|
|
from common.db.search import native_search, get_dynamics_model
|
|||
|
|
from common.utils.common import get_file_content
|
|||
|
|
from maxkb.conf import PROJECT_DIR
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ApplicationStatsSerializer(serializers.Serializer):
|
|||
|
|
chat_record_count = serializers.IntegerField(required=True, label=_("Number of conversations"))
|
|||
|
|
customer_added_count = serializers.IntegerField(required=True, label=_("Number of new users"))
|
|||
|
|
customer_num = serializers.IntegerField(required=True, label=_("Total number of users"))
|
|||
|
|
day = serializers.CharField(required=True, label=_("date"))
|
|||
|
|
star_num = serializers.IntegerField(required=True, label=_("Number of Likes"))
|
|||
|
|
tokens_num = serializers.IntegerField(required=True, label=_("Tokens consumption"))
|
|||
|
|
trample_num = serializers.IntegerField(required=True, label=_("Number of thumbs-downs"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ApplicationStatisticsSerializer(serializers.Serializer):
|
|||
|
|
application_id = serializers.UUIDField(required=True, label=_("Application ID"))
|
|||
|
|
start_time = serializers.DateField(format='%Y-%m-%d', label=_("Start time"))
|
|||
|
|
end_time = serializers.DateField(format='%Y-%m-%d', label=_("End time"))
|
|||
|
|
|
|||
|
|
def get_end_time(self):
|
|||
|
|
return datetime.datetime.combine(
|
|||
|
|
datetime.datetime.strptime(self.data.get('end_time'), '%Y-%m-%d'),
|
|||
|
|
datetime.datetime.max.time())
|
|||
|
|
|
|||
|
|
def get_start_time(self):
|
|||
|
|
return self.data.get('start_time')
|
|||
|
|
|
|||
|
|
def get_customer_count_trend(self, with_valid=True):
|
|||
|
|
if with_valid:
|
|||
|
|
self.is_valid(raise_exception=True)
|
|||
|
|
start_time = self.get_start_time()
|
|||
|
|
end_time = self.get_end_time()
|
|||
|
|
return native_search(
|
|||
|
|
{'default_sql': QuerySet(ApplicationChatUserStats).filter(
|
|||
|
|
application_id=self.data.get('application_id'),
|
|||
|
|
create_time__gte=start_time,
|
|||
|
|
create_time__lte=end_time)},
|
|||
|
|
select_string=get_file_content(
|
|||
|
|
os.path.join(PROJECT_DIR, "apps", "application", 'sql', 'customer_count_trend.sql')))
|
|||
|
|
|
|||
|
|
def get_chat_record_aggregate_trend(self, with_valid=True):
|
|||
|
|
if with_valid:
|
|||
|
|
self.is_valid(raise_exception=True)
|
|||
|
|
start_time = self.get_start_time()
|
|||
|
|
end_time = self.get_end_time()
|
|||
|
|
chat_record_aggregate_trend = native_search(
|
|||
|
|
{'default_sql': QuerySet(model=get_dynamics_model(
|
|||
|
|
{'application_chat.application_id': models.UUIDField(),
|
|||
|
|
'application_chat_record.create_time': models.DateTimeField()})).filter(
|
|||
|
|
**{'application_chat.application_id': self.data.get('application_id'),
|
|||
|
|
'application_chat_record.create_time__gte': start_time,
|
|||
|
|
'application_chat_record.create_time__lte': end_time}
|
|||
|
|
)},
|
|||
|
|
select_string=get_file_content(
|
|||
|
|
os.path.join(PROJECT_DIR, "apps", "application", 'sql', 'chat_record_count_trend.sql')))
|
|||
|
|
customer_count_trend = self.get_customer_count_trend(with_valid=False)
|
|||
|
|
return self.merge_customer_chat_record(chat_record_aggregate_trend, customer_count_trend)
|
|||
|
|
|
|||
|
|
def merge_customer_chat_record(self, chat_record_aggregate_trend: List[Dict], customer_count_trend: List[Dict]):
|
|||
|
|
|
|||
|
|
return [{**self.find(chat_record_aggregate_trend, lambda c: c.get('day').strftime('%Y-%m-%d') == day,
|
|||
|
|
{'star_num': 0, 'trample_num': 0, 'tokens_num': 0, 'chat_record_count': 0,
|
|||
|
|
'customer_num': 0,
|
|||
|
|
'day': day}),
|
|||
|
|
**self.find(customer_count_trend, lambda c: c.get('day').strftime('%Y-%m-%d') == day,
|
|||
|
|
{'customer_added_count': 0})}
|
|||
|
|
for
|
|||
|
|
day in
|
|||
|
|
self.get_days_between_dates(self.data.get('start_time'), self.data.get('end_time'))]
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def find(source_list, condition, default):
|
|||
|
|
value_list = [row for row in source_list if condition(row)]
|
|||
|
|
if len(value_list) > 0:
|
|||
|
|
return value_list[0]
|
|||
|
|
return default
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_days_between_dates(start_date, end_date):
|
|||
|
|
start_date = datetime.datetime.strptime(start_date, '%Y-%m-%d')
|
|||
|
|
end_date = datetime.datetime.strptime(end_date, '%Y-%m-%d')
|
|||
|
|
days = []
|
|||
|
|
current_date = start_date
|
|||
|
|
while current_date <= end_date:
|
|||
|
|
days.append(current_date.strftime('%Y-%m-%d'))
|
|||
|
|
current_date += datetime.timedelta(days=1)
|
|||
|
|
return days
|