Merge pull request #19 from sotetsuk/cross-platform-support

Cross platform support

paicli/config.py

@@ -17,6 +17,8 @@
 except NameError:
     FileNotFoundError = IOError
 
+HOME = os.path.expanduser("~")
+
 
 class Config(object):
 
@@ -26,8 +28,8 @@ class Config(object):
     _accesstoken = ".token_{}"
 
     # paths
-    path_to_configdir = os.path.join(os.environ['HOME'], _paicli_dirname)
-    path_to_configfile = os.path.join(os.environ['HOME'], _paicli_dirname, _configfile)
+    path_to_configdir = os.path.join(HOME, _paicli_dirname)
+    path_to_configfile = os.path.join(HOME, _paicli_dirname, _configfile)
 
     def __init__(self, profile="default"):
         self.profile = to_str(profile)
@@ -37,7 +39,7 @@ def __init__(self, profile="default"):
         self.access_token = to_str("")
         self.api_version = to_str("v1")
 
-        self.path_to_accesstoken = os.path.join(os.environ['HOME'],
+        self.path_to_accesstoken = os.path.join(HOME,
                                                 self._paicli_dirname, self._accesstoken.format(self.profile))
 
     def add_profile(self):

paicli/jobs.py

@@ -3,12 +3,18 @@
 Author: Sotetsu KOYAMADA
 """
 from __future__ import division
+import sys
 from datetime import datetime
 import json
 from prettytable import PrettyTable
 from termcolor import colored
 from .utils import to_str
 
+if sys.platform.lower()[:3] == "win":
+    import colorama
+    colorama.init()
+
+
 class Jobs(object):
 
     def __init__(self, api, username=""):

paicli/main.py

@@ -24,6 +24,10 @@
 except NameError:
     FileNotFoundError = IOError
 
+if sys.platform.lower()[:3] == "win":
+    import colorama
+    colorama.init()
+
 
 def _load(config):
     try:

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='paicli',
-    version="0.3.2",
+    version="0.3.4",
     description='Client for PAI',
     author='Sotetsu KOYAMADA',
     url='',
@@ -13,6 +13,7 @@
                       "prompt_toolkit==1.0.9",
                       "prettytable",
                       "termcolor",
+                      "colorama",
                       "click",
                       ],
     packages=find_packages(),