From 4bbffa6dc3469832609b6e56ae926dcdf7e729ac Mon Sep 17 00:00:00 2001
From: tech234a <46801700+tech234a@users.noreply.github.com>
Date: Tue, 22 Sep 2020 21:37:12 -0400
Subject: [PATCH] Retry on account error in subtitle retrieval

---
 export.py | 81 ++++++++++++++++++++++++++++++-------------------------
 1 file changed, 44 insertions(+), 37 deletions(-)

diff --git a/export.py b/export.py
index 39944ce..309f010 100644
--- a/export.py
+++ b/export.py
@@ -33,6 +33,8 @@ from gc import collect
 
 import requests
 
+from time import sleep
+
 # https://docs.python.org/3/library/html.parser.html
 from html.parser import HTMLParser
 
@@ -84,43 +86,48 @@ def subprrun(jobs, mysession):
         vid = vid.strip()
         print(langcode, vid)
 
-        if mode == "default":
-            pparams = (
-                ("v", vid),
-                ("lang", langcode),
-                ("action_mde_edit_form", 1),
-                ("bl", "vmp"),
-                ("ui", "hd"),
-                ("tab", "captions"),
-                ("o", "U")
-            )
-
-            page = mysession.get("https://www.youtube.com/timedtext_editor", params=pparams)
-        elif mode == "forceedit-metadata":
-            pparams = (
-                ("v", vid),
-                ("lang", langcode),
-                ("action_mde_edit_form", 1),
-                ('forceedit', 'metadata'),
-                ('tab', 'metadata')
-            )
-
-            page = mysession.get("https://www.youtube.com/timedtext_editor", params=pparams)
-        elif mode == "forceedit-captions":
-            pparams = (
-                ("v", vid),
-                ("lang", langcode),
-                ("action_mde_edit_form", 1),
-                ("bl", "vmp"),
-                ("ui", "hd"),
-                ('forceedit', 'captions'),
-                ("tab", "captions"),
-                ("o", "U")
-            )
-
-            page = mysession.get("https://www.youtube.com/timedtext_editor", params=pparams)
-
-        assert not "accounts.google.com" in page.url, "Please supply authentication cookie information in config.json or environment variables. See README.md for more information."
+        while True:
+            if mode == "default":
+                pparams = (
+                    ("v", vid),
+                    ("lang", langcode),
+                    ("action_mde_edit_form", 1),
+                    ("bl", "vmp"),
+                    ("ui", "hd"),
+                    ("tab", "captions"),
+                    ("o", "U")
+                )
+
+                page = mysession.get("https://www.youtube.com/timedtext_editor", params=pparams)
+            elif mode == "forceedit-metadata":
+                pparams = (
+                    ("v", vid),
+                    ("lang", langcode),
+                    ("action_mde_edit_form", 1),
+                    ('forceedit', 'metadata'),
+                    ('tab', 'metadata')
+                )
+
+                page = mysession.get("https://www.youtube.com/timedtext_editor", params=pparams)
+            elif mode == "forceedit-captions":
+                pparams = (
+                    ("v", vid),
+                    ("lang", langcode),
+                    ("action_mde_edit_form", 1),
+                    ("bl", "vmp"),
+                    ("ui", "hd"),
+                    ('forceedit', 'captions'),
+                    ("tab", "captions"),
+                    ("o", "U")
+                )
+
+                page = mysession.get("https://www.youtube.com/timedtext_editor", params=pparams)
+
+            if not "accounts.google.com" in page.url:
+                break
+            else:
+                print("[Retrying in 30 seconds] Please supply authentication cookie information in config.json or environment variables. See README.md for more information.")
+                sleep(30)
 
         inttext = page.text