update and cleanup more demos

1 files changed, 10 insertions, 2 deletions

diff --git a/demo/readdir2.py b/demo/readdir2.py
index 5927a77..64fc60f 100644
--- a/demo/readdir2.py
+++ b/demo/readdir2.py
@@ -1,11 +1,19 @@
-# A Linux-only demo, using verify() instead of hard-coding the exact layouts
+# A Linux-only demo, using set_source() instead of hard-coding the exact layouts
 #
 import sys
-from _readdir2 import ffi, lib
 
 if not sys.platform.startswith('linux'):
     raise Exception("Linux-only demo")
 
+# If the build script was run immediately before this script, the cffi module
+# ends up in the current directory. Make sure we can import it.
+sys.path.append('.')
+
+try:
+    from _readdir2 import ffi, lib
+except ImportError:
+    print 'run readdir2_build first, then make sure the shared object is on sys.path'
+    sys.exit(-1)
 
 def walk(basefd, path):
     print '{', path