diff --git a/html_report.py b/html_report.py
new file mode 100644
index 0000000..ed08177
--- /dev/null
+++ b/html_report.py
@@ -0,0 +1,357 @@
+"""HTML report generation for missing albums"""
+
+from html import escape
+from typing import Dict, List
+
+
+def generate_html_report(albums_to_add: List[Dict], albums_to_update: List[Dict]):
+ """Generate an HTML report with clickable submission links"""
+ html_content = """
+
+
+
+
+ MusicBrainz Albums - Add & Update
+
+
+
+ 🎵 MusicBrainz Albums - Add & Update
+
+ Albums to ADD: {add_count} | Albums to UPDATE: {update_count}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ {artist_buttons}
+
+
+
+
{title}
+
by {artist}
+ {mb_info}
+ {issues_info}
+
+
+
Issues: {", ".join(issues)}
'
+
+ title = album.get("title", "Unknown Title")
+ artist = album.get("artist_name", "Unknown Artist")
+ artist_escaped = escape(artist)
+ atisket_link = submission_links.get("atisket_link", "#")
+ harmony_link = submission_links.get("harmony_link", "#")
+ deezer_url = submission_links.get("deezer_url", "#")
+
+ return album_html.format(
+ title=escape(title),
+ artist=artist,
+ artist_escaped=artist_escaped,
+ mb_info=mb_info,
+ issues_info=issues_info,
+ atisket_link=atisket_link,
+ harmony_link=harmony_link,
+ deezer_url=deezer_url,
+ )
+
+ all_albums = albums_to_add + albums_to_update
+ unique_artists = sorted(
+ set(album.get("artist_name", "Unknown") for album in all_albums)
+ )
+
+ artist_buttons_html = "".join(
+ f''
+ for artist in unique_artists
+ )
+
+ albums_html = ""
+ if albums_to_add:
+ albums_html += ''
+ albums_html += "
📥 Albums to ADD (Not in MusicBrainz)
"
+ formatted_add = map(lambda album: format_album(album, False), albums_to_add)
+ albums_html += "".join(formatted_add)
+ albums_html += ""
+
+ if albums_to_update:
+ albums_html += ''
+ albums_html += "
🔄 Albums to UPDATE (Need Linking/Updates)
"
+ formatted_update = map(
+ lambda album: format_album(album, True), albums_to_update
+ )
+ albums_html += "".join(formatted_update)
+ albums_html += ""
+
+ add_count = len(albums_to_add)
+ update_count = len(albums_to_update)
+ html_header = html_content.format(
+ add_count=add_count,
+ update_count=update_count,
+ artist_buttons=artist_buttons_html,
+ )
+ html_footer = "\n\n\n"
+ html_content = html_header + albums_html + html_footer
+
+ with open("missing_albums.html", "w", encoding="utf-8") as f:
+ f.write(html_content)
+ print(f"📄 HTML report saved to missing_albums.html")
diff --git a/main.py b/main.py
index dac37bc..6f3efb2 100755
--- a/main.py
+++ b/main.py
@@ -13,6 +13,8 @@ from urllib.parse import quote
import requests
from dotenv import load_dotenv
+from html_report import generate_html_report
+
load_dotenv()
@@ -342,179 +344,5 @@ def main():
generate_html_report(all_albums_to_add, all_albums_to_update)
-def generate_html_report(albums_to_add: List[Dict], albums_to_update: List[Dict]):
- """Generate an HTML report with clickable submission links"""
- html_content = """
-
-
-
-
- MusicBrainz Albums - Add & Update
-
-
-
- 🎵 MusicBrainz Albums - Add & Update
-
- Albums to ADD: {add_count} | Albums to UPDATE: {update_count}
-
-"""
-
- album_html = """
-
-
{title}
-
by {artist}
- {mb_info}
- {issues_info}
-
-
-
Issues: {", ".join(issues)}
'
-
- title = album.get("title", "Unknown Title")
- artist = album.get("artist_name", "Unknown Artist")
- atisket_link = submission_links.get("atisket_link", "#")
- harmony_link = submission_links.get("harmony_link", "#")
- deezer_url = submission_links.get("deezer_url", "#")
-
- return album_html.format(
- title=title,
- artist=artist,
- mb_info=mb_info,
- issues_info=issues_info,
- atisket_link=atisket_link,
- harmony_link=harmony_link,
- deezer_url=deezer_url,
- )
-
- albums_html = ""
- if albums_to_add:
- albums_html += "📥 Albums to ADD (Not in MusicBrainz)
"
- formatted_add = map(lambda album: format_album(album, False), albums_to_add)
- albums_html += "".join(formatted_add)
-
- if albums_to_update:
- albums_html += "🔄 Albums to UPDATE (Need Linking/Updates)
"
- formatted_update = map(
- lambda album: format_album(album, True), albums_to_update
- )
- albums_html += "".join(formatted_update)
-
- add_count = len(albums_to_add)
- update_count = len(albums_to_update)
- html_header = html_content.format(add_count=add_count, update_count=update_count)
- html_footer = "\n\n\n"
- html_content = html_header + albums_html + html_footer
-
- with open("missing_albums.html", "w", encoding="utf-8") as f:
- f.write(html_content)
- print(f"📄 HTML report saved to missing_albums.html")
-
-
if __name__ == "__main__":
main()