调用内部含线程的API时，如何等待其执行完成？

• 检查API是否支持同步调用或返回可等待对象
  很多带内部线程的API会提供同步模式参数（比如blocking=True），或者返回Future、Task这类可等待对象。如果demo_v1.main有这类设计，直接调用时开启同步模式，或者对返回值调用result()/wait()方法，就能阻塞主线程直到内部线程执行完成。

• 将API调用封装到子进程中
  利用进程的特性：子进程退出前会等待其内部所有线程执行完毕。把每次API调用放到独立子进程里，主进程等待子进程结束后再进入下一次循环。示例代码（Python）：

  from multiprocessing import Process
  import demo_v1
  
  # 定义所有测试输入
  task_list = [("img_01.jpg", "text_01.txt", 1), ("img_02.jpg", "text_02.txt", 2)]
  
  for img_path, txt_path, num in task_list:
      # 创建子进程执行API调用
      proc = Process(target=demo_v1.main, args=(img_path, txt_path, num))
      proc.start()
      proc.join()  # 阻塞等待子进程完成
  

  若需要获取API返回结果，可借助multiprocessing.Queue或Pipe在主进程和子进程间传递数据。

• 修改API源码添加等待标记（若有权限）
  如果能编辑demo_v1的源码，可以在内部添加线程计数或事件标记：

  1. 在demo_v1.py中添加全局线程同步工具：

     import threading
     # 用信号量记录活跃线程数，初始为0
     active_threads = threading.Semaphore(0)
     

  2. 在main函数启动的线程逻辑末尾，释放信号量：

     def main(img_file, text_file, integer):
         def thread_worker():
             # 原线程执行逻辑
             # ...
             active_threads.release()  # 线程结束时释放信号量
     
         t = threading.Thread(target=thread_worker)
         t.start()
     

  3. 你的调用代码中，每次调用API后等待活跃线程数归零：

     import demo_v1
     
     task_list = [...]
     for img, txt, num in task_list:
         demo_v1.main(img, txt, num)
         # 等待所有内部线程完成
         while demo_v1.active_threads._value > 0:
             pass
     

• 应急方案：固定时长等待（不推荐）
  若以上方法都不可行，可在每次API调用后添加固定时长的睡眠，确保内部线程有足够时间完成。示例：

  import time
  import demo_v1
  
  task_list = [...]
  for img, txt, num in task_list:
      demo_v1.main(img, txt, num)
      time.sleep(15)  # 根据实际任务耗时设置足够长的时间
  

  缺点是灵活性差，过短仍会崩溃，过长则浪费资源。

内容的提问来源于stack exchange，提问作者beluga