8

我正在尝试学习一些 python，但我在想要测试的逻辑上遇到了问题。目前我的代码是以绑定到 source_address 的方式编写的，当进程开始时不会改变

import socket
import requests

real_create_conn = socket.create_connection

def set_src_addr(*args):
    address, timeout = args[0], args[1]
    source_address = ('201.X.X.1', 0)
    return real_create_conn(address, timeout, source_address)

socket.create_connection = set_src_addr

r = requests.get('http://www.mywebpage.com/main')
print r.status_code 
if r.status_code == 404
   print "Webpage Down!"

r = requests.get('http://www.mywebpage.com/blog')
print r.status_code 
if r.status_code == 204
   print "Error occured!"

我正在寻找在哪里做这样的事情

import socket
import requests


While 1: 

      #bind to source address 201.X.X.1
      #Send request to main webpage
      #print result
      time.sleep(300) # 5 minutes

      #bind to source address 201.X.X.12
      #Send request to blog webpage
      #print result
      time.sleep(300) # 5 minutes
4

1 回答 1

18

对于每个请求，没有优雅的解决方案，但您需要使用 requests Session 对象并为每个请求安装一个新的传输适配器。

您可以在此问题评论中找到传输适配器的示例代码，或者您可以使用requests-toolbelt包中包含的适配器，如下所示

import requests
from requests_toolbelt.adapters import source

responses = []
s = requests.Session()
for source_ip, url in list_of_sources_and_urls:
    new_source = source.SourceAddressAdapter(source_ip)
    s.mount('http://', new_source)
    s.mount('https://', new_source)
    responses.append(s.get(url))

假设list_of_sources_and_urls看起来像

 [('127.0.0.1', 'https://google.com'),
  ('255.255.254.0', 'https://yahoo.com'),
  # ...
  ]
于 2015-02-28T15:53:19.837 回答