model_script/client/AIGCClient.py

import json

import requests


class AIGCClient(object):
    def __init__(self, token, base_url):
        self.token = token
        self.base_url = base_url

    def update_produce_plan_status(self, produce_plan_id: str, status: int):
        params = {
            "id": produce_plan_id,
            "status": status,
        }
        url = f"{self.base_url}/aigc/produce/plan/updatePlanStatus"
        self.post(url, params)

    def post(self, url: str, params: dict):
        request_param = {
            "params": params,
            "baseInfo": {
                "token": self.token,
            }
        }
        print(f"invoke aigc platform url: {url} request: {json.dumps(request_param)}")
        response = requests.post(url, json=request_param)
        print(f"invoke aigc platform url: {url} request: {json.dumps(request_param)} response: {response.json()}")
        return response.json()