1 files changed, 43 insertions, 0 deletions
diff --git a/module/plugins/hooks/UserAgentSwitcher.py b/module/plugins/hooks/UserAgentSwitcher.py
new file mode 100644
index 000000000..0637936f9
--- /dev/null
+++ b/module/plugins/hooks/UserAgentSwitcher.py
@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+
+import pycurl
+
+from module.plugins.internal.Hook import Hook
+
+
+class UserAgentSwitcher(Hook):
+    __name__    = "UserAgentSwitcher"
+    __type__    = "hook"
+    __version__ = "0.09"
+
+    __config__ = [("activated"     , "bool", "Activated"                             , True                                                                      ),
+                  ("connecttimeout", "int" , "Connection timeout in seconds"         , 60                                                                        ),
+                  ("maxredirs"     , "int" , "Maximum number of redirects to follow" , 10                                                                        ),
+                  ("useragent"     , "str" , "Custom user-agent string"              , "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:38.0) Gecko/20100101 Firefox/38.0")]
+
+    __description__ = """Custom user-agent"""
+    __license__     = "GPLv3"
+    __authors__     = [("Walter Purcaro", "vuolter@gmail.com")]
+
+
+    interval = 0  #@TODO: Remove in 0.4.10
+
+
+    def setup(self):
+        self.info = {}  #@TODO: Remove in 0.4.10
+
+
+    def downloadPreparing(self, pyfile):
+        connecttimeout = self.getConfig('connecttimeout')
+        maxredirs      = self.getConfig('maxredirs')
+        useragent      = self.getConfig('useragent').encode("utf8", "replace")  #@TODO: Remove `encode` in 0.4.10
+
+        if connecttimeout:
+            pyfile.plugin.req.http.c.setopt(pycurl.CONNECTTIMEOUT, connecttimeout)
+
+        if maxredirs:
+            pyfile.plugin.req.http.c.setopt(pycurl.MAXREDIRS, maxredirs)
+
+        if useragent:
+            self.logDebug("Use custom user-agent string: " + useragent)
+            pyfile.plugin.req.http.c.setopt(pycurl.USERAGENT, useragent)