Переглянути джерело

refactor: Use Path for file paths in process_data.py

mrh 1 рік тому
батько
коміт
d9278a3ba8
1 змінених файлів з 4 додано та 2 видалено
  1. 4 2
      process_data.py

+ 4 - 2
process_data.py

@@ -1,5 +1,6 @@
 import csv
 import chardet
+from pathlib import Path
 from ai_trans import translate_sentences
 from brand_add_url_link import create_hyperlink
 
@@ -103,6 +104,7 @@ def main(input_file, output_file):
         raise
 
 if __name__ == "__main__":
-    input_file = "测试.csv"
-    output_file = "processed_测试.csv"
+    output_dir = Path('temp')
+    input_file = output_dir/"测试.csv"
+    output_file = output_dir/"processed_测试.csv"
     main(input_file, output_file)