From cbe4d60d401e4f2cedc38efb375ae38b40fb0c89 Mon Sep 17 00:00:00 2001
From: RaNaN <Mast3rRaNaN@hotmail.de>
Date: Sun, 16 Aug 2009 17:29:36 +0200
Subject: reconnect-, uploaded.to fix

---
 module/network/Request.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

(limited to 'module/network/Request.py')

diff --git a/module/network/Request.py b/module/network/Request.py
index ccf12e756..0427bbe25 100755
--- a/module/network/Request.py
+++ b/module/network/Request.py
@@ -163,9 +163,12 @@ class Request:
 
     def add_auth(self, user, pw):
 
-        # @TODO: pycurl auth
-
-        self.downloader.addheaders.append(['Authorization', 'Basic ' + base64.encodestring(user + ':' + pw)[:-1]])
+        if self.curl:
+            self.pycurl.setopt(pycurl.USERNAME, user)
+            self.pycurl.setopt(pycurl.PASSWORD, pw)
+            self.pycurl.setopt(pycurl.HTTPAUTH, pycurl.HTTPAUTH_ANY)
+        else:
+            self.downloader.addheaders.append(['Authorization', 'Basic ' + base64.encodestring(user + ':' + pw)[:-1]])
 
     def add_cookies(self, req):
         cookie_head = ""
@@ -182,9 +185,15 @@ class Request:
             del self.cookies[:]
 
     def add_proxy(self, protocol, adress):
-        handler = urllib2.ProxyHandler({protocol: adress})
-        self.opener.add_handler(handler)
-        self.downloader.add_handler(handler)
+
+        # @TODO: pycurl proxy protocoll selection
+        if self.curl:
+            self.pycurl.setopt(pycurl.PROXY, adress.split(":")[0])
+            self.pycurl.setopt(pycurl.PROXYPORT, adress.split(":")[1])
+        else:
+            handler = urllib2.ProxyHandler({protocol: adress})
+            self.opener.add_handler(handler)
+            self.downloader.add_handler(handler)
 
     def download(self, url, filename, get={}, post={}, ref=True, cookies=False):
 
-- 
cgit v1.2.3