alist-airia2-py/tests/test.py

import configparser
import json
import logging
import os

from src.api.Aria2API import Aria2API
from src.api.radarr import RadarClient


# 运行测试

def test_find_movie_id_by_filename():
    # 初始化日志和配置
    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
    # 获取当前脚本的绝对路径
    current_directory = os.path.dirname(os.path.abspath(__file__))

    # 获取当前脚本所在目录的上级目录
    parent_directory = os.path.dirname(current_directory)

    # 构建 config.ini 文件的路径
    config_path = os.path.join(parent_directory, 'config', 'config.ini')

    # 读取配置文件
    config = configparser.ConfigParser()
    config.read(config_path)

    # 创建 RadarClient 实例
    radar_client = RadarClient(config['RADAR']['URL'],
                               config['RADAR']['API_KEY'])

    movies = radar_client.get_all_movie_names()

    print(f"movies_jons: {movies}")

    aria2_api = Aria2API(config['ARIA2']['RPC_URL'], config['ARIA2']['RPC_SECRET'])
    downloads = aria2_api.get_downloads()
    for download in downloads:
        if download.is_complete:
            completed_files = [os.path.basename(file.path) for file in download.files if file.selected]
            print(f"completed_files: {completed_files}")
            # 查找电影ID
            for completed_file in completed_files:
                ids = radar_client.find_movie_id_by_filename(completed_file)
                if ids is not None:
                    print(f"ids: {ids}")


# 运行测试
test_find_movie_id_by_filename()