Except YTMusicUserError in auth.py fixes #110 (#111)

Add exceptions thrown by `ytmusicapi`
apastel · Feb 24, 2025 · 565d7fd · 565d7fd
1 parent c86fc1c
commit 565d7fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ytmusic_deleter/auth.py b/ytmusic_deleter/auth.py
@@ -4,6 +4,7 @@
 
 import ytmusicapi
 from ytmusicapi import YTMusic
+from ytmusicapi.exceptions import YTMusicUserError
 
 from . import common as const
 
@@ -23,7 +24,7 @@ def ensure_auth(credential_dir, oauth) -> YTMusic:
         logging.info(f"Attempting authentication with: {auth_file_path}")
         yt_auth = YTMusic(auth_file_path)
         logging.info(f'Authenticated with: {auth_file_path}"')
-    except JSONDecodeError:
+    except (JSONDecodeError, YTMusicUserError, FileNotFoundError):
         logging.info(f"Creating file: {auth_file_name}")
         if oauth:
             ytmusicapi.setup_oauth(filepath=auth_file_path, open_browser=True)