From 9c81a7bbda89a00a9ecbe1cf94db5b129afc11b7 Mon Sep 17 00:00:00 2001
From: Vangelis Banos <vangelis@archive.org>
Date: Mon, 31 Jul 2017 08:57:47 +0000
Subject: [PATCH] Make youtube-dl optional in BrozzlerWorker.brozzle_page

Enabled by default (of course).
Speed is significantly improved when disabled.
---
 brozzler/worker.py | 48 ++++++++++++++++++++++++----------------------
 1 file changed, 25 insertions(+), 23 deletions(-)

diff --git a/brozzler/worker.py b/brozzler/worker.py
index 3169e47..f201b9a 100644
--- a/brozzler/worker.py
+++ b/brozzler/worker.py
@@ -316,30 +316,32 @@ class BrozzlerWorker:
 
         return full_jpeg, thumb_jpeg
 
-    def brozzle_page(self, browser, site, page, on_screenshot=None):
+    def brozzle_page(self, browser, site, page, on_screenshot=None,
+                     enable_youtube_dl=True):
         self.logger.info("brozzling {}".format(page))
-        try:
-            with tempfile.TemporaryDirectory(prefix='brzl-ydl-') as tempdir:
-                ydl = self._youtube_dl(tempdir, site)
-                ydl_spy = ydl.brozzler_spy # remember for later
-                self._try_youtube_dl(ydl, site, page)
-        except brozzler.ReachedLimit as e:
-            raise
-        except brozzler.ShutdownRequested:
-            raise
-        except brozzler.ProxyError:
-            raise
-        except Exception as e:
-            if (hasattr(e, 'exc_info') and len(e.exc_info) >= 2
-                    and hasattr(e.exc_info[1], 'code')
-                    and e.exc_info[1].code == 430):
-                self.logger.info(
-                        'youtube-dl got %s %s processing %s',
-                        e.exc_info[1].code, e.exc_info[1].msg, page.url)
-            else:
-                self.logger.error(
-                        'youtube_dl raised exception on %s', page,
-                        exc_info=True)
+        if enable_youtube_dl:
+            try:
+                with tempfile.TemporaryDirectory(prefix='brzl-ydl-') as tempdir:
+                    ydl = self._youtube_dl(tempdir, site)
+                    ydl_spy = ydl.brozzler_spy # remember for later
+                    self._try_youtube_dl(ydl, site, page)
+            except brozzler.ReachedLimit as e:
+                raise
+            except brozzler.ShutdownRequested:
+                raise
+            except brozzler.ProxyError:
+                raise
+            except Exception as e:
+                if (hasattr(e, 'exc_info') and len(e.exc_info) >= 2
+                        and hasattr(e.exc_info[1], 'code')
+                        and e.exc_info[1].code == 430):
+                    self.logger.info(
+                            'youtube-dl got %s %s processing %s',
+                            e.exc_info[1].code, e.exc_info[1].msg, page.url)
+                else:
+                    self.logger.error(
+                            'youtube_dl raised exception on %s', page,
+                            exc_info=True)
 
         if self._needs_browsing(page, ydl_spy):
             self.logger.info('needs browsing: %s', page)