search for keyfile in sys.path as last resort; reword exception
diff --git a/Demo/pdist/security.py b/Demo/pdist/security.py
index a11294a..f993682 100755
--- a/Demo/pdist/security.py
+++ b/Demo/pdist/security.py
@@ -9,10 +9,17 @@
keyfile = '.python_keyfile'
if env.has_key('HOME'):
keyfile = os.path.join(env['HOME'], keyfile)
+ if not os.path.exists(keyfile):
+ import sys
+ for dir in sys.path:
+ kf = os.path.join(dir, keyfile)
+ if os.path.exists(kf):
+ keyfile = kf
+ break
try:
self._key = eval(open(keyfile).readline())
except IOError:
- raise IOError, "python keyfile %s not found" % keyfile
+ raise IOError, "python keyfile %s: cannot open" % keyfile
def _generate_challenge(self):
import whrandom