更简单的登录多个站点/页面的方法

Question

我对我编写代码的方式不是很满意。有没有一种更简单方便的方法来在一个函数中编写代码并返回多个页面的输出。

def login():
        url = "http://192.168.2.45/pricelogin.php"
        r = requests.get(url, auth=('pstats', 'pStats'))
        page = r.text
        return page

def loginhighpricingerror():
        pricingerrorurl = "http://192.168.2.45/airline_error.pl"
        peu = requests.get(pricingerrorurl, auth=('pstats', 'pstats'))
        peupage = peu.text
        return peupage

def loginsuccessfullbookings():
        sucurl = "http://192.168.2.45/airlinessucbookings.php"
        suc = requests.get(sucbookingurl, auth=('pstats', 'pstats'))
        sucpage = suc.text
        return sucpage

Answer 1

使用会话而不是无会话模块函数：

s = requests.Session()
s.auth=('pstats', 'pStats')

def login():
        url = "http://192.168.2.45/pricelogin.php"
        r = s.get(url)
        page = r.text
        return page

def loginhighpricingerror():
        pricingerrorurl = "http://192.168.2.45/airline_error.pl"
        peu = s.get(pricingerrorurl)
        peupage = peu.text
        return peupage

def loginsuccessfullbookings():
        sucurl = "http://192.168.2.45/airlinessucbookings.php"
        suc = s.get(sucbookingurl)
        sucpage = suc.text
        return sucpage

当然，这应该被重构，但希望你能明白我的意思。

Answer 2

我将泛化登录函数，将url作为参数传递：

def login(url):
   try:
      r = requests.get(url, auth=('pstats', 'pStats'))
   except requests.exceptions.RequestException as e:
      print e
      return '' # but maybe you want to do something else

   page = r.text
   return page

然后，您可以为每个url运行它，将页面累加到一个数组中，例如：

urls = ["http://192.168.2.45/pricelogin.php", "http://192.168.2.45/airline_error.pl", "http://192.168.2.45/airlinessucbookings.php"]

pages = [] # resulting array
for url in urls:
   pages.append(login(url))

注释：我为requests.get添加了一个异常检查，因为当存在连接问题时，这可能会失败。