DiTo97 · January 10, 2024 05:34 · Jan 10, 2024 · Jan 10, 2024 · Jan 10, 2024 · Jan 10, 2024
diff --git a/python-to-js-webserver.py b/python-to-js-webserver.py
@@ -54,9 +54,6 @@ def close(self) -> None:
     def execute(self, script: str) -> Any:
         return self.webdriver.execute_script(script)
 
-    async def async_execute(self, script: str) -> Any:
-        return self.webdriver.execute_async_script(script)
-
 
 if __name__ == "__main__":
     myserver = MyServer("localhost", 8000, None)

diff --git a/python-to-js-webserver.py b/python-to-js-webserver.py
@@ -5,7 +5,6 @@
 - https://stackoverflow.com/a/39607115
 """
 import http.server
-import pathlib
 import socketserver
 import threading
 from typing import Any
@@ -14,9 +13,6 @@
 from selenium.webdriver.chrome import options as chromeoptions
 
 
-ROOT = pathlib.Path(__file__).parent
-
-
 def from_webroot_dir(directory: str | None):
     def closure(*args: Any, **kwargs: Any) -> http.server.SimpleHTTPRequestHandler:
         return http.server.SimpleHTTPRequestHandler(

diff --git a/python-to-js-webserver.py b/python-to-js-webserver.py
@@ -11,13 +11,13 @@
 from typing import Any
 
 from selenium import webdriver
-from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome import options as chromeoptions
 
 
 ROOT = pathlib.Path(__file__).parent
 
 
-def from_webroot_dir(directory: str):
+def from_webroot_dir(directory: str | None):
     def closure(*args: Any, **kwargs: Any) -> http.server.SimpleHTTPRequestHandler:
         return http.server.SimpleHTTPRequestHandler(
             *args, directory=directory, **kwargs
@@ -27,33 +27,32 @@ def closure(*args: Any, **kwargs: Any) -> http.server.SimpleHTTPRequestHandler:
 
 
 class MyServer:
-    def __init__(self, host: str = "localhost", port: int = 8000):
-        self.httpd = socketserver.TCPServer((host, port), from_webroot("webroot"))
-        self.server_thread = threading.Thread(target=self.httpd.serve_forever)
+    def __init__(self, host: str = "localhost", port: int = 8000, webroot: str | None = None):
+        self.server = socketserver.TCPServer((host, port), from_webroot_dir(webroot))
+        self.executor = threading.Thread(target=self.server.serve_forever)
 
-    def start(self):
-        # Start the server in a new thread
-        self.server_thread.start()
+    def begin(self) -> None:
+        self.executor.start()
+
+    def close(self) -> None:
+        self.server.shutdown()
+        self.executor.join()
 
-    def stop(self):
-        # Stop the server and wait for the server thread to finish
-        self.httpd.shutdown()
-        self.server_thread.join()
 
 class MyBrowser:
     def __init__(self, host: str, port: int) -> None:
         self.URL = f"http://{host}:{port}"
 
-        options = Options()
+        options = chromeoptions.Options()
         options.add_argument("--headless")
 
         self.webdriver = webdriver.Chrome(options=options)
 
-    def get(self, path: str):
-        endpoint = f"{self.URL}/{path}"
+    def get(self, resource: str) -> None:
+        endpoint = f"{self.URL}/{resource}"
         self.webdriver.get(endpoint)
 
-    def quit(self):
+    def close(self) -> None:
         self.webdriver.quit()
 
     def execute(self, script: str) -> Any:
@@ -62,24 +61,22 @@ def execute(self, script: str) -> Any:
     async def async_execute(self, script: str) -> Any:
         return self.webdriver.execute_async_script(script)
 
+
 if __name__ == "__main__":
-    # Usage
-    server = MyServer("localhost", 8000)
-    server.start()
+    myserver = MyServer("localhost", 8000, None)
+    myserver.begin()
 
-    browser = MyBrowser("localhost", 8000)
+    mybrowser = MyBrowser("localhost", 8000)
 
     try:
-        browser.get("index.html")  # Navigate to the local HTML file
-
-        X = browser.execute("return X;")
-        helloworld = browser.execute("return printHelloWorld();")
+        mybrowser.get("index.html")
 
-        print(X, helloworld)  # Output: The value of the global JavaScript variable
+        X = mybrowser.execute("return X;")
+        helloworld = mybrowser.execute("return helloworld();")
 
-        # Do something with the browser...
+        print(X, helloworld)
     except Exception as x:
         print(x)
     finally:
-        browser.quit()
-        server.stop()
+        mybrowser.close()
+        myserver.close()
diff --git a/python-to-js-webserver.py b/python-to-js-webserver.py
@@ -1,3 +1,9 @@
+"""A simple python-to-JS HTTP web server from a specific webroot
+
+For more information on python-to-JS transfer possibilities, see the following resources:
+- https://stackoverflow.com/questions/22554313/send-captured-images-from-python-server-to-javascript-client
+- https://stackoverflow.com/a/39607115
+"""
 import http.server
 import pathlib
 import socketserver

diff --git a/index.html b/index.html
@@ -0,0 +1,16 @@
+<!DOCTYPE html>
+<html>
+
+<head>
+    <script type="text/javascript">
+        const X = "I am X";
+
+        function helloworld() {
+            return "Hello world";
+        }
+    </script>
+</head>
+
+<body></body>
+
+</html>
diff --git a/python-to-js-webserver.py b/python-to-js-webserver.py
@@ -0,0 +1,79 @@
+import http.server
+import pathlib
+import socketserver
+import threading
+from typing import Any
+
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+
+
+ROOT = pathlib.Path(__file__).parent
+
+
+def from_webroot_dir(directory: str):
+    def closure(*args: Any, **kwargs: Any) -> http.server.SimpleHTTPRequestHandler:
+        return http.server.SimpleHTTPRequestHandler(
+            *args, directory=directory, **kwargs
+        )
+
+    return closure
+
+
+class MyServer:
+    def __init__(self, host: str = "localhost", port: int = 8000):
+        self.httpd = socketserver.TCPServer((host, port), from_webroot("webroot"))
+        self.server_thread = threading.Thread(target=self.httpd.serve_forever)
+
+    def start(self):
+        # Start the server in a new thread
+        self.server_thread.start()
+
+    def stop(self):
+        # Stop the server and wait for the server thread to finish
+        self.httpd.shutdown()
+        self.server_thread.join()
+
+class MyBrowser:
+    def __init__(self, host: str, port: int) -> None:
+        self.URL = f"http://{host}:{port}"
+
+        options = Options()
+        options.add_argument("--headless")
+
+        self.webdriver = webdriver.Chrome(options=options)
+
+    def get(self, path: str):
+        endpoint = f"{self.URL}/{path}"
+        self.webdriver.get(endpoint)
+
+    def quit(self):
+        self.webdriver.quit()
+
+    def execute(self, script: str) -> Any:
+        return self.webdriver.execute_script(script)
+
+    async def async_execute(self, script: str) -> Any:
+        return self.webdriver.execute_async_script(script)
+
+if __name__ == "__main__":
+    # Usage
+    server = MyServer("localhost", 8000)
+    server.start()
+
+    browser = MyBrowser("localhost", 8000)
+
+    try:
+        browser.get("index.html")  # Navigate to the local HTML file
+
+        X = browser.execute("return X;")
+        helloworld = browser.execute("return printHelloWorld();")
+
+        print(X, helloworld)  # Output: The value of the global JavaScript variable
+
+        # Do something with the browser...
+    except Exception as x:
+        print(x)
+    finally:
+        browser.quit()
+        server.stop()
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1 @@
+selenium~=4.16.0