dickreckard
/
distribusi
forked from varia/distribusi
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
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
b613eef36c
commit
52830ff7c4
2 changed files with 103 additions and 88 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 180
      distribusi/distribusi.py
  2. 11
      distribusi/mappings.py

180
distribusi/distribusi.py
View File

@ -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
View File

@ -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>')
}

Write Preview
Loading…
Cancel
Save