bugfixes for webdriver arguments

Author: maybe-hello-world

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "netunicorn-library"
-version = "0.3.6"
+version = "0.3.7"
 authors = [
     {name = "Roman Beltiukov", email = "[email protected]"},
 ]

tasks/video_watchers/twitch_watcher.py

@@ -47,11 +47,26 @@ def watch(
 
 
 class WatchTwitchStream(TaskDispatcher):
-    def __init__(self, video_url: str, duration: Optional[int] = None, *args, **kwargs):
+    def __init__(
+            self,
+            video_url: str,
+            duration: Optional[int] = None,
+            chrome_location: Optional[str] = None,
+            webdriver_arguments: Optional[list] = None,
+            *args, **kwargs
+    ):
         super().__init__(*args, **kwargs)
         self.video_url = video_url
         self.duration = duration
-        self.linux_implementation = WatchTwitchStreamLinuxImplementation(self.video_url, self.duration, name=self.name)
+        self.chrome_location = chrome_location
+        self.webdriver_arguments = webdriver_arguments
+        self.linux_implementation = WatchTwitchStreamLinuxImplementation(
+            self.video_url,
+            self.duration,
+            self.chrome_location,
+            self.webdriver_arguments,
+            name=self.name
+        )
 
     def dispatch(self, node: Node) -> Task:
         if node.architecture in {Architecture.LINUX_AMD64, Architecture.LINUX_ARM64}:
@@ -69,17 +84,23 @@ class WatchTwitchStreamLinuxImplementation(Task):
     ]
 
     def __init__(
-        self, video_url: str, duration: int = 10, chrome_location: Optional[str] = None, *args, **kwargs
+            self,
+            video_url: str,
+            duration: int = 10,
+            chrome_location: Optional[str] = None,
+            webdriver_arguments: Optional[list] = None,
+            *args, **kwargs
     ):
         self.video_url = video_url
         self.duration = duration
         self.chrome_location = chrome_location
         if not self.chrome_location:
             self.chrome_location = "/usr/bin/chromium"
+        self.webdriver_arguments = webdriver_arguments
         super().__init__(*args, **kwargs)
 
     def run(self):
-        return watch(self.video_url, self.duration, self.chrome_location)
+        return watch(self.video_url, self.duration, self.chrome_location, self.webdriver_arguments)
 
 
 if __name__ == "__main__":

tasks/video_watchers/vimeo_watcher.py

@@ -77,11 +77,26 @@ def watch(
 
 
 class WatchVimeoVideo(TaskDispatcher):
-    def __init__(self, video_url: str, duration: Optional[int] = None, *args, **kwargs):
+    def __init__(
+            self,
+            video_url: str,
+            duration: Optional[int] = None,
+            chrome_location: Optional[str] = None,
+            webdriver_arguments: Optional[list] = None,
+            *args, **kwargs
+    ):
         super().__init__(*args, **kwargs)
         self.video_url = video_url
         self.duration = duration
-        self.linux_implementation = WatchVimeoVideoLinuxImplementation(self.video_url, self.duration, name=self.name)
+        self.chrome_location = chrome_location
+        self.webdriver_arguments = webdriver_arguments
+        self.linux_implementation = WatchVimeoVideoLinuxImplementation(
+            self.video_url,
+            self.duration,
+            self.chrome_location,
+            self.webdriver_arguments,
+            name=self.name
+        )
 
     def dispatch(self, node: Node) -> Task:
         if node.architecture in {Architecture.LINUX_AMD64, Architecture.LINUX_ARM64}:
@@ -103,6 +118,7 @@ def __init__(
         video_url: str,
         duration: Optional[int] = None,
         chrome_location: Optional[str] = None,
+        webdriver_arguments: Optional[list] = None,
         *args,
         **kwargs
     ):
@@ -111,10 +127,11 @@ def __init__(
         self.chrome_location = chrome_location
         if not self.chrome_location:
             self.chrome_location = "/usr/bin/chromium"
+        self.webdriver_arguments = webdriver_arguments
         super().__init__(*args, **kwargs)
 
     def run(self):
-        return watch(self.video_url, self.duration, self.chrome_location)
+        return watch(self.video_url, self.duration, self.chrome_location, self.webdriver_arguments)
 
 
 if __name__ == "__main__":

tasks/video_watchers/youtube_watcher.py

@@ -100,11 +100,26 @@ def watch(
 
 
 class WatchYouTubeVideo(TaskDispatcher):
-    def __init__(self, video_url: str, duration: Optional[int] = None, *args, **kwargs):
+    def __init__(
+            self,
+            video_url: str,
+            duration: Optional[int] = None,
+            chrome_location: Optional[str] = None,
+            webdriver_arguments: Optional[list] = None,
+            *args, **kwargs
+    ):
         super().__init__(*args, **kwargs)
         self.video_url = video_url
         self.duration = duration
-        self.linux_implementation = WatchYouTubeVideoLinuxImplementation(self.video_url, self.duration, name=self.name)
+        self.chrome_location = chrome_location
+        self.webdriver_arguments = webdriver_arguments
+        self.linux_implementation = WatchYouTubeVideoLinuxImplementation(
+            self.video_url,
+            self.duration,
+            self.chrome_location,
+            self.webdriver_arguments,
+            name=self.name
+        )
 
     def dispatch(self, node: Node) -> Task:
         if node.architecture in {Architecture.LINUX_AMD64, Architecture.LINUX_ARM64}:
@@ -126,6 +141,7 @@ def __init__(
         video_url: str,
         duration: Optional[int] = None,
         chrome_location: Optional[str] = None,
+        webdriver_arguments: Optional[list] = None,
         *args,
         **kwargs,
     ):
@@ -134,10 +150,11 @@ def __init__(
         self.chrome_location = chrome_location
         if not self.chrome_location:
             self.chrome_location = "/usr/bin/chromium"
+        self.webdriver_arguments = webdriver_arguments
         super().__init__(*args, **kwargs)
 
     def run(self):
-        return watch(self.video_url, self.duration, self.chrome_location)
+        return watch(self.video_url, self.duration, self.chrome_location, self.webdriver_arguments)
 
 
 if __name__ == "__main__":