1 files changed, 48 insertions, 30 deletions
diff --git a/module/plugins/hoster/BasePlugin.py b/module/plugins/hoster/BasePlugin.py
index ed11c378b..15e35ce24 100644
--- a/module/plugins/hoster/BasePlugin.py
+++ b/module/plugins/hoster/BasePlugin.py
@@ -1,13 +1,13 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-
-from module.plugins.Hoster import Hoster
-from module.utils import html_unescape
-
 from urlparse import urlparse
 from re import search
 from urllib import unquote
 
+from module.network.HTTPRequest import BadHeader
+from module.plugins.Hoster import Hoster
+from module.utils import html_unescape, removeChars
+
 class BasePlugin(Hoster):
     __name__ = "BasePlugin"
     __type__ = "hoster"
@@ -41,31 +41,49 @@ class BasePlugin(Hoster):
 #
 #        return
         if pyfile.url.startswith("http"):
-            header = self.load(pyfile.url, just_header = True)
-            #self.logDebug(header)
-
-            if 'location' in header:
-                self.logDebug("Location: " + header['location'])
-                url = unquote(header['location'])
-            else:
-                url = pyfile.url
-
-            name = html_unescape(urlparse(url).path.split("/")[-1])
-
-            if 'content-disposition' in header:
-                self.logDebug("Content-Disposition: " + header['content-disposition'])
-                m = search("filename(?P<type>=|\*=(?P<enc>.+)'')(?P<name>.*)", header['content-disposition'])
-                if m:
-                    disp = m.groupdict()
-                    self.logDebug(disp)
-                    if not disp['enc']: disp['enc'] = 'utf-8'
-                    name = disp['name'].replace('"', "").replace("'", "").replace(";", "").strip()
-                    name = unicode(unquote(name), disp['enc'])
-
-            if not name: self.offline()
-            pyfile.name = name
-            self.logDebug("Filename: %s" % pyfile.name)
-            self.download(url, disposition=True)
+
+            try:
+                self.downloadFile(pyfile)
+            except BadHeader, e:
+                if e.code in (401, 403):
+                    self.logDebug("Auth required")
+
+                    pwd = pyfile.package().password.strip()
+                    if ":" not in pwd:
+                        self.fail(_("Authorization required (username:password)"))
+
+                    self.req.addAuth(pwd)
+                    self.downloadFile(pyfile)
+                else:
+                    raise
 
         else:
-            self.fail("No Plugin matched and not a downloadable url.")
-\ No newline at end of file
+            self.fail("No Plugin matched and not a downloadable url.")
+
+
+    def downloadFile(self, pyfile):
+        header = self.load(pyfile.url, just_header = True)
+        #self.logDebug(header)
+
+        if 'location' in header:
+            self.logDebug("Location: " + header['location'])
+            url = unquote(header['location'])
+        else:
+            url = pyfile.url
+
+        name = html_unescape(urlparse(url).path.split("/")[-1])
+
+        if 'content-disposition' in header:
+            self.logDebug("Content-Disposition: " + header['content-disposition'])
+            m = search("filename(?P<type>=|\*=(?P<enc>.+)'')(?P<name>.*)", header['content-disposition'])
+            if m:
+                disp = m.groupdict()
+                self.logDebug(disp)
+                if not disp['enc']: disp['enc'] = 'utf-8'
+                name = removeChars(disp['name'], "\"';").strip()
+                name = unicode(unquote(name), disp['enc'])
+
+        if not name: name = url
+        pyfile.name = name
+        self.logDebug("Filename: %s" % pyfile.name)
+        self.download(url, disposition=True)
+\ No newline at end of file