Permalink
Browse files

MB-7086: Add bin directory to python path

Since cbworkloadgen depends on cbtransfer, which in turn depends on
couchstore and related dlls, it is needed to add bin directory to python
path in case cbworkloadgen is moved to a different directory.

Change-Id: Id7fac8c07542ba5a0c2b2d5af9130b0a68f9c7a2
Reviewed-on: http://review.couchbase.org/22272
Tested-by: Bin Cui <bin.cui@gmail.com>
Reviewed-by: Steve Yen <steve.yen@gmail.com>
Reviewed-by: Pavel Paulau <pavel.paulau@gmail.com>
Tested-by: Steve Yen <steve.yen@gmail.com>
  • Loading branch information...
1 parent 7fe4121 commit dea2bae7b4a3a062d22e3d5db82e48961dce0804 @bcui6611 bcui6611 committed with steveyen Nov 5, 2012
Showing with 6 additions and 0 deletions.
  1. +6 −0 cbworkloadgen
View
@@ -10,6 +10,8 @@ import pump
import pump_gen
import sys
import optparse
+import platform
+import os
class WorkloadGen(pump_transfer.Transfer):
"""Entry point for 2.0 cbworkloadgen."""
@@ -145,4 +147,8 @@ class WorkloadGen(pump_transfer.Transfer):
return pump_gen.GenSource, pump.PumpingStation.find_handler(opts, sink, pump_transfer.SINKS)
if __name__ == '__main__':
+ if platform.system() == "Windows":
+ python_lib = os.path.join(os.path.dirname(sys.argv[0]), '..')
+ sys.path.append(python_lib)
+
pump_transfer.exit_handler(WorkloadGen().main(sys.argv))

0 comments on commit dea2bae

Please sign in to comment.