file_reader.py 973 B

12345678910111213141516171819202122
  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. lines = f.readlines()
  15. meta_header = lines[0]
  16. content = ''.join(lines[1:])
  17. TrunksService().create_trunk({'file_path': relative_path, 'content': content,'type':'trunk','meta_header':meta_header})
  18. if __name__ == '__main__':
  19. directory = 'e:\\project\\knowledge\\utils\\files'
  20. FileReader.find_and_print_split_files(directory)