test.py 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. import configparser
  2. import json
  3. import logging
  4. import os
  5. from src.api.Aria2API import Aria2API
  6. from src.api.radarr import RadarClient
  7. # 运行测试
  8. def test_find_movie_id_by_filename():
  9. # 初始化日志和配置
  10. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
  11. # 获取当前脚本的绝对路径
  12. current_directory = os.path.dirname(os.path.abspath(__file__))
  13. # 获取当前脚本所在目录的上级目录
  14. parent_directory = os.path.dirname(current_directory)
  15. # 构建 config.ini 文件的路径
  16. config_path = os.path.join(parent_directory, 'config', 'config.ini')
  17. # 读取配置文件
  18. config = configparser.ConfigParser()
  19. config.read(config_path)
  20. # 创建 RadarClient 实例
  21. radar_client = RadarClient(config['RADAR']['URL'],
  22. config['RADAR']['API_KEY'])
  23. movies = radar_client.get_all_movie_names()
  24. print(f"movies_jons: {movies}")
  25. aria2_api = Aria2API(config['ARIA2']['RPC_URL'], config['ARIA2']['RPC_SECRET'])
  26. downloads = aria2_api.get_downloads()
  27. for download in downloads:
  28. if download.is_complete:
  29. pass
  30. # <<<<<<< HEAD
  31. # # 检查下载的每个文件
  32. # for file in download.files:
  33. # if file.selected:
  34. # completed_files = [os.path.basename(file.path) for file in download.files if file.selected]
  35. # for completed_file in completed_files:
  36. #
  37. # file_path = os.path.join(config['ARIA2']['DESTINATION_PATH'], completed_file)
  38. # temp_file = file_path + '.aria2'
  39. # print(f'temp_file: {temp_file}')
  40. # =======
  41. # completed_files = [os.path.basename(file.path) for file in download.files if file.selected]
  42. # print(f"completed_files: {completed_files}")
  43. # # 查找电影ID
  44. # for completed_file in completed_files:
  45. # ids = radar_client.find_movie_id_by_filename(completed_file)
  46. # if ids is not None:
  47. # print(f"ids: {ids}")
  48. # >>>>>>> main
  49. # 运行测试
  50. test_find_movie_id_by_filename()