Fix path issue on ppc

Author: Kevin D Smith

swat/clib.py

@@ -25,6 +25,7 @@
 
 import glob
 import os
+import platform
 import struct
 import sys
 from .utils.compat import PY3, WIDE_CHARS, a2u
@@ -41,11 +42,11 @@ def _import_pyswat():
 
     import importlib
 
-    platform = 'linux'
+    plat = 'linux'
     if sys.platform.lower().startswith('win'):
-        platform = 'win'
+        plat = 'win'
     elif sys.platform.lower().startswith('darwin'):
-        platform = 'mac'
+        plat = 'mac'
 
     if PY3:
         libname = '_py%s%sswat' % (sys.version_info[0], sys.version_info[1])
@@ -55,14 +56,14 @@ def _import_pyswat():
         libname = '_pyswat'
 
     # Bail out if we aren't on Linux
-#   if platform != 'linux':
+#   if plat != 'linux':
 #       raise ValueError('Currently, Linux is the only platform with support '
 #                        'for the binary protocol.  You must connect to CAS '
 #                        'using the REST interface on this platform.')
 
     # Bail out if the C extension doesn't exist
     if not glob.glob(os.path.join(os.path.dirname(__file__), 'lib',
-                                  platform, libname + '.*')):
+                                  plat, libname + '.*')):
         raise ValueError('The extensions for the binary protocol have not been '
                          'installed.  You can either install them using the full '
                          'platform-dependent install file, or use the REST interface '
@@ -85,7 +86,7 @@ def _import_pyswat():
 
     # Try to import the C extension
     try:
-        _pyswat = importlib.import_module('.lib.%s.%s' % (platform, libname),
+        _pyswat = importlib.import_module('.lib.%s.%s' % (plat, libname),
                                           package='swat')
 
     except ImportError:
@@ -141,11 +142,24 @@ def InitializeTK(*args, **kwargs):
     ''' Initialize the TK subsystem (importing _pyswat as needed) '''
     if _pyswat is None:
         _import_pyswat()
-    out = _pyswat.InitializeTK(*args, **kwargs)
-    # Override TKPATH after initialization so that other TK applications
-    # won't be affected.
-    if sys.platform.lower().startswith('win') and 'TKPATH' not in os.environ:
-        os.environ['TKPATH'] = os.pathsep
+
+    # Patch ppc linux path
+    set_tkpath_env = 'ppc' in platform.machine() and 'TKPATH' not in os.environ
+    if set_tkpath_env and args:
+        os.environ['TKPATH'] = args[0]
+
+    try:
+        out = _pyswat.InitializeTK(*args, **kwargs)
+
+    finally:
+        if set_tkpath_env:
+            del os.environ['TKPATH']
+
+        # Override TKPATH after initialization so that other TK applications
+        # won't be affected (Windows only).
+        if sys.platform.lower().startswith('win') and 'TKPATH' not in os.environ:
+            os.environ['TKPATH'] = os.pathsep
+
     return out