Initial commit

BruceZhang1993 · BruceZhang1993 · commit f4e80ceb338f · 2019-06-03T21:45:18.000+08:00
diff --git a/index_generator/__init__.py b/index_generator/__init__.py
diff --git a/index_generator/__main__.py b/index_generator/__main__.py
@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+# *-- coding: utf-8 --*
+import sys
+
+import jinja2
+import argparse
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--version', '-V', action='store_true', default=False,
+                        help='Print version infomation and quit.')
+    parser.add_argument('--template', '-t', type=str, default='templates/default',
+                        help='Select template to generate html.')
+    parser.add_argument('--no-recursive', action='store_true', default=False, help='Do not generate recursively.')
+    parser.add_argument('--name', '-n', type=str, default='index.html',
+                        help='Default output filename.')
+    parser.add_argument('--print', '-p', action='store_true', default=False, help='Whether to print to stdout.')
+    arguments = parser.parse_args(sys.argv[1:])
+
+
+def generate(templateDir):
+    environment = jinja2.Environment(
+        loader=jinja2.PackageLoader('index_generator', templateDir),
+        autoescape=jinja2.select_autoescape(['html', 'htm'])
+    )
+    template = environment.get_template('index.html')
+    print(template.render(ig={
+        'currentPath': '/example',
+        'files': [
+            {
+                'name': 'test.txt',
+                'modified': '-',
+                'size': '12'
+            }
+        ]
+    }))
+
+
+if __name__ == '__main__':
+    main()
diff --git a/index_generator/templates/default/index.html b/index_generator/templates/default/index.html
@@ -0,0 +1,25 @@
+{% extends "layout.html" %}
+{% block extra %}
+<style>
+    body { background-color: #F8F8F8; }
+</style>
+{% endblock %}
+{% block files %}
+<table>
+    <col width="800">
+    <col width="200">
+    <col width="80">
+    <tr>
+        <th align="left">Filename</th>
+        <th align="left">Last modified</th>
+        <th align="left">Size</th>
+    </tr>
+    {% for file in ig.files %}
+    <tr>
+        <td><a href="#">{{ file.name }}</a></td>
+        <td>{{ file.modified }}</td>
+        <td>{{ file.size }}</td>
+    </tr>
+    {% endfor %}
+</table>
+{% endblock %}
diff --git a/index_generator/templates/default/layout.html b/index_generator/templates/default/layout.html
@@ -0,0 +1,18 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>{% block title %}Files: {{ ig.currentPath }}{% endblock %}</title>
+    {% block extra %}{% endblock %}
+</head>
+<body>
+{% block head %}
+<h1>Index of {{ ig.currentPath }}</h1>
+<hr>
+{% block files %}
+{% endblock %}
+<hr>
+{% endblock %}
+<div>{% block footer %}Generated by index-generator.{% endblock %}</div>
+</body>
+</html>
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1 @@
+Jinja2
