file_reader.py 852 B

1234567891011121314151617181920
  1. import os
  2. import sys
  3. from service.trunks_service import TrunksService
  4. class FileReader:
  5. @staticmethod
  6. def find_and_print_split_files(directory):
  7. for root, dirs, files in os.walk(directory):
  8. for file in files:
  9. if '_split_' in file and file.endswith('.txt'):
  10. #if file.endswith('.md'):
  11. file_path = os.path.join(root, file)
  12. relative_path = '\\report\\' + os.path.relpath(file_path, directory)
  13. with open(file_path, 'r', encoding='utf-8') as f:
  14. content = f.read()
  15. TrunksService().create_trunk({'file_path': relative_path, 'content': content,'type':'trunk'})
  16. if __name__ == '__main__':
  17. directory = 'e:\\project\\knowledge\\utils\\files'
  18. FileReader.find_and_print_split_files(directory)