ASJServer/Roomumy/Controller/FeedDiaryController.py

# -*- encoding: utf-8 -*-
"""
@File    : FeedDiaryController.py
@Time    : 2024-10-7 14:37:30
@Author  : peng
@Email   :
@Software: PyCharm
"""
from datetime import datetime, timedelta
import time

from django.views import View

from Roomumy.models import FeedDiary, FeedType
from Service.CommonService import CommonService


class FeedDiaryView(View):
    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.GET, request, operation)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.POST, request, operation)

    def validation(self, request_dict, request, operation):
        token_code, user_id, response = CommonService.verify_token_get_user_id(request_dict, request)
        if token_code != 0:
            return response.json(token_code)
        if operation == 'query':  # 查询喂养记录
            return self.get_feed_diary_data(request_dict, response)
        elif operation == 'queryStatistics':  # 查询喂养统计
            return self.get_statistics_data(request_dict, response)
        elif operation == 'addOrEdit':  # 添加或编辑喂养日记
            return self.add_or_edit_feed_diary(request_dict, response)
        elif operation == 'delete':  # 删除喂养日记
            return self.delete_feed_diary(request_dict, response)
        elif operation == 'getFeedList':  # 查询喂养类型列表
            return self.get_feed_list(request_dict, response)
        elif operation == 'feedTypeConfig':
            return self.feed_type_config(request_dict, response)
        elif operation == 'getDailyRecordStatus':  # 查询天数记录
            return self.get_daily_record_status(request_dict, response)
        else:
            return response.json(414)

    @classmethod
    def get_feed_diary_data(cls, request_dict, response):
        """
        查询喂养记录
        @param request_dict: 请求参数
        @request_dict start_time: 开始时间戳
        @request_dict end_time: 结束时间戳
        @request_dict baby_id: 宝宝id
        @request_dict event_type: 喂养类型
        @param response: 响应对象
        @return: response
        """
        start_time = request_dict.get('start_time', None)
        end_time = request_dict.get('end_time', None)
        baby_id = request_dict.get('baby_id', None)
        feed_type = request_dict.get('feed_type', None)
        if not all([start_time, end_time, baby_id]):
            return response.json(444, {'error param': 'start_time, end_time or baby_id'})
        try:
            feed_diary = FeedDiary.objects.filter(date_time__gte=start_time, date_time__lte=end_time,
                                                  baby_id=baby_id)
            if feed_type:
                feed_diary = feed_diary.filter(feed_type=feed_type)
            feed_diary = feed_diary.values('feed_type__name', 'feed_type__icon_url', 'feed_content', 'date_time',
                                           'id', 'feed_type')
            feed_diary = list(feed_diary)

            # 统计各类型的喂养次数
            feed_statistics = {}
            for item in feed_diary:
                feed_type_id = item['feed_type']
                feed_type_name = item['feed_type__name']

                # 初始化统计项
                if feed_type_id not in feed_statistics:
                    feed_statistics[feed_type_id] = {
                        "feed_type_name": feed_type_name,
                        "frequency": 0,
                        "total": 0  # 仅对类型1/2/3有效
                    }

                # 更新统计
                feed_statistics[feed_type_id]["frequency"] += 1

                # 仅对类型1/2/3计算total
                if item['feed_type'] in [1, 2, 3] and "total" in item.get("feed_content", {}):
                    feed_statistics[feed_type_id]["total"] += item["feed_content"]["total"]

            # 转换统计结果为列表
            feed_statistics_list = [
                {
                    "feed_type": feed_type_id,
                    "feed_type_name": data["feed_type_name"],
                    "frequency": str(data["frequency"]),
                    "total": str(data["total"])
                }
                for feed_type_id, data in feed_statistics.items()
            ]

            return response.json(0, {
                'feed_diary': feed_diary,
                'feed_statistics': feed_statistics_list  # 改为列表形式，更清晰
            })
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def get_statistics_data(cls, request_dict, response):
        """
        查询喂养记录
        @param request_dict: 请求参数
        @request_dict start_time: 开始时间戳
        @request_dict end_time: 结束时间戳
        @request_dict baby_id: 宝宝id
        @request_dict event_type: 喂养类型
        @request_dict sub_event_type: 喂养子类型
        @param response: 响应对象
        @return: response
        """
        start_time = request_dict.get('start_time', None)
        end_time = request_dict.get('end_time', None)
        baby_id = request_dict.get('baby_id', None)
        feed_type = request_dict.get('feed_type', None)
        sub_feed_type = request_dict.get('sub_feed_type', None)
        if not all([start_time, end_time, baby_id, feed_type]):
            return response.json(444, {'error param': 'start_time, end_time, event_type or baby_id'})
        s_time = datetime.datetime.fromtimestamp(int(start_time))
        e_time = datetime.datetime.fromtimestamp(int(end_time))
        time_list = CommonService.cutting_time(s_time, e_time, 'day')
        try:
            result_list = []
            for time_range in time_list:
                content_list = []
                feed_diary = FeedDiary.objects.filter(created_time__gte=time_range[0], created_time__lt=time_range[1],
                                                      baby_id=baby_id, feed_type_id=feed_type).values_list(
                    'feed_content', flat=True)
                count = 0
                if feed_type == '5' and sub_feed_type:  # 换尿布
                    sub_feed_type = int(sub_feed_type)
                    for item in feed_diary:
                        if item['sub_feed_type'] in [sub_feed_type, 3]:  # 1：嘘嘘 2：臭臭 3：嘘嘘+臭臭 4：干爽
                            count += 1
                            content_list.append(item)
                else:
                    count = feed_diary.count()
                    content_list = list(feed_diary)
                result_list.append({
                    'date_time': time_range[0],
                    'count': count,
                    'feed_content': content_list
                })
            return response.json(0, result_list)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def add_or_edit_feed_diary(cls, request_dict, response):
        """
        添加编辑喂养记录
        @param request_dict: 请求参数
        @request_dict date_time: 喂养时间戳
        @request_dict baby_id: 宝宝id
        @request_dict feed_id: 记录id
        @request_dict event_type: 喂养类型
        @request_dict event_content: 喂养事件
        @param response: 响应对象
        @return: response
        """
        date_time = request_dict.get('date_time', None)
        baby_id = request_dict.get('baby_id', None)
        feed_id = request_dict.get('feed_id', None)
        feed_type = request_dict.get('feed_type', None)
        feed_content = request_dict.get('feed_content', None)
        if not all([date_time, baby_id, feed_type, feed_content]):
            return response.json(444, {'error param': 'date_time, feed_content, feed_type or baby_id'})
        feed_content = eval(feed_content)
        try:
            now_time = int(time.time())
            feed_type_qs = FeedType.objects.filter(id=feed_type)
            if not feed_type_qs:
                return response.json(444, {'error param': 'feed_type'})
            if feed_id:
                FeedDiary.objects.filter(id=feed_id, baby_id=baby_id).update(date_time=date_time,
                                                                             updated_time=now_time,
                                                                             feed_content=feed_content)
            else:
                FeedDiary.objects.create(baby_id=baby_id, feed_type_id=feed_type, feed_content=feed_content,
                                         date_time=date_time, created_time=now_time, updated_time=now_time)
            return response.json(0)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def delete_feed_diary(cls, request_dict, response):
        """
        删除喂养记录
        @param request_dict: 请求参数
        @request_dict baby_id: 宝宝id
        @request_dict feed_id: 记录id
        @param response: 响应对象
        @return: response
        """
        baby_id = request_dict.get('baby_id', None)
        feed_id = request_dict.get('feed_id', None)
        if not all([feed_id, baby_id]):
            return response.json(444, {'error param': 'feed_id or baby_id'})
        try:
            FeedDiary.objects.filter(id=feed_id, baby_id=baby_id).delete()
            return response.json(0)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def get_feed_list(cls, request_dict, response):
        """
        获取喂养列表
        @param request_dict: 请求参数
        @param response: 响应对象
        @return: response
        """
        try:
            feed_type = FeedType.objects.filter(is_show=True).values('id', 'name', 'icon_url', 'sort').order_by('sort')
            return response.json(0, list(feed_type))
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def feed_type_config(cls, request_dict, response):
        """
        根据喂养类型返回文本列表
        @param request_dict: 请求参数
        @param response: 响应对象
        @return: response
        """
        try:
            feed_type = request_dict.get('feed_type_id', None)
            if feed_type == '4':#辅食
                feed_type_text = [
                    "米粉", "面条", "红薯", "山药", "南瓜",
                    "胡萝卜", "菠菜", "番茄", "土豆", "猪肉",
                    "猪肝", "鱼肉", "鸡肉", "牛肉", "虾",
                    "鸡蛋", "苹果", "香蕉", "猕猴桃", "火龙果"
                ]

            elif feed_type == '5':
                feed_type_text = ["嘘嘘", "臭臭", "嘘嘘+臭臭", "干爽"]

            elif feed_type == '10':
                feed_type_text = ["额温", "耳温", "腋温", "口温", "肛温"]

            else:
                feed_type_text = []
            return response.json(0, {"feed_type_text": feed_type_text})
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @classmethod
    def get_daily_record_status(cls, request_dict, response):
        """
        获取记录的日期
        @param request_dict: 请求参数
        @request_dict baby_id: 宝宝id
        @request_dict timestamp: 时间戳
        @param response: 响应对象
        @return: response
        """
        baby_id = request_dict.get('baby_id')
        start_time = request_dict.get('start_time')
        end_time = request_dict.get('end_time')

        if not all([baby_id, start_time, end_time]):
            return response.json(444, {'error': '缺少baby_id、start_time或end_time参数'})

        try:
            # 转换为整数
            start_time = int(start_time)
            end_time = int(end_time)

            records = FeedDiary.objects.filter(
                baby_id=baby_id,
                date_time__gte=start_time,
                date_time__lte=end_time
            ).values('date_time')
            record_dates = {
                datetime.fromtimestamp(r['date_time']).strftime('%Y-%m-%d')
                for r in records
            }
            # 生成时间段内所有日期
            all_dates = []
            current_date = datetime.fromtimestamp(start_time).date()
            end_date = datetime.fromtimestamp(end_time).date()

            while current_date <= end_date:
                all_dates.append(current_date.strftime('%Y-%m-%d'))
                current_date += timedelta(days=1)

            # 构建返回结果
            result = []
            for date in all_dates:
                result.append({
                    'date': date,
                    'has_record': date in record_dates
                })

            return response.json(0, {
                'baby_id': baby_id,
                'days': result
            })

        except ValueError as e:
            return response.json(444, {'error': '时间戳格式不正确'})
        except Exception as e:
            return response.json(500, {'error': f'服务器错误: {str(e)}'})