如何减少这段代码

Question

我对 Python 完全陌生，刚刚了解了列表理解，这令人印象深刻。我想知道是否有类似的方法可以减少以下if 语句。

if len(browser.find_elements_by_xpath('//span[@gogo-test="backers"]')) > 0:
    backers = [sub for sub in browser.find_element_by_xpath('//span[@gogo-test="backers"]').text.split() if sub.isdigit()][0]

理想情况下，我想使用backers = [sub for sub in browser.find_element_by_xpath('//span[@gogo-test="backers"]').text.split() if sub.isdigit()][0]，但是，与 Scrapy 不同的是，Selenium 在找不到任何东西时不会返回None，而且肯定会有browser.find_element_by_xpath('//span[@gogo-test="backers"]') 不存在的情况.

谢谢

Answer 1

您可以使用 Page-Object-Model 来减少代码块，这是一种常见的 Selenium 约定：

if  page.get_spans_array_size() > 0:
    backers = [sub for sub in  page.get_spans_array() if sub.isdigit()][0]

页面代码：

class Page:

    def __init__(self, driver):
        self.driver = driver

    def get_spans_array_size(self):
        return len(
                   ## self.driver.find..  )

    def get_spans_array(self):
        return ## self.driver.find..  

我的开源项目中有一个 POM 的 python 示例： https://github.com/AutomatedOwl/selenium-python-pom-example

Answer 2

backers = [sub for sub in browser.find_element_by_xpath('//span[@gogo-test="backers"]').text.split() if sub.isdigit()]
backers = backers[0] if len(backers) > 0 else None

它已经很短了，几乎没有改进的余地:) 或者

backers = list(filter(lambda sub: sub.isdigit(),
                      browser.find_element_by_xpath('//span[@gogo-test="backers"]').text.split()
               ))

backers = backers[0] if len(backers) > 0 else None

或者用正则表达式：

import re
backers = browser.find_elements_by_xpath('//span[@gogo-test="backers"]')
backers = backers[0].text if len(backers) > 0 else ''
backers = re.findall('(\d+)', backers)