loader.py 888 B

123456789101112131415161718192021222324252627
  1. # -*- coding: utf-8 -*-
  2. from .__load__ import *
  3. class Loader(object):
  4. def get(self, file, param = {}):
  5. if 'http' in file:
  6. loader = 'web'
  7. elif File.exists(file):
  8. if '.csv' in file:
  9. loader = 'csv'
  10. elif '.pdf' in file:
  11. loader = 'pdf'
  12. elif '.html' in file:
  13. loader = 'html'
  14. elif '.json' in file:
  15. loader = 'json'
  16. elif '.xls' in file:
  17. loader = 'excel'
  18. elif '.ppt' in file:
  19. loader = 'ppt'
  20. elif '.doc' in file:
  21. loader = 'word'
  22. elif '.jpg' in file or '.png' in file or '.gif' in file or '.webp' in file:
  23. loader = 'img'
  24. else:
  25. loader = 'text'
  26. return Demeter.service(loader, 'extract.docs').init(file, param)