90 lines
2.9 KiB
Python
Executable File
90 lines
2.9 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
''' A script that creates index.html indexes for a folder.
|
|
'''
|
|
|
|
import os,sys
|
|
import urllib
|
|
INDEXFILE='index.html'
|
|
|
|
def setup():
|
|
''' Setup the command line options '''
|
|
from argparse import ArgumentParser
|
|
parser=ArgumentParser()
|
|
parser.add_argument("-f",action="store_true",dest="overwrite",default=False,
|
|
help="Overwrite existing "+INDEXFILE)
|
|
parser.add_argument("-H",action="store_true",dest="hidden",default=False,
|
|
help="Show hidden files")
|
|
parser.add_argument("-t",type=str,dest="title",default=None,
|
|
help="Name for the title (Default: Folder name)")
|
|
parser.add_argument("-p",action="store_false",dest="parent",default=True,
|
|
help="Do no print .. link for parent folder.")
|
|
parser.add_argument("startpath",type=str,action="store",default=os.path.abspath('.'),nargs='?',
|
|
help="Root path of the index")
|
|
options=parser.parse_args()
|
|
options.startpath=os.path.abspath(options.startpath)
|
|
if options.title==None:
|
|
options.title=os.path.basename(options.startpath)
|
|
return options
|
|
|
|
def generate_index(opts):
|
|
for path,dirs,files in os.walk(opts.startpath):
|
|
if INDEXFILE in files:
|
|
if not opts.overwrite:
|
|
print(INDEXFILE+" exists")
|
|
sys.exit(1)
|
|
files = [ f for f in files if f != INDEXFILE]
|
|
if not opts.hidden:
|
|
files = [ f for f in files if not f.startswith(".")]
|
|
dirs = [ d for d in dirs if not d.startswith(".")]
|
|
f=open(os.path.join(path,INDEXFILE),'wt')
|
|
dirs.sort()
|
|
files.sort()
|
|
f.write(get_header(opts.title))
|
|
if opts.parent:
|
|
f.write(get_pathlink('..'))
|
|
for pa in dirs:
|
|
f.write(get_pathlink(pa))
|
|
for fi in files:
|
|
f.write(get_filelink(path,fi))
|
|
f.write(get_footer())
|
|
f.close()
|
|
return
|
|
|
|
def get_filelink(path,fname):
|
|
fsize=os.path.getsize(os.path.join(path,fname))
|
|
fsstr=sizeof(fsize)
|
|
return '<tr><td><a href="'+urllib.quote(fname)+'">'+fname+'</a><td>'+fsstr+'</td></tr>\n'
|
|
|
|
|
|
def get_pathlink(path):
|
|
return '<tr><td><a href="'+urllib.quote(path)+'">'+path+'</a><td></td></tr>\n'
|
|
|
|
def get_header(title):
|
|
header='''<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">
|
|
<html>
|
|
<head>
|
|
<title>Index of '''+title+'''</title>
|
|
</head>
|
|
<body>
|
|
<h1>Index of '''+title+'''</h1>
|
|
<table><tr><th>Name</th><th>Size</th></tr>
|
|
'''
|
|
return header
|
|
|
|
def get_footer():
|
|
footer='''</table>
|
|
</body></html>'''
|
|
return footer
|
|
|
|
def sizeof(num):
|
|
for x in ['bytes','KB','MB','GB','TB']:
|
|
if num < 1024.0:
|
|
if x=='bytes':
|
|
return "%d %s" % (num, x)
|
|
return "%3.1f %s" % (num, x)
|
|
num /= 1024.0
|
|
|
|
opts=setup()
|
|
generate_index(opts)
|