from ai.models import *
from ai.utils.show_prompt import *
import datetime
from django_redis import get_redis_connection
import re

# ai排片主流程
def show_main_process():
    # 查看状态
    redis_conn = get_redis_connection()
    redis_key = f'ai_show{datetime.date.today().strftime("%Y%m%d")}'
    if redis_conn.exists(redis_key):
        return False
    # 获取影院列表
    redis_conn.set(redis_key, 1, ex=60*60*20)
    test_cinema_list = TestCinema.objects.filter(is_active=True).all()
    # 生成目标日期
    show_date = datetime.date.strftime(datetime.date.today() + datetime.timedelta(days=3), '%Y-%m-%d')
    for cinema in test_cinema_list:
        print(cinema.name)
        show_ai = ShowAI(cinema, show_date)
        prompt = show_ai.general_prompt()
        start = datetime.datetime.now()
        # result, message, tokens = show_ai.get_show_result_ai()
        # end = datetime.datetime.now()
        # print('prompt:', prompt)
        # print('result:', result)
        # print('message:', message)
        # print('tokens:', tokens)
        # # 获取排片数据
        # _show = next((s for s in result.split('------') if s.startswith('\n影厅别名,影厅id')), '')
        # _show = _show.strip()
        # print('_show:', _show)
        # # 获取销售额
        # _sales = next((s for s in result.split('\n') if re.search('预估销售数据', s)), '')
        # _sales = str(re.findall(r'\d+\.?\d*', _sales)[0])
        # print('_sales:', _sales)
        # # 处理返回结果
        # try:
        #     AiShow.objects.create(
        #         cinema=cinema.name,
        #         zz_code=cinema.zz_code,
        #         show_date=show_date,
        #         is_ai_show=True,
        #         show=_show,
        #         sales=_sales,
        #         prompt=prompt,
        #         result=result,
        #         message=message,
        #         take_times=int((end - start).seconds),
        #         take_tokens=tokens
        #     )
        # except Exception as e:
        #     print(e)
    return True