main_view.py 1.2 KB

1234567891011121314151617181920212223242526272829303132
  1. from .tool.func import *
  2. def main_view(name = ''):
  3. with get_db_connect() as conn:
  4. file_name = re.search(r'([^/]+)$', name)
  5. if not file_name:
  6. return ''
  7. else:
  8. file_name = file_name.group(1)
  9. dir_name = './views/' + re.sub(r'\.{2,}', '', name[:-len(file_name)])
  10. file_name = re.sub(r'\.cache_v(?:[0-9]+)$', '', file_name)
  11. mime_type = file_name.split('.')
  12. if len(mime_type) < 2:
  13. mime_type = 'text/plain'
  14. else:
  15. mime_type = mime_type[len(mime_type) - 1].lower()
  16. image_type = ['jpeg', 'jpg', 'gif', 'png', 'webp', 'ico', 'svg']
  17. if mime_type in image_type:
  18. if not mime_type == 'svg':
  19. mime_type = 'image/' + mime_type
  20. else:
  21. mime_type = 'image/svg+xml'
  22. elif mime_type == 'js':
  23. mime_type = 'text/javascript'
  24. elif mime_type == 'txt':
  25. mime_type = 'text/plain'
  26. else:
  27. mime_type = 'text/' + mime_type
  28. return flask.send_from_directory(dir_name, file_name, mimetype = mime_type)