bin/suricata-update: handle non-standard install prefixes.

Look to see if the modules are in a known location relative to
the location of suricata-update and to the sys.path if found.

Allows suricata-update to be installed with non-standard
--prefixes.

diff --git a/bin/suricata-update b/bin/suricata-update
index e8230cfb0fcc54b2388581cf64f401ef559ead7d..f2e85577380d43704c1124e409af5f5d1031e96a 100755 (executable)
--- a/bin/suricata-update
+++ b/bin/suricata-update
@@ -19,6 +19,13 @@
 import sys
 import os
 
+exec_dir = os.path.dirname(__file__)
+version_info = sys.version_info
+pyver = "%d.%d" % (version_info.major, version_info.minor)
+site_path = "%s/../lib/python%s/site-packages" % (exec_dir, pyver)
+
+if os.path.exists("%s/suricata/update" % (site_path)):
+    sys.path.insert(0, site_path)
 sys.path.insert(
     0, os.path.dirname(os.path.dirname(os.path.abspath(sys.argv[0]))))