py-hutun-robot/api/search.py

"""
搜索API
"""
from flask import Blueprint
from flask import request

import api
from huitun import HuiTunBrowser
from util import lock_util
from xhs import XhsBrowser
from playwright.sync_api import Playwright, sync_playwright

search_opt = Blueprint('search', __name__)

@search_opt.route('/note', methods=["POST"])
def search_note():
    """
    根据关键字搜索抖音笔记
    :return:
    """
    request_body = request.json
    playwright = sync_playwright().start()
    browser = HuiTunBrowser(lock_util.get_idle_phone('huitun'), playwright)
    result = browser.search_note(request_body.get('tagName'), request_body.get('searchLimit'))
    xhs_browser = XhsBrowser(lock_util.get_idle_phone('xhs'), playwright)
    xhs_browser.polish_huitun_note(result)
    playwright.stop()
    return api.success(result)