fix: 'common' version

This commit is contained in:
2026-07-10 18:31:45 +08:00
parent f773dce315
commit d62a772faa
3 changed files with 8 additions and 8 deletions
+7 -7
View File
@@ -33,7 +33,7 @@ WEBURL = "https://crm.xiaoman.cn/business/export"
APIURL = "https://crm.xiaoman.cn/api/opportunityRead/export"
def main(driver: WebDriver, logger = logging.getLogger('main')):
identity = short()
unique = identity()
parameters = vars(args)
http = PoolManager()
sp = ServiceProvider.default()
@@ -82,7 +82,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
while True:
try:
if not (buttons := driver.find_elements(By.CSS_SELECTOR, ".export-actions .okki-btn-primary")): raise ValueError()
for button in buttons: driver.execute_script("arguments[0].addEventListener('click', () => arguments[0].parentElement.setAttribute(arguments[1], '1'));", button, identity)
for button in buttons: driver.execute_script("arguments[0].addEventListener('click', () => arguments[0].parentElement.setAttribute(arguments[1], '1'));", button, unique)
break
except:
sleep(1)
@@ -90,9 +90,9 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
try:
verified = False
button = locate(".filter-actions .okki-btn-primary")
driver.execute_script("window.__%s__ = () => arguments[0].click();" % identity, buttons[1])
driver.execute_script("arguments[0].addEventListener('click', __%s__);" % identity, button)
driver.execute_script("__%s__();" % identity)
driver.execute_script("window.__%s__ = () => arguments[0].click();" % unique, buttons[1])
driver.execute_script("arguments[0].addEventListener('click', __%s__);" % unique, button)
driver.execute_script("__%s__();" % unique)
except Exception as e:
logger.critical("Error while setting up button listeners", exc_info=e)
return 4
@@ -100,8 +100,8 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
while True:
try:
for index in count(0):
if (driver.execute_script("return arguments[0].parentElement.getAttribute(arguments[1]);", buttons[index], identity)):
driver.execute_script("arguments[0].parentElement.removeAttribute(arguments[1]);", buttons[index], identity)
if (driver.execute_script("return arguments[0].parentElement.getAttribute(arguments[1]);", buttons[index], unique)):
driver.execute_script("arguments[0].parentElement.removeAttribute(arguments[1]);", buttons[index], unique)
raise ValueError()
except StaleElementReferenceException:
logger.critical('Error while listening to button events')