chaos/musicutil
1
0
Fork 0
Code Issues Releases Activity

add by folder

Browse source
This commit is contained in:
ChaotiCryptidz 2022-02-10 12:03:23 +00:00
parent 8a6502bf45
commit 6c6a3151ca
1 changed files with 28 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
musicutil/commands/genhtml_command.py
View file

@ -9,6 +9,7 @@ class GenHTMLArgs():
dest: str dest: str
title: str title: str
description: str description: str
by_folder: bool
def add_genhtml_command(subparsers): def add_genhtml_command(subparsers):
genhtml_parser = subparsers.add_parser('genhtml') genhtml_parser = subparsers.add_parser('genhtml')
@ -28,6 +29,9 @@ def add_genhtml_command(subparsers):
'description', 'description',
type=str, type=str,
help='description') help='description')
genhtml_parser.add_argument(
'--by-folder',
action='store_true')
def get_genhtml_args(args) -> GenHTMLArgs: def get_genhtml_args(args) -> GenHTMLArgs:
command_args = GenHTMLArgs() command_args = GenHTMLArgs()
@ -35,15 +39,18 @@ def get_genhtml_args(args) -> GenHTMLArgs:
command_args.dest = args.dest command_args.dest = args.dest
command_args.title = args.title command_args.title = args.title
command_args.description = args.description command_args.description = args.description
command_args.by_folder = args.by_folder
return command_args return command_args
def make_table_for_files(files: list[File]) -> str: def make_table_for_files(files: list[File], by_folder: bool = False) -> str:
html_content = "" html_content = ""
html_content += """ path_head = "<th>Path</th>" if not by_folder else ""
html_content += f"""
<table class="pure-table pure-table-horizontal"> <table class="pure-table pure-table-horizontal">
<thead> <thead>
<tr> <tr>
<th>Path</th> {path_head}
<th>Title</th> <th>Title</th>
<th>Artist</th> <th>Artist</th>
<th>Format</th> <th>Format</th>
@ -52,15 +59,18 @@ def make_table_for_files(files: list[File]) -> str:
<tbody> <tbody>
""" """
isOdd = True isOdd = True
print(files)
for file in files: for file in files:
tdClass = "pure-table-odd" if isOdd else "pure-table-even" tdClass = "pure-table-odd" if isOdd else "pure-table-even"
data_path = escape_html(file.path_from_src) data_path = escape_html(file.path_from_src)
data_title = escape_html(file.tags.title) data_title = escape_html(file.tags.title)
data_artist = escape_html(file.tags.artist) data_artist = escape_html(file.tags.artist)
data_extension = escape_html(file.extension) data_extension = escape_html(file.extension)
path_data = f"<td>{data_path}</td>" if not by_folder else ""
html_content += f""" html_content += f"""
<tr class="{tdClass}"> <tr class="{tdClass}">
<td>{data_path}</td> {path_data}
<td>{data_title}</td> <td>{data_title}</td>
<td>{data_artist}</td> <td>{data_artist}</td>
<td>{data_extension}</td> <td>{data_extension}</td>
@ -71,8 +81,7 @@ def make_table_for_files(files: list[File]) -> str:
</tbody> </tbody>
</table> </table>
""" """
return html_content
class GenHTMLCommand(): class GenHTMLCommand():
def __init__(self, args: GenHTMLArgs): def __init__(self, args: GenHTMLArgs):
@ -105,7 +114,19 @@ class GenHTMLCommand():
<body> <body>
""" """
html_content += make_table_for_files(files) if self.args.by_folder:
files_per_folder: dict[str, list[File]] = {}
for file in files:
if not file.path_from_src in files_per_folder.keys():
files_per_folder[file.path_from_src] = []
files_per_folder[file.path_from_src].append(file)
for folder in files_per_folder.keys():
folder_name = folder
folder_files = files_per_folder[folder_name]
html_content += f"<h1>{folder_name}</h1>"
html_content += make_table_for_files(folder_files, True)
else:
html_content += make_table_for_files(files)
html_content += """ html_content += """
</body> </body>