Browse Source

Clean up imports and formatting

pull/3/head
decentral1se 3 years ago
parent
commit
df71f44550
No known key found for this signature in database GPG Key ID: 5E2EF5A63E3718CC
  1. 30
      distribusi/cli.py
  2. 27
      distribusi/distribusi.py

30
distribusi/cli.py

@ -2,61 +2,55 @@ import argparse
from distribusi.distribusi import distribusify
def build_argparser():
parser = argparse.ArgumentParser("""
parser = argparse.ArgumentParser(
"""
distbusi is a content management system for the web that produces static
index pages based on folders in the filesystem. It is inspired by the
automatic index functions featured in several web servers. It works by
traversing the file system and directory hierarchy to automatically list
all the files in the directory and providing them with html classes and
tags for easy styling.
""")
parser.add_argument(
'-d',
'--directory',
help="Select which directory to distribute"
"""
)
parser.add_argument(
'-s',
'--style',
help="Select a CSS style to include"
'-d', '--directory', help="Select which directory to distribute"
)
parser.add_argument('-s', '--style', help="Select a CSS style to include")
parser.add_argument(
'-v',
'--verbose',
help="Print verbose debug output",
action="store_true"
'-v', '--verbose', help="Print verbose debug output", action="store_true"
)
parser.add_argument(
'-t',
'--thumbnail',
help="Generate 150x150 thumbnails for images",
action="store_true"
action="store_true",
)
parser.add_argument(
'-n',
'--no-template',
help="Don't use the template to output html",
action="store_true"
action="store_true",
)
parser.add_argument(
'-nf',
'--no-filenames',
help="Don't include image filenames",
action="store_true"
action="store_true",
)
parser.add_argument(
'-c',
'--captions',
help="Print captions stored in exif metadata",
action="store_true"
action="store_true",
)
return parser

27
distribusi/distribusi.py

@ -1,16 +1,14 @@
import base64
import os
import subprocess
from io import BytesIO
import magic
from distribusi.page_template import html_footer, html_head
from PIL import Image
import subprocess
CODE_TYPES = [
'x-c',
'html'
]
from distribusi.page_template import html_footer, html_head
CODE_TYPES = ['x-c', 'html']
FILE_TYPES = {
'image': '<figure><img class="image" src="{}">{}</figure>',
@ -19,19 +17,14 @@ FILE_TYPES = {
'<embed src="{}" type="application/pdf" /></object>'
),
'text': '<a href="{}" class="text">{}</a>',
'video': (
'<video class="video" controls>'
'<source src="{}"></source></video>'
),
'audio': (
'<audio controls class="audio">'
'<source src="{}"></source></audio>'
),
'video': ('<video class="video" controls>' '<source src="{}"></source></video>'),
'audio': ('<audio controls class="audio">' '<source src="{}"></source></audio>'),
}
MIME_TYPE = magic.Magic(mime=True)
def caption(image):
process = subprocess.Popen(['exiftool', '-Comment', image], stdout=subprocess.PIPE)
out, err = process.communicate()
@ -41,6 +34,7 @@ def caption(image):
caption = ''
return caption
def thumbnail(image, name, args):
size = (450, 450)
im = Image.open(image)
@ -58,6 +52,7 @@ def thumbnail(image, name, args):
"<figure><a href='{}'><img class='thumbnail' src='data:image/jpg;base64,{}'></a>{}</figure>"
).format(name, data_url, cap)
def div(args, mime, tag, *values):
id_name = values[0].split('.')[0].replace(' ', '_')
if not args.no_filenames:
@ -101,7 +96,7 @@ def distribusify(args, directory): # noqa
elif format in CODE_TYPES:
# if the plain text is code,
# which types do we wrap in pre-tags?
a = "<pre>"+open(full_path).read()+"</pre>"
a = "<pre>" + open(full_path).read() + "</pre>"
else:
a = FILE_TYPES[mime]
@ -146,7 +141,7 @@ def distribusify(args, directory): # noqa
f.write(styled_html_head)
for line in html:
f.write(line+'\n')
f.write(line + '\n')
if not args.no_template:
f.write(html_footer)

Loading…
Cancel
Save