From 534778c5625a9f023563a906e56cddaca6c9a531 Mon Sep 17 00:00:00 2001
From: break27 <buzybreakerbear@gmail.com>
Date: Thu, 25 Jun 2026 12:02:45 +0800
Subject: [PATCH] fix: added 'short' function

---
 src/common/utils/__init__.py | 2 ++
 src/common/utils/selenium.py | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/common/utils/__init__.py b/src/common/utils/__init__.py
index 0dfe878..c2fa5c8 100644
--- a/src/common/utils/__init__.py
+++ b/src/common/utils/__init__.py
@@ -3,6 +3,7 @@ from common.utils.selenium import (
     sleep,
     locate,
     click,
+    short,
     setup,
 )
 
@@ -11,5 +12,6 @@ __all__ = [
     'sleep',
     'locate',
     'click',
+    'short',
     'setup',
 ]
diff --git a/src/common/utils/selenium.py b/src/common/utils/selenium.py
index b67fb1b..a831ca3 100644
--- a/src/common/utils/selenium.py
+++ b/src/common/utils/selenium.py
@@ -73,8 +73,11 @@ def click(selector: str|WebElement, wait=True, condition=False):
     try: driver.execute_script("arguments[0].removeEventListener('click', __%s__);" % identity, element)
     except: pass
 
+def short():
+    return ''.join(random.choices(string.digits + string.ascii_uppercase + string.ascii_lowercase, k=8))
+
 def setup(a: WebDriver, b: dict):
     global identity, driver
-    identity = ''.join(random.choices(string.ascii_uppercase + string.digits, k=8))
+    identity = short()
     driver = a
     parameters.update(b)