lvm · August 19, 2020 22:54 · Aug 19, 2020 · Aug 19, 2020
diff --git a/youtube-mp3 b/youtube-mp3
@@ -22,6 +22,17 @@ def ffmpeg(args: list) -> None:
     sp.call(shlex.split(f"ffmpeg -v quiet -y {args}"))
 
 
+def sanitize_album(album: str) -> str:
+    clean_these = [
+        "full album",  # from youtube probably"
+        "\d{4}",  # year
+        "(\[|\().*(\]|\))",
+        "HD",
+    ]
+    album = re.sub("|".join(clean_these), "", album, flags=re.IGNORECASE).strip()
+    return album
+
+
 def guess_band_song(audio_title: str) -> dict:
     "Guesses the band / album based on the (non-sliced) filename"
     dct = {"band":"", "song":""}
@@ -56,7 +67,7 @@ def save_mp3(video_url):
     title = ""
     YTDL_OPTS = {
         "quiet": True,
-        "restrictfilenames": True,
+        # "restrictfilenames": True,
         "writethumbnail": False,
         "ignoreerrors": True,
         "geo_bypass": True,
@@ -78,7 +89,7 @@ def save_mp3(video_url):
         ydl.download([video_url])
 
         title = ytdl_info.get("title")
-        audio_file = sanitize_filename(title, True, False)
+        audio_file = sanitize_filename(title, False, False)
         audio_file = f"{audio_file}.mp3"
         save_id3(title, audio_file)
 
@@ -92,7 +103,12 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     if args.video:
-        title, output = save_mp3(args.video)
+        video = (
+            args.video
+            if args.video.startswith("https://")
+            else f"https://www.youtube.com/watch?v={args.video}"
+        )
+        title, output = save_mp3(video)
         print (f"Title: {title}\nFile: {output}")
     else:
-        parser.print_help()
+        parser.print_help()
diff --git a/youtube-mp3 b/youtube-mp3
@@ -0,0 +1,98 @@
+#!/usr/bin/env python3
+
+import re
+import eyed3
+import shlex
+import string
+import argparse
+import youtube_dl
+import subprocess as sp
+from pathlib import Path
+from youtube_dl.utils import sanitize_filename
+
+
+USER_AGENT = "Mozilla/5.0 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)"
+BAND_ALBUM_RE = (
+    f"(?P<band>[{string.printable} ]+)( — | ▲ | - )(?P<song>[{string.printable}]+)"
+)
+
+
+def ffmpeg(args: list) -> None:
+    "Base ffmpeg call."
+    sp.call(shlex.split(f"ffmpeg -v quiet -y {args}"))
+
+
+def guess_band_song(audio_title: str) -> dict:
+    "Guesses the band / album based on the (non-sliced) filename"
+    dct = {"band":"", "song":""}
+    result = re.search(BAND_ALBUM_RE, audio_title)
+    if result:
+        dct = result.groupdict()
+        dct["band"] = dct.get("band").strip()
+        dct["song"] = sanitize_album(dct.get("song"))
+    else:
+        dct["band"] = "Various artists"
+        dct["song"] = audio_title
+
+    return dct
+
+
+def save_id3(title:str, audio_file:str) -> None:
+    gbs = guess_band_song(title)
+
+    audiofile = eyed3.load(audio_file)
+    audiofile.tag.artist = gbs.get("band")
+    audiofile.tag.album = gbs.get("album")
+    audiofile.tag.album_artist = gbs.get("band")
+    audiofile.tag.title = gbs.get("song")
+
+    audiofile.tag.save()
+
+
+
+def save_mp3(video_url):
+    "Get video info, save it as mp3 and save id3 tag"
+    assert video_url, "Missing video, can't continue"
+    title = ""
+    YTDL_OPTS = {
+        "quiet": True,
+        "restrictfilenames": True,
+        "writethumbnail": False,
+        "ignoreerrors": True,
+        "geo_bypass": True,
+        "format": "mp4",
+        "outtmpl": f"%(title)s.%(ext)s",
+        "format": "bestaudio/best",
+        "postprocessors": [
+            {
+                "key": "FFmpegExtractAudio",
+                "preferredcodec": "mp3",
+                "preferredquality": "192",
+            }
+        ]
+    }
+
+    with youtube_dl.YoutubeDL(YTDL_OPTS) as ydl:
+        youtube_dl.utils.std_headers["User-Agent"] = USER_AGENT
+        ytdl_info = ydl.extract_info(video_url, download=False)
+        ydl.download([video_url])
+
+        title = ytdl_info.get("title")
+        audio_file = sanitize_filename(title, True, False)
+        audio_file = f"{audio_file}.mp3"
+        save_id3(title, audio_file)
+
+    return title, audio_file
+
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("video", type=str, default="", help="Video URL")
+    args = parser.parse_args()
+
+    if args.video:
+        title, output = save_mp3(args.video)
+        print (f"Title: {title}\nFile: {output}")
+    else:
+        parser.print_help()