no more args, just run distribusi

This commit is contained in:
crunk 2024-12-21 12:35:00 +01:00
parent c8862dfb59
commit cc25eb73c1
6 changed files with 67 additions and 191 deletions

View File

@ -1,89 +0,0 @@
import argparse
import os
from distribusi.distribusi import distribusify
def build_argparser():
parser = argparse.ArgumentParser(
"""
distribusi is a content management system for the web that produces static
index pages based on folders in the files system. It is inspired by the
automatic index functions featured in several popular web servers.
distribusi works by traversing the file system and directory hierarchy to
automatically list all the files in the directory, detect the file types
and providing them with relevant html classes and tags for easy styling.
"""
)
parser.add_argument(
"-d",
"--directory",
help="Select which directory to distribute",
default=".",
)
parser.add_argument(
"-s", "--style", help="Select a CSS style sheet to include"
)
parser.add_argument(
"-v",
"--verbose",
help="Print verbose debug output",
action="store_true",
)
parser.add_argument(
"-t",
"--thumbnail",
help="Generate 450x450 thumbnails for images",
action="store_true",
)
parser.add_argument(
"-a",
"--alttexts",
help="Adds file alttext based on same named files",
action="store_true",
)
parser.add_argument(
"-r",
"--remove-index",
help="Recursively removes all instances of index.html that have been previously made by distribusi",
action="store_true",
)
parser.add_argument(
"-e",
"--exclude-directory",
help="Exclude one or multiple directories from indexing",
nargs="*",
metavar="DIR",
)
parser.add_argument(
"-f",
"--force",
help="Force whether distribusi overwrites or removes instances of index.html not generated by distribusi, use at own risk!",
action="store_true",
)
parser.add_argument(
"--no-hidden", help="Exclude hidden directories", action="store_true"
)
parser.add_argument(
"--menu-with-index",
help="Append index.html to menu items to aid navigation",
action="store_true",
)
return parser
def cli_entrypoint():
parser = build_argparser()
args = parser.parse_args()
distribusify(args, args.directory)

View File

@ -46,7 +46,7 @@ def _read_matching_text_file(image_filename_no_ext, filename):
return text_file.read()
def thumbnail(full_path_image, name, args):
def make_thumbnail(full_path_image, name):
if full_path_image.endswith("_thumbnail.jpg"):
return
try:
@ -68,7 +68,7 @@ def thumbnail(full_path_image, name, args):
return
def format_div(args, type_, subtype, tag, name):
def format_div(type_, subtype, tag, name):
id_name = name.split(".")[0].replace(" ", "_")
filename = f'<span class="filename">{name}</span>'
@ -84,25 +84,19 @@ def format_div(args, type_, subtype, tag, name):
return html.format(id_name, subtype, tag)
def check_distribusi_index(args, index):
def check_distribusi_index(index):
"""
check whether a index.html file is generated by distribusi
"""
if not args.force:
with open(index, "r") as f:
if '<meta name="generator" content="distribusi" />' in f.read():
return True
else:
if args.verbose:
print(index, "not generated by distribusi, skipping")
return False
elif args.force:
return True
def write_index_html(args, index, html):
def write_index_html(index, html, cssfile):
with open(index, "w") as index_file:
styled_html_head = html_head.format(cssfile=args.style)
styled_html_head = html_head.format(cssfile=cssfile)
index_file.write(styled_html_head)
for line in html:
index_file.write(line + "\n")
@ -127,9 +121,8 @@ def handle_text_files(name, full_path, subtype):
return subtype, tag
def handle_image_files(name, full_path, args):
if args.thumbnail:
thumbnail_filename = thumbnail(full_path, name, args)
def handle_image_files(name, full_path):
thumbnail_filename = make_thumbnail(full_path, name)
if thumbnail_filename is None:
return
@ -158,16 +151,13 @@ def handle_image_files(name, full_path, args):
image_alttext=image_alttext,
image_description=image_description,
)
return FILE_TYPES["image"].format(name, image_alttext)
def remove_index_html(root, files, args):
def remove_index_html(root, files):
index = os.path.join(root, "index.html")
if "index.html" in files:
try:
if check_distribusi_index(args, index):
if args.verbose:
print("Removing index.html from", root)
if check_distribusi_index(index):
os.remove(index)
except Exception as e:
print(e)
@ -180,7 +170,7 @@ def remove_hidden_files_and_folders(dirs, files):
return dirs, files
def is_skipped_file(name, args):
def is_skipped_file(name, cssfile):
if "index.html" in name:
return True
@ -193,32 +183,18 @@ def is_skipped_file(name, args):
if name.endswith("_dv_description.txt"):
return True
if args.style in name:
if cssfile in name:
return True
def distribusify(args, directory): # noqa
def distribusify(cssfile, directory):
for root, dirs, files in os.walk(directory):
html = []
if args.exclude_directory:
if args.verbose:
print(
"Excluding directory:", ", ".join(args.exclude_directory)
)
dirs[:] = [d for d in dirs if d not in args.exclude_directory]
if args.no_hidden:
dirs, files = remove_hidden_files_and_folders(dirs, files)
if args.remove_index:
remove_index_html(root, files, args)
if args.verbose:
print("Generating directory listing for", root)
remove_index_html(root, files)
for name in sorted(files):
if is_skipped_file(name, args):
if is_skipped_file(name, cssfile):
continue
full_path = os.path.join(root, name)
@ -226,8 +202,6 @@ def distribusify(args, directory): # noqa
type_, subtype = mime.split("/")
alttext = name
if args.verbose:
print("Found", name, "as", mime)
if type_ in FILE_TYPES:
match type_:
case "text":
@ -235,7 +209,7 @@ def distribusify(args, directory): # noqa
name, full_path, subtype
)
case "image":
tag = handle_image_files(name, full_path, args)
tag = handle_image_files(name, full_path)
if tag is None:
continue
case _:
@ -247,29 +221,16 @@ def distribusify(args, directory): # noqa
if type_ not in FILE_TYPES and subtype not in SUB_TYPES:
# catch exceptions not yet defined in FILE_TYPES or SUB_TYPES
tag = "<a href='{}'>{}</a>"
if args.verbose:
message = (
"not in list of file types, adding as plain href: \n"
)
print(type_, subtype, message, name)
subtype = subtype + " unkown-file"
tag = tag.replace("{}", name)
html.append(format_div(args, type_, subtype, tag, name))
html.append(format_div(type_, subtype, tag, name))
if root != directory:
if args.menu_with_index:
html.append('<a href="../index.html">../</a>')
else:
html.append('<a href="../">../</a>')
for name in sorted(dirs):
if args.menu_with_index:
tag = "<a href='{}/index.html'>{}</a>".replace("{}", name)
else:
tag = "<a href='{}'>{}/</a>".replace("{}", name)
html.insert(0, format_div(args, "dir", "dir", tag, "folder"))
html.insert(0, format_div("dir", "dir", tag, "folder"))
index = os.path.join(root, "index.html")
write_index_html(args, index, html)
write_index_html(index, html, cssfile)

View File

@ -1,4 +1,4 @@
html_head = '''\
html_head = """\
<!DOCTYPE html>
<html lang="en">
<head>
@ -8,9 +8,9 @@ html_head = '''\
<link rel="stylesheet" href="{cssfile}" crossorigin="anonymous">
</head>
<body>
'''
"""
html_footer ='''\
html_footer = """\
</body>
</html>
'''
"""

View File

@ -112,6 +112,10 @@ def _delete_distribusi_files(distribusiname):
cssfolder = os.path.join("themes/userthemes", distribusiname)
if os.path.exists(cssfolder):
shutil.rmtree(cssfolder)
distribusi = Distribusis.query.filter_by(
distribusiname=distribusiname
).first()
if distribusi.publictheme is not None:
publicthemefolder = os.path.join("themes/publicthemes", distribusiname)
if os.path.exists(publicthemefolder):

View File

@ -79,7 +79,9 @@ def upload_updates_files(uploadfolder):
db.session.commit()
except (InvalidRequestError, DataError, InterfaceError, DatabaseError):
db.session.rollback()
uploadform.sitename.errors.append("Something went wrong with the database!")
uploadform.sitename.errors.append(
"Something went wrong with the database!"
)
zipfilename = "{}.zip".format(distribusi.distribusiname)
zipfile = uploadform.zipfile.data

View File

@ -14,19 +14,17 @@ config = context.config
# Interpret the config file for Python logging.
# This line sets up loggers basically.
fileConfig(config.config_file_name)
logger = logging.getLogger("alembic.env")
logger = logging.getLogger('alembic.env')
# add your model's MetaData object here
# for 'autogenerate' support
# from myapp import mymodel
# target_metadata = mymodel.Base.metadata
config.set_main_option(
"sqlalchemy.url",
str(current_app.extensions["migrate"].db.get_engine().url).replace(
"%", "%%"
),
)
target_metadata = current_app.extensions["migrate"].db.metadata
'sqlalchemy.url',
str(current_app.extensions['migrate'].db.get_engine().url).replace(
'%', '%%'))
target_metadata = current_app.extensions['migrate'].db.metadata
# other values from the config, defined by the needs of env.py,
# can be acquired:
@ -67,20 +65,20 @@ def run_migrations_online():
# when there are no changes to the schema
# reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html
def process_revision_directives(context, revision, directives):
if getattr(config.cmd_opts, "autogenerate", False):
if getattr(config.cmd_opts, 'autogenerate', False):
script = directives[0]
if script.upgrade_ops.is_empty():
directives[:] = []
logger.info("No changes in schema detected.")
logger.info('No changes in schema detected.')
connectable = current_app.extensions["migrate"].db.get_engine()
connectable = current_app.extensions['migrate'].db.get_engine()
with connectable.connect() as connection:
context.configure(
connection=connection,
target_metadata=target_metadata,
process_revision_directives=process_revision_directives,
**current_app.extensions["migrate"].configure_args,
**current_app.extensions['migrate'].configure_args
)
with context.begin_transaction():