From 1372bfdcbd599071565e807f0109ea0710bc4f60 Mon Sep 17 00:00:00 2001
From: Michael Mintz <mdmintz@gmail.com>
Date: Thu, 6 Mar 2025 20:46:15 -0500
Subject: [PATCH] Update command-line options

---
 seleniumbase/plugins/driver_manager.py | 17 +++++++++++++++++
 seleniumbase/plugins/sb_manager.py     | 17 +++++++++++++++++
 2 files changed, 34 insertions(+)

diff --git a/seleniumbase/plugins/driver_manager.py b/seleniumbase/plugins/driver_manager.py
index 72609637..663f06d0 100644
--- a/seleniumbase/plugins/driver_manager.py
+++ b/seleniumbase/plugins/driver_manager.py
@@ -781,6 +781,23 @@ def Driver(
             swiftshader = False
     if locale is not None and locale_code is None:
         locale_code = locale
+    if locale_code is None:
+        if '--locale="' in arg_join:
+            locale_code = (
+                arg_join.split('--locale="')[1].split('"')[0]
+            )
+        elif '--locale=' in arg_join:
+            locale_code = (
+                arg_join.split('--locale=')[1].split(' ')[0]
+            )
+        elif '--locale-code="' in arg_join:
+            locale_code = (
+                arg_join.split('--locale-code="')[1].split('"')[0]
+            )
+        elif '--locale-code=' in arg_join:
+            locale_code = (
+                arg_join.split('--locale-code=')[1].split(' ')[0]
+            )
     if ad_block is not None and ad_block_on is None:
         ad_block_on = ad_block
     if ad_block_on is None:
diff --git a/seleniumbase/plugins/sb_manager.py b/seleniumbase/plugins/sb_manager.py
index b3081c0a..0ed56d60 100644
--- a/seleniumbase/plugins/sb_manager.py
+++ b/seleniumbase/plugins/sb_manager.py
@@ -942,6 +942,23 @@ def SB(
             swiftshader = False
     if locale is not None and locale_code is None:
         locale_code = locale
+    if locale_code is None:
+        if '--locale="' in arg_join:
+            locale_code = (
+                arg_join.split('--locale="')[1].split('"')[0]
+            )
+        elif '--locale=' in arg_join:
+            locale_code = (
+                arg_join.split('--locale=')[1].split(' ')[0]
+            )
+        elif '--locale-code="' in arg_join:
+            locale_code = (
+                arg_join.split('--locale-code="')[1].split('"')[0]
+            )
+        elif '--locale-code=' in arg_join:
+            locale_code = (
+                arg_join.split('--locale-code=')[1].split(' ')[0]
+            )
     if ad_block is not None and ad_block_on is None:
         ad_block_on = ad_block
     if ad_block_on is None: