codes/test.py

import os
# import spaces
os.environ["NUMEXPR_MAX_THREADS"] = '192'
os.chdir('/dev/shm/codes')
from service.trans_dh_service import TransDhTask, Status,a
from y_utils.config import GlobalConfig
from y_utils.logger import logger
import argparse
import requests
import sys
import json

def download_file(url, local_path):
    print(f"下载文件：{url}, 保存路径：{local_path}")
    response = requests.get(url, stream=True)
    response.raise_for_status()
    with open(local_path, 'wb') as f:
        for chunk in response.iter_content(chunk_size=8192):
            f.write(chunk)
# @spaces.GPU(duration=120)
def run():
    with open(os.path.join(GlobalConfig.instance().temp_dir,'config.json'), 'r', encoding='utf-8') as f:
        data = json.load(f)
    code = data['code']
    GlobalConfig.instance().temp_dir = os.path.join(GlobalConfig.instance().temp_dir, code)
    GlobalConfig.instance().result_dir = os.path.join(GlobalConfig.instance().result_dir, code)
    _watermark_switch = 0
    _digital_auth = 0
    _chaofen = 0
    _pn = 1
    # TransDhTask.instance().run_lock.acquire()
    # TransDhTask.instance().run_flag = True
    TransDhTask.instance().task_dic[code] = (Status.run, 0, '', '')
    TransDhTask.instance().work(data['audio'], data['video'], code, _watermark_switch, _digital_auth, _chaofen, _pn)

if __name__ == '__main__':
    print('开始运行')
    TransDhTask.instance()
    run()