title_with_video/applications/routes.py

"""
@author: luojunhui
"""
import json
import time
import uuid
import asyncio
from quart import Blueprint, jsonify, request

from applications.functions.log import logging
from applications.schedule import recall_videos, search_videos
from applications.functions.kimi import KimiServer

my_blueprint = Blueprint('LongArticles', __name__)


def Routes(mysql_client):
    """
    路由代码
    """

    @my_blueprint.route('/healthcheck')
    def healthcheck():
        """
        Hello World Test
        :return:
        """
        logging(
            code="1001",
            info="请求接口成功",
            port="healthcheck"
        )
        return jsonify({'message': 'Hello, World!'})

    @my_blueprint.route('/search_videos', methods=['POST'])
    async def search_videos_from_the_web():
        """
        从web 搜索视频并且存储到票圈的视频库中
        :return:
        """
        params = await request.get_json()
        K = KimiServer()
        gh_id = params['ghId']
        trace_id = "search-{}-{}".format(str(uuid.uuid4()), str(int(time.time())))
        params['trace_id'] = trace_id
        title = params['title'].split("@@")[-1].replace("'", "")
        contents = params['content'].replace("'", "")
        account_name = params['accountName'].replace("'", "")
        logging(
            code="2000",
            info="搜索视频内容接口请求成功",
            port="title_to_search",
            function="search_videos_from_the_web",
            trace_id=trace_id
        )
        insert_sql = f"""
            INSERT INTO long_articles_video_dev 
                (trace_id, gh_id, article_title, article_text, account_name)
            VALUES 
                ('{trace_id}', '{gh_id}', '{title}', '{contents}', '{account_name}');"""
        await mysql_client.async_insert(insert_sql)
        try:
            kimi_info = await K.search_kimi_schedule(params=params)
            print(json.dumps(kimi_info, ensure_ascii=False, indent=4))
            kimi_title = kimi_info['k_title']
            content_title = kimi_info['content_title']
            content_keys = json.dumps(kimi_info['content_keys'], ensure_ascii=False)
            update_kimi_sql = f"""
                UPDATE long_articles_video_dev SET
                kimi_title = '{kimi_title}',
                kimi_summary = '{content_title}',
                kimi_keys = '{content_keys}'
                WHERE trace_id = '{trace_id}';
            """
            await mysql_client.async_insert(update_kimi_sql)
            asyncio.ensure_future(
                search_videos(
                    kimi_info=kimi_info,
                    trace_id=trace_id,
                    gh_id=gh_id,
                    mysql_client=mysql_client
                )
            )
            res = {
                "status": "success",
                "code": 0,
                "traceId": trace_id
            }
        except Exception as e:
            res = {
                "status": "fail",
                "code": 1,
                "message": str(e)
            }
        return jsonify(res)

    @my_blueprint.route('/recall_videos', methods=['POST'])
    async def find_videos():
        """
        请求接口代码
        :return:
        """
        data = await request.get_json()
        trace_id = data['traceId']
        logging(
            code="1001",
            info="请求接口成功",
            port="recall_videos",
            trace_id=trace_id
        )
        try:
            result = await recall_videos(
                trace_id=trace_id,
                mysql_client=mysql_client
            )
        except Exception as e:
            result = {
                "traceId": trace_id,
                "error": e
            }
        return jsonify(result)

    return my_blueprint