Install missing optional dependencies as needed

Author: mdmintz

seleniumbase/core/s3_manager.py

@@ -2,6 +2,7 @@
 Methods for uploading/managing files on Amazon S3.
 """
 from seleniumbase.config import settings
+from seleniumbase.fixtures import shared_utils
 
 already_uploaded_files = []
 
@@ -19,15 +20,23 @@ def __init__(
         selenium_access_key=settings.S3_SELENIUM_ACCESS_KEY,
         selenium_secret_key=settings.S3_SELENIUM_SECRET_KEY,
     ):
-        from boto.s3.connection import S3Connection
+        try:
+            from boto.s3.connection import S3Connection
+        except Exception:
+            shared_utils.pip_install("boto", version="2.49.0")
+            from boto.s3.connection import S3Connection
 
         self.conn = S3Connection(selenium_access_key, selenium_secret_key)
         self.bucket = self.conn.get_bucket(log_bucket)
         self.bucket_url = bucket_url
 
     def get_key(self, file_name):
         """Create a new Key instance with the given name."""
-        from boto.s3.key import Key
+        try:
+            from boto.s3.key import Key
+        except Exception:
+            shared_utils.pip_install("boto", version="2.49.0")
+            from boto.s3.key import Key
 
         return Key(bucket=self.bucket, name=file_name)
 

seleniumbase/fixtures/base_case.py

@@ -5498,7 +5498,12 @@ def get_pdf_text(
 
         with warnings.catch_warnings():
             warnings.simplefilter("ignore", category=UserWarning)
-            from pdfminer.high_level import extract_text
+            try:
+                from pdfminer.high_level import extract_text
+            except Exception:
+                shared_utils.pip_install("pdfminer.six")
+                from pdfminer.high_level import extract_text
+
         if not password:
             password = ""
         if not maxpages:
@@ -5723,7 +5728,11 @@ def save_element_as_image_file(
             file.write(element_png)
         # Add a text overlay if given
         if type(overlay_text) is str and len(overlay_text) > 0:
-            from PIL import Image, ImageDraw
+            try:
+                from PIL import Image, ImageDraw
+            except Exception:
+                shared_utils.pip_install("Pillow")
+                from PIL import Image, ImageDraw
 
             text_rows = overlay_text.split("\n")
             len_text_rows = len(text_rows)