model_script/client/PQClient.py

from typing import List, Dict, Any

import requests


class PQClient:
    def __init__(self):
        self._base_url = "http://longvideoapi-internal.piaoquantv.com"

    def batch_select_video_info(self, video_ids: List[int]) -> List[Dict[str, Any]]:
        url = f"{self._base_url}/longvideoapi/openapi/video/batchSelectVideoInfo"
        data = {
            "videoIdList": video_ids,
        }
        response = self._post(url, {}, data)
        return response['data']

    def get_video_info(self, video_id: int) -> Dict[str, Any]:
        url = f"{self._base_url}/longvideoapi/openapi/video/getVideoInfo"
        data = {
            "videoId": video_id,
        }
        response = self._post(url, {}, data)
        return response['data']

    @staticmethod
    def _post(url: str, headers: Dict[str, str], data: Dict[str, Any]) -> Dict[str, Any]:
        return requests.post(url, headers=headers, json=data).json()