basic functions achieved

edward-p · edward-p · commit 09d4a5d04c63 · 2019-06-04T03:23:53.000+08:00
diff --git a/index_generator/__main__.py b/index_generator/__main__.py
@@ -2,11 +2,18 @@
 # *-- coding: utf-8 --*
 import sys
 
+import os
+import os.path as path
+from datetime import datetime
+
 import jinja2
 import argparse
 
+indexIgnore=['index.html', 'templates']
 
 def main():
+    global template
+    global arguments
     parser = argparse.ArgumentParser()
     parser.add_argument('--version', '-V', action='store_true', default=False,
                         help='Print version infomation and quit.')
@@ -17,25 +24,52 @@ def main():
                         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),
+        loader=jinja2.PackageLoader('index_generator', arguments.template),
         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'
-            }
-        ]
-    }))
+    
+    generate()
+
+
+def generate(currentDir=''):
+    filelist=[]
+    dirlist=[]
+    for file in os.listdir():
+        if file in indexIgnore:
+            continue
+        if path.isdir(file):
+            dirlist.append({
+                'name': file,
+                'modified': datetime.fromtimestamp(path.getmtime(file)).strftime('%Y-%m-%d %H:%M')
+                })
+        else:
+            filelist.append({
+                'name': file,
+                'modified': datetime.fromtimestamp(path.getmtime(file)).strftime('%Y-%m-%d %H:%M'),
+                'size': path.getsize(file)
+                })
+    
+    index = template.render(ig={
+            'currentPath': currentDir,
+            'dirs': dirlist,
+            'files': filelist
+            })
+    if arguments.print:
+        print(index)
+
+    with open('index.html','w') as f:
+        print(index, file=f)
 
+    if not arguments.no_recursive and dirlist:
+        for file in dirlist:
+            if arguments.print:
+                print('------------------------------------------------')
+            os.chdir(file['name'])
+            generate(currentDir+'/'+file['name'])
+    
+    os.chdir('..')
 
 if __name__ == '__main__':
-    main()
+    main()
diff --git a/index_generator/templates/default/index.html b/index_generator/templates/default/index.html
@@ -14,9 +14,21 @@
         <th align="left">Last modified</th>
         <th align="left">Size</th>
     </tr>
+    <tr>
+        <td><a href="../">Parent directory/</a></td>
+        <td>-</td>
+        <td>-</td>
+    </tr>
+    {% for dir in ig.dirs %}
+    <tr>
+        <td><a href="{{ ig.currentPath }}/{{ dir.name }}">{{ dir.name }}/</a></td>
+        <td>{{ dir.modified }}</td>
+        <td>-</td>
+    </tr>
+    {% endfor %}
     {% for file in ig.files %}
     <tr>
-        <td><a href="#">{{ file.name }}</a></td>
+        <td><a href="{{ ig.currentPath }}/{{ file.name }}">{{ file.name }}</a></td>
         <td>{{ file.modified }}</td>
         <td>{{ file.size }}</td>
     </tr>
diff --git a/index_generator/templates/default/layout.html b/index_generator/templates/default/layout.html
@@ -2,12 +2,12 @@
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>{% block title %}Files: {{ ig.currentPath }}{% endblock %}</title>
+    <title>{% block title %}Files: {{ ig.currentPath }}/{% endblock %}</title>
     {% block extra %}{% endblock %}
 </head>
 <body>
 {% block head %}
-<h1>Index of {{ ig.currentPath }}</h1>
+<h1>Index of {{ ig.currentPath }}/</h1>
 <hr>
 {% block files %}
 {% endblock %}
