Fix modinput url split code to work with an IPv6 address

Shakeel Mohamed · Shakeel Mohamed · commit 4c2ae3b72676 · 2014-09-11T17:31:42.000-07:00
this uses the same code as custom search commands
diff --git a/splunklib/modularinput/script.py b/splunklib/modularinput/script.py
@@ -98,7 +98,8 @@ def run_script(self, args, event_writer, input_stream):
 
                     return 1
             else:
-                err_string = "ERROR Invalid arguments to modular input script:" + ' '.join(args)
+                err_string = "ERROR Invalid arguments to modular input script:" + ' '.join(
+                    args)
                 event_writer._err.write(err_string)
 
         except Exception as e:
@@ -129,13 +130,14 @@ def service(self):
         splunkd_uri = self._input_definition.metadata["server_uri"]
         session_key = self._input_definition.metadata["session_key"]
 
-        scheme, netloc, _, _, _ = urlsplit(splunkd_uri, allow_fragments=False)
-
-        splunkd_host, splunkd_port = netloc.split(':')
+        splunkd = urlsplit(splunkd_uri, allow_fragments=False)
 
         self._service = Service(
-            scheme=scheme, host=splunkd_host, port=splunkd_port,
-            token=session_key)
+            scheme=splunkd.scheme,
+            host=splunkd.hostname,
+            port=splunkd.port,
+            token=session_key,
+        )
 
         return self._service
 
