创建类或功能以供使用代理

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