Add experimental file icon support

Author: BruceZhang1993

index_generator/__main__.py

@@ -62,7 +62,7 @@ def app(args):
         raise IndexGeneratorTemplateNotFound(str(e))
 
 
-def generate_once(theme, root, files, name, if_print, base=os.path.sep, human=False, template=''):
+def generate_once(theme, root, files, name, if_print, base=os.path.sep, human=False, template='', iconset='material'):
     if not template:
         environment = jinja2.Environment(
             loader=jinja2.PackageLoader('index_generator', 'templates/' + theme),
@@ -88,7 +88,8 @@ def generate_once(theme, root, files, name, if_print, base=os.path.sep, human=Fa
             'size': entry.size,
             'modified': time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(entry.modified)),
             'mime': entry.mime,
-            'isDir': entry.isDir
+            'isDir': entry.isDir,
+            'icon': entry.icon
         })
     html = template.render(ig={
         'root': base + root.lstrip('.*' + os.path.sep),
@@ -107,7 +108,7 @@ def generate_once(theme, root, files, name, if_print, base=os.path.sep, human=Fa
             print(html, file=f)
 
 
-def generate_recursively(theme, path, name, if_print, max_depth=0, base=os.path.sep, human=False, template=''):
+def generate_recursively(theme, path, name, if_print, max_depth=0, base=os.path.sep, human=False, template='', iconset='material'):
     os.chdir(path)
     for root, dirs, files in os.walk('.'):
         if max_depth != 0 and root.count(os.sep) >= max_depth:
@@ -124,7 +125,7 @@ def generate_recursively(theme, path, name, if_print, max_depth=0, base=os.path.
             print('files: {}'.format(files))
             print('-----------------------------------------')
 
-        generate_once(theme, root, dirs + files, name, if_print, base=base, human=human, template=template)
+        generate_once(theme, root, dirs + files, name, if_print, base=base, human=human, template=template, iconset=iconset)
 
 
 if __name__ == '__main__':

index_generator/icons/material/default.svg

@@ -1,5 +1,6 @@
 import os
 import mimetypes
+import base64
 
 
 def sizeof_fmt(num, suffix='B'):
@@ -14,7 +15,7 @@ def sizeof_fmt(num, suffix='B'):
 
 
 class Entry(object):
-    def __init__(self, file, root, base=os.path.sep, human=False):
+    def __init__(self, file, root, base=os.path.sep, human=False, iconset='material'):
         path = root + os.path.sep + file
         self.path = base + path.lstrip('.*' + os.path.sep)
         self.name = os.path.basename(path)
@@ -25,3 +26,5 @@ def __init__(self, file, root, base=os.path.sep, human=False):
             self.size = os.path.getsize(path)
         self.modified = os.path.getmtime(path)
         self.isDir = os.path.isdir(path)
+        with open(os.path.dirname(__file__) + os.path.sep + '..' + os.path.sep + 'icons' + os.path.sep + iconset + os.path.sep + 'default.svg', 'rb') as f:
+            self.icon = base64.b64encode(f.read()).decode()

index_generator/models/entries.py

@@ -6,25 +6,30 @@
 {% endblock %}
 {% block files %}
 <table>
-    <col width="600">
+    <col width="40">
+    <col width="560">
     <col width="180">
     <col width="300">
     <tr>
+        <th align="left"></th>
         <th align="left">Filename</th>
         <th align="left">Size</th>
         <th align="left">Last modified</th>
     </tr>
     <tr>
+        <td></td>
         <td><a href="../">Parent directory/</a></td>
         <td>-</td>
         <td>-</td>
     </tr>
     {% for file in ig.files %}
     <tr>
         {% if file.isDir %}
+        <td><img src="data:image/svg+xml;base64,{{ file.icon }}" width="30" /></td>
         <td><a href="{{ file.path }}">{{ file.name }}/</a></td>
         <td>-</td>
         {% else %}
+        <td><img src="data:image/svg+xml;base64,{{ file.icon }}" width="30" /></td>
         <td><a href="{{ file.path }}">{{ file.name }}</a></td>
         <td>{{ file.size }}</td>
         {% endif %}

index_generator/templates/default/index.html

@@ -18,7 +18,7 @@
         'index_generator.models'
     ],
     package_data={
-        'index_generator': ['templates/*/*']
+        'index_generator': ['templates/*/*', 'icons/*/*']
     },
     include_package_data=True,
     url=APP_URL,