implement no recursive print power

Author: BruceZhang1993

.gitignore

@@ -1,3 +1,4 @@
 .idea
 *.iml
 __pycache__
+/*.html

index_generator/__init__.py

@@ -0,0 +1,3 @@
+APP_NAME = 'Index Generator'
+APP_VERSION = '0.0.1'
+APP_URL = 'https://github.com/BruceZhang1993/index-generator'

index_generator/__main__.py

@@ -1,9 +1,15 @@
 #!/usr/bin/env python3
 # *-- coding: utf-8 --*
 import sys
+import time
 
 import jinja2
 import argparse
+import os
+
+from index_generator.models.entries import Entry
+from index_generator.models.exceptions import IndexGeneratorException
+from . import *
 
 
 def main():
@@ -15,27 +21,50 @@ def main():
     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.')
+    parser.add_argument('--print', '-P', action='store_true', default=False, help='Whether to print to stdout.')
+    parser.add_argument('path', type=str, default='.', help='Path')
     arguments = parser.parse_args(sys.argv[1:])
+    app(arguments)
+
+
+def app(args):
+    if args.version:
+        print(APP_NAME + ' ' + APP_VERSION + ' ' + APP_URL)
+        sys.exit(0)
+    if args.no_recursive:
+        generate_once(args.template, args.path, args.name, args.print)
+    else:
+        raise IndexGeneratorException(IndexGeneratorException.NOT_IMPLEMENTED)
 
 
-def generate(templateDir):
+def generate_once(template_dir, path='.', name='index.html', if_print=False):
     environment = jinja2.Environment(
-        loader=jinja2.PackageLoader('index_generator', templateDir),
+        loader=jinja2.PackageLoader('index_generator', template_dir),
         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'
-            }
-        ]
-    }))
+    template = environment.get_template(name)
+    entries = list(map(lambda f: Entry(f), os.listdir(path)))
+    files = []
+    for entry in entries:
+        files.append({
+            'path':     entry.path,
+            'name':     entry.name,
+            'size':     entry.size,
+            'modified': time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(entry.modified)),
+            'mime':     entry.mime
+        })
+    html = template.render(ig={
+        'currentPath': '/',
+        'files': files
+    })
+    if if_print:
+        print(html)
+    else:
+        raise IndexGeneratorException(IndexGeneratorException.NOT_IMPLEMENTED)
 
 
 if __name__ == '__main__':
-    main()
+    try:
+        main()
+    except Exception as e:
+        print(e)

index_generator/models/entries.py

@@ -0,0 +1,11 @@
+import os
+import mimetypes
+
+
+class Entry(object):
+    def __init__(self, path, root='/'):
+        self.name = os.path.basename(path)
+        self.mime = mimetypes.guess_type(path)[0]
+        self.path = root + path
+        self.size = os.path.getsize(path)
+        self.modified = os.path.getmtime(path)

index_generator/models/exceptions.py

@@ -0,0 +1,2 @@
+class IndexGeneratorException(BaseException):
+    NOT_IMPLEMENTED = 'Power has not been implemented.'

index_generator/templates/default/index.html

@@ -16,7 +16,7 @@
     </tr>
     {% for file in ig.files %}
     <tr>
-        <td><a href="#">{{ file.name }}</a></td>
+        <td><a href="{{ file.path }}">{{ file.name }}</a></td>
         <td>{{ file.modified }}</td>
         <td>{{ file.size }}</td>
     </tr>