Check session before extracting images (#130)

Jaded-Encoding-Thaumaturgy · Mar 12, 2024 · 3d99895 · 3d99895
1 parent c5ad99c
commit 3d99895
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/vspreview/toolbars/comp/toolbar.py b/vspreview/toolbars/comp/toolbar.py
@@ -165,6 +165,15 @@ def run(self, conf: WorkerConfiguration) -> None:
         all_images = list[list[Path]]()
         conf.path.mkdir(parents=True, exist_ok=False)
 
+        if conf.browser_id and conf.session_id:
+            with Session() as sess:
+                sess.cookies.set('SLP-SESSION', conf.session_id, domain='slow.pics')
+                browser_id = conf.browser_id
+                base_page = sess.get('https://slow.pics/comparison')
+                if base_page.text.find('id="logoutBtn"') == -1:
+                    self.progress_status.emit(conf.uuid, 'Session Expired', 0, 0)
+                    return
+
         try:
             for i, output in enumerate(conf.outputs):
                 if self.isFinished():