main.py 723 B

1234567891011121314151617181920212223
  1. import os
  2. import sys
  3. import json
  4. from backend.api import app
  5. from backend.services import process_case_dir
  6. def main():
  7. base_dir = os.path.join(os.path.dirname(__file__), "test_complete")
  8. if len(sys.argv) > 1:
  9. case_id = sys.argv[1]
  10. case_dir = os.path.join(base_dir, case_id)
  11. results = [process_case_dir(case_dir)]
  12. else:
  13. results = []
  14. if os.path.exists(base_dir):
  15. for name in os.listdir(base_dir):
  16. case_dir = os.path.join(base_dir, name)
  17. if os.path.isdir(case_dir):
  18. results.append(process_case_dir(case_dir))
  19. print(json.dumps(results, ensure_ascii=False, indent=2))
  20. if __name__ == "__main__":
  21. main()