为使用代理创建类或函数

Question

我想对特定请求使用代理设置而不是我的所有脚本。 我其实：

import whois
from pymongo import MongoClient
import socks
import socket



socks.setdefaultproxy(socks.PROXY_TYPE_HTTP, "IP", 13012)
socket.socket = socks.socksocket


client = MongoClient()
db = client.pro
collection = db.domain
find_document = collection.find({"Field":"ERROR"},{'domain': 1, '_id':0})

domaine = d['domain']
print(domaine)
w = whois.whois(domaine) # I want to use proxy for this request
date = w.expiration_date
print date
collection.update({"domain": domaine}, {"$set": {"expire": date}})

但在那之后，无法连接到localhost上的mongodb。 我的问题是我如何才能将代理仅用于特定请求？ 提前感谢您的回答。

Answer 1

我自己找到了答案，并为您分享：

import socks

socks.setdefaultproxy(socks.PROXY_TYPE_HTTP, 'YOUR_IP', 8080)
socks.wrapmodule(whois) # Here is Whois Module force to use proxy