Use Sanic

utils.py

@@ -0,0 +1,97 @@
+from datetime import datetime
+from mimetypes import guess_type
+from pathlib import Path
+
+from jinja2 import Environment, FileSystemLoader
+
+from conf import BASE_HOSTS, SORT_KEYS, ICONS
+
+
+def icon_html(cls):
+    return f'<i class="{cls}"></i>'
+
+
+class ListEntry:
+    def __init__(self, path: Path):
+        self.name = path.name
+        self.mime = guess_type(self.name)[0]
+        self.is_dir = path.is_dir()
+        self.size = path.stat().st_size
+        self.created = path.stat().st_ctime
+
+    @property
+    def link_name(self):
+        if self.is_dir:
+            return f'{self.name}/'
+        return self.name
+
+    @property
+    def display_name(self):
+        if self.is_dir:
+            return f'{self.name}/'
+        return f'{self.name} [{self.mime}]'
+
+    @property
+    def icon(self):
+        if self.name == '..':
+            return icon_html(ICONS['types']['parent'])
+        elif self.is_dir:
+            return icon_html(ICONS['types']['dir'])
+        elif self.mime:
+            t = self.mime.split('/')[0]
+            if t in ICONS['types']:
+                return icon_html(ICONS['types'][t])
+        return icon_html(ICONS['types']['file'])
+
+    @property
+    def formatted_date(self):
+        return datetime.fromtimestamp(self.created).strftime('%Y-%m-%d %H:%M:%S')
+
+    @property
+    def formatted_size(self):
+        size = self.size
+        for unit in ['B', 'KiB', 'MiB', 'GiB', 'TiB']:
+            if size < 2048:
+                return f'{size} {unit}'
+            else:
+                size //= 1024
+        return f'{size} PiB'
+
+
+def get_j2env(debug=False):
+    return Environment(loader=FileSystemLoader([Path('.').absolute()]),
+                       trim_blocks=True, optimized=debug, cache_size=0 if debug else 400)
+
+
+def get_sort_icon(test, sort):
+    if sort == test:
+        return icon_html(ICONS['sort_asc'])
+    elif sort == f'-{test}':
+        return icon_html(ICONS['sort_desc'])
+    else:
+        return ''
+
+
+def get_sort_link(current, sort, hidden):
+    return f'?sort={"-" if current == sort else ""}{sort}{"&hidden" if hidden else ""}'
+
+
+def resolve_path(domain, path):
+    path_parts = [i for i in path.split('/') if i not in ['.', '..']]
+    joined_parts = '/'.join(path_parts)
+    if domain in BASE_HOSTS:
+        return Path('files').joinpath(*path_parts), joined_parts
+    else:
+        for host in BASE_HOSTS:
+            if host in domain:
+                return Path('subdomain_files').joinpath(domain[:domain.index(host)-1], *path_parts), joined_parts
+
+    raise ValueError
+
+
+def list_dir(directory: Path, sort, hidden=False, root=True):
+    lst = [ListEntry(path) for path in directory.iterdir() if hidden or not path.name.startswith('.')]
+    lst = sorted(lst, key=SORT_KEYS[sort], reverse=sort.startswith('-'))
+    if not root:
+        lst = [ListEntry(Path('..'))] + lst
+    return lst