Take browser user data dir from CLI arguments

Author: lubo

ytm/ytm/__main__.py

@@ -157,6 +157,11 @@ def sync_playlist(client, playlist_id):
 
 def main():
     arg_parser = argparse.ArgumentParser()
+    arg_parser.add_argument(
+        "browser_data_dir",
+        help="Browser user data directory",
+        type=Path,
+    )
     subparsers = arg_parser.add_subparsers(dest="command", required=True)
     subparsers.add_parser(
         "download-library",
@@ -179,9 +184,7 @@ def main():
 
     with sync_playwright() as p:
         browser = p.chromium.launch_persistent_context(
-            # Must not be Default.
-            # TODO: Load from CLI.
-            user_data_dir="/home/lubo/.config/google-chrome/Profile 13",
+            user_data_dir=args.browser_data_dir,
             executable_path=shutil.which("google-chrome-stable"),
             args=[
                 # Required for Google authentication.
@@ -196,7 +199,7 @@ def main():
         with page.expect_response(
             lambda response: (
                 response.request.method == "POST"
-                and urlsplit(response.url)[0:3]
+                and urlsplit(response.url)[:3]
                 == ("https", "music.youtube.com", "/youtubei/v1/browse")
             ),
             timeout=PLATWRIGHT_TIMEOUT,