import re
import time

from database.database import get_session
from database.tcontentdispatch.curd import get_content_by_title_and_category, create_or_update_content
from database.tcontentdispatch.model import TContentDispatch
from database.tmaterial.crud import update_material_by_id, \
    get_materials_for_generate_real_estate_reference_message
from task.manager_task import execute_task


def generate_real_estate_reference_message_task():
    with get_session() as db:
        # 1. 构建楼市参考消息文章标题，格式：楼市参考消息yyyy-MM-dd
        title = ("楼市参考消息" +
                 time.strftime("%Y", time.localtime()) + '年' +
                 time.strftime("%m", time.localtime()) + '月' +
                 time.strftime("%d", time.localtime()) + '日')
        # 2. 从内容分发数据表获取当前标题和分类的文章是否存在
        content_dispatch = get_content_by_title_and_category(db, title, "房地产")
        content = ""
        if content_dispatch is not None:
            content = content_dispatch.content
            # 从最后一条获取并计算开始编号
            result = re.findall(r'(?<!\.)\d+\. ', content)
            start_num = int(re.findall(r'\d+', result[-1])[-1]) + 1
        else:
            content_dispatch = TContentDispatch(category="房地产", title=title, ai_generate=1)
            start_num = 1
        # 3. 从新闻素材数据表获取房产类的新闻列表
        news_list = get_materials_for_generate_real_estate_reference_message(db)
        # 4. 拼接成文章正文content
        for i, news in enumerate(news_list, start=start_num):  # Using enumerate to control the index starting from 1
            content += f"{i}. {news.ai_summary}\n"
        # 5. 把content写入数据库
        if content is not None and news_list is not None:
            content_dispatch.content = content
            content_dispatch.is_sent = False
            create_or_update_content(db, content_dispatch)
        # 6. 把news_list更新入数据库更新字段is_usage为True
        for news in news_list:
            news.is_usage = True
            update_material_by_id(db, news)


if __name__ == "__main__":
    execute_task(generate_real_estate_reference_message_task)