summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xbin/hostinfo5
1 files changed, 4 insertions, 1 deletions
diff --git a/bin/hostinfo b/bin/hostinfo
index 25c7509..11356c4 100755
--- a/bin/hostinfo
+++ b/bin/hostinfo
@@ -7,7 +7,10 @@ import yaml
import os.path
from dns import resolver,reversename
-import hostinfo.printer
+if os.path.exists(os.path.join(os.path.dirname(__file__), '..', 'hostinfo')):
+ sys.path = ['..'] + sys.path
+
+from hostinfo import printer
def print_info(path, key=None):
stream = file(path, 'r')