loader.py 860 B

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