Skip to content

Instantly share code, notes, and snippets.

@jamiesun jamiesun/sh
Created Feb 24, 2015

Embed
What would you like to do?
python color shell
#!/usr/bin/env python
#coding:utf-8
import sys
import os
import subprocess
import time
class ToughError(Exception):
def __init__(self, message):
self.message = message
class ToughShell(object):
def __init__(self,logfile=None,debug=False):
self.logfile = logfile
self.is_debug = debug
self.is_win32 = sys.platform in ['win32']
# error
def inred(self,s):
return self.is_win32 and s or "%s[31;2m%s%s[0m"%(chr(27),s, chr(27))
# success
def ingreen(self,s):
return self.is_win32 and s or "%s[32;2m%s%s[0m"%(chr(27),s, chr(27))
# operate
def inblue(self,s):
return self.is_win32 and s or "%s[34;2m%s%s[0m"%(chr(27),s, chr(27))
# info
def incblue(self,s):
return self.is_win32 and s or "%s[36;2m%s%s[0m"%(chr(27),s, chr(27))
# warning Magenta
def inwarn(self,s):
return self.is_win32 and s or "%s[35;2m%s%s[0m"%(chr(27),s, chr(27))
def log(self,msg,_font=None,logfile=None):
print _font(msg)
if self.logfile:
with open(self.logfile,'ab') as fs:
fs.write(msg)
fs.write('\n')
def info(self,msg):
self.log('[INFO] - %s'%msg,_font=self.incblue)
def debug(self,msg):
self.log('[DEBUG] - %s'%msg,_font=self.inblue)
def succ(self,msg):
self.log('[SUCC] - %s'%msg,_font=self.ingreen)
def err(self,msg):
self.log('[ERROR] - %s'%msg,_font=self.inred)
def warn(self,msg):
self.log('[WARN] - %s'%msg,_font=self.inwarn)
def read(self,ask):
result = raw_input(self.incblue('[INPUT] - %s'%ask))
if self.is_debug:
self.debug('<question - %s | answer - %s>'%(ask,result))
return result
def wait(self,sec=0):
if not sec:return
sec = int(sec)
_range = range(1,sec+1)
_range.reverse()
for i in _range:
self.debug(str(i))
time.sleep(1.0)
def run(self,command, raise_on_fail=False, shell=True, env=None,wait=0):
self.info(">> run command : %s"%command)
_result = dict(code=0)
run_env = os.environ.copy()
if env:run_env.update(env)
if wait > 0:
subprocess.Popen(command, shell=True)
self.wait(wait)
else:
proc = subprocess.Popen(command,shell=shell,
stdout=subprocess.PIPE,stderr=subprocess.PIPE,
env=run_env)
stdout, stderr = proc.communicate('through stdin to stdout')
result = proc.returncode, stdout, stderr
if proc.returncode > 0 and raise_on_fail:
error_string = "# Could not run command (return code= %s)\n" % proc.returncode
error_string += "# Error was:\n%s\n" % (stderr.strip())
error_string += "# Command was:\n%s\n" % command
error_string += "# Output was:\n%s\n" % (stdout.strip())
if proc.returncode == 127: # File not found, lets print path
path = os.getenv("PATH")
error_string += "# Check if y/our path is correct: %s" % path
self.err(error_string)
raise ToughError(error_string)
else:
if self.is_debug:
if stdout.strip():
self.debug(stdout)
if stderr.strip():
self.err(stderr)
if proc.returncode == 0:
self.succ(">> run command : %s success!"%command)
else:
self.err(">> run command : %s failure!"%command)
return result
shell = ToughShell()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.