Browse Source

- changed variable names for clarity

- added f;ag to remove indexes
- made '--no-filename more consistent'
- styled verbose output better.
master
rra 5 years ago
parent
commit
52830ff7c4
  1. 180
      distribusi/distribusi.py
  2. 11
      distribusi/mappings.py

180
distribusi/distribusi.py

@ -7,7 +7,7 @@ import magic
from PIL import Image from PIL import Image
from distribusi.page_template import html_footer, html_head from distribusi.page_template import html_footer, html_head
from distribusi.mappings import CODE_TYPES, FILE_TYPES from distribusi.mappings import CODE_TYPES, FILE_TYPES, SUB_TYPES
MIME_TYPE = magic.Magic(mime=True) MIME_TYPE = magic.Magic(mime=True)
@ -43,99 +43,113 @@ def thumbnail(image, name, args):
"<figure><a href='{}'><img class='thumbnail' src='data:image/jpg;base64,{}'></a><figcaption>{}</figcaption></figure>" "<figure><a href='{}'><img class='thumbnail' src='data:image/jpg;base64,{}'></a><figcaption>{}</figcaption></figure>"
).format(name, data_url, cap) ).format(name, data_url, cap)
except Exception as e: except Exception as e:
print(e) print('Thumbnailer:',e)
return "<figure><a href='{}'><img class='thumbnail' src='{}'></a><figcaption>{}</figcaption></figure>".format(name, name,name) return "<figure><a href='{}'><img src='{}'></a><figcaption>{}</figcaption></figure>".format(name, name,name)
def div(args, mime, tag, *values): def div(args, type_, subtype, tag, name):
id_name = values[0].split('.')[0].replace(' ', '_') id_name = name.split('.')[0].replace(' ', '_')
if not args.no_filenames: if args.no_filenames:
filename = '<br><span class="filename">{}</span>'
else:
filename = '' filename = ''
if 'image' in mime:
html = '<div id="{}">{}</div>'
elif 'pdf' in mime:
html = '<div id="{}">{}' + filename + '</div>'
else: else:
html = '<div id="{}">{}</div>' filename = '<span class="filename">{}</span>'.format(name)
if 'image' in type_:
html = '<div id="{}" class="{}">{}</div>'
elif 'pdf' in subtype:
html = '<div id="{}" class="{}">{}' + filename + '</div>'
elif 'dir' in type_ or 'html' in subtype or 'unkown-file' in subtype:
html = '<div id="{}" class="{}">{}</div>'
else:
html = '<div id="{}" class="{}">{}' + filename +'</div>'
return html.format(id_name, tag, values[0]) return html.format(id_name, subtype, tag)
def distribusify(args, directory): # noqa def distribusify(args, directory): # noqa
for root, dirs, files in os.walk(directory): for root, dirs, files in os.walk(directory):
html = [] if not args.remove_index:
html = []
if args.verbose:
print('Listing', root)
for name in files:
if args.verbose: if args.verbose:
print('Adding', name) print('Generating directory listing for', root)
if 'index.html' not in name:
full_path = os.path.join(root, name) for name in files:
mime = MIME_TYPE.from_file(full_path) if 'index.html' not in name:
mime, format = mime.split('/') # example: plain text full_path = os.path.join(root, name)
mime = MIME_TYPE.from_file(full_path)
# example: MIME plain/text becomes 'type' plain 'subtype' text
type_, subtype = mime.split('/')
if args.verbose:
print(mime, format)
if mime in FILE_TYPES:
# expansion for different kind of textfiles
if mime == 'text':
if name.endswith('.html') or name.endswith('.txt'):
# what types of text files to expand
a = open(full_path).read()
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>"
else:
a = FILE_TYPES[mime]
if mime == 'image' and args.thumbnail:
a = thumbnail(full_path, name, args)
else:
if args.captions:
cap = caption(full_path)
else:
cap = name
a = FILE_TYPES[mime].format(full_path, cap)
if format in FILE_TYPES:
a = FILE_TYPES[format]
if mime not in FILE_TYPES and format not in FILE_TYPES:
# catch exceptions not defined in FILE_TYPES before
a = "<a href='{}'>{}</a>"
if args.verbose: if args.verbose:
message = 'mime-type not in list, adding as href: \n' print('Found', name,'as', mime)
print(message, mime, format, name)
if type_ in FILE_TYPES:
a = a.replace('{}', name) # expansion for different kind of textfiles
html.append(div(args, mime, a, name)) if type_ == 'text':
if name.endswith('.html') or subtype == 'html':
if root != directory: subtype = 'html'
html.append('<a href="../">../</a>') # what types of text files to expand
a = '<section id="{}">{}</section>'.format(name, open(full_path).read())
for name in dirs: elif subtype in CODE_TYPES or name.endswith('.txt'):
a = "<a href='{}' class='dir'>{}/</a>".replace('{}', name) # if the plain text is code,
html.append(div(args, 'dir', a, 'folder')) # which types do we wrap in pre-tags?
a = "<pre>" + open(full_path).read() + "</pre>"
with open(os.path.join(root, 'index.html'), 'w') as f: else:
if not args.no_template: subtype = subtype+' unkown-file'
if args.style: a = "<a href='{}'>{}</a>"
fs = open(args.style, "r") #a = FILE_TYPES[type_]
style = fs.read()
styled_html_head = html_head % style if type_ == 'image':
else: caption = name
styled_html_head = html_head % '' if args.thumbnail:
f.write(styled_html_head) a = thumbnail(full_path, name, args)
if args.captions:
for line in html: caption = caption(full_path)
f.write(line + '\n') a = FILE_TYPES[type_].format(name, caption)
if not args.no_template: if subtype in SUB_TYPES:
f.write(html_footer) a = SUB_TYPES[subtype]
if type_ not in FILE_TYPES and subtype not in SUB_TYPES:
# catch exceptions not yet defined in FILE_TYPES or SUB_TYPES
a = "<a href='{}'>{}</a>"
if args.verbose:
message = 'not in list of file types, adding as plain href: \n'
print(type_, subtype, message, name)
type_ ='unkown-file'
a = a.replace('{}', name)
html.append(div(args, type_, subtype, a, name))
if root != directory:
html.append('<a href="../">../</a>')
for name in dirs:
a = "<a href='{}'>{}/</a>".replace('{}', name)
html.append(div(args,'dir', 'dir', a, 'folder'))
with open(os.path.join(root, 'index.html'), 'w') as f:
if not args.no_template:
if args.style:
fs = open(args.style, "r")
style = fs.read()
styled_html_head = html_head % style
else:
styled_html_head = html_head % ''
f.write(styled_html_head)
for line in html:
f.write(line + '\n')
if not args.no_template:
f.write(html_footer)
if args.remove_index:
index = os.path.join(root, 'index.html')
if 'index.html' in files:
if args.verbose:
print('Removing index.html from', root)
try:
os.remove(index)
except Exception as e:
print(e)

11
distribusi/mappings.py

@ -1,13 +1,14 @@
CODE_TYPES = ['x-c', 'html'] CODE_TYPES = ['x-c', 'x-shellscript', 'x-python']
FILE_TYPES = { FILE_TYPES = {
'image': '<figure><img class="image" src="{}"><figcaption>{}</figcaption></figure>', 'image': '<figure><img class="image" src="{}"><figcaption>{}</figcaption></figure>',
'pdf': (
'<object data="{}" class="pdf" type="application/pdf">'
'<embed src="{}" type="application/pdf" /></object>'
),
'text': '<a href="{}" class="text">{}</a>', 'text': '<a href="{}" class="text">{}</a>',
'video': ('<video class="video" controls>' '<source src="{}"></source></video>'), 'video': ('<video class="video" controls>' '<source src="{}"></source></video>'),
'audio': ('<audio controls class="audio">' '<source src="{}"></source></audio>'), 'audio': ('<audio controls class="audio">' '<source src="{}"></source></audio>'),
} }
SUB_TYPES ={
'pdf': (
'<object data="{}" class="pdf" type="application/pdf">'
'<embed src="{}" type="application/pdf" /></object>')
}

Loading…
Cancel
Save