功能：
1、通过简单的命令行添加修改shell脚本。
2、增加分类功能，让自己的环境变量分类清晰。
3、快速查看脚本。
安装：
1、文件名存储为eshell.py,执行命令python3 eshell.py install（需安装python3）
2、source ~/.bash_profile
使用示例：
1、eshell add java 'export java=/Users/xxxx/jdk/java' 添加分类java，并新增脚本。添加完成后执行reshell生效。
2、eshell list 显示所有分类
3、eshell list java 显示java分类下定义的脚本
4、eshell rm java 移除java类的所有脚本
5、
a、eshell vim iterm 手动编辑（需安装vim）。修改完成后执行reshell生效。
b、输入以下内容：
echo '配置iterm'
export CLICOLOR=1
export LSCOLORS=gxfxcxdxbxegedabagacad
export PS1='\[\033[01;32m\]\u@\h\[\033[00m\]:\[\033[01;36m\]\w\[\033[00m\]\$ '
export TERM=xterm-256color
c、reshell
eshell.py：
import os
import sys
import subprocess
import datetime
_PYTHON = 'default'
_ENCODING = 'utf-8'
_BAK_PREFIX = 'bak/'
class Log(object):
def __init__(self, debug):
self.debug = debug
def debug(self, msg):
# if self.debug is True:
print_on_screen(msg)
@classmethod
def info(self, msg):
print(msg)
log: Log = None
def shell_str(s):
return '\'{}\''.format(s)
def create_backup_name(list):
return eshell_dir() + os.path.sep + _BAK_PREFIX + list + '-' + str(datetime.datetime.now())
def back_up(list):
command = 'cp \'{0}\' \'{1}\''.format(cmd_list_path(list), create_backup_name(list))
# log.debug(command)
print_on_screen(command)
os.system(command)
def print_on_screen(s):
os.system('echo ' + '\'{}\''.format(s))
def bash_sh_path():
return subprocess.getoutput('echo ~/.bash_profile')
def source_bash_shell():
return os.system('source ' + bash_sh_path())
def eshell_dir():
root_path = subprocess.getoutput('echo ~')
dir_path = root_path + '/.eshell'
return dir_path
def cmd_list_path(cmd_list):
return eshell_dir() + '/' + cmd_list
def edit_file_replace(file_path, old_str, new_str):
with open(file_path, 'r', encoding=_ENCODING) as fr:
content = fr.read()
content = content.replace(old_str, new_str)
with open(file_path, "w", encoding = _ENCODING) as fw:
fw.write(content)
def read_cmd_list_file(cmd_list):
cmd_list_check(cmd_list)
with open(cmd_list_path(cmd_list), 'r', encoding=_ENCODING) as fr:
return fr.read()
def write_cmd_list_file(cmd_list, cmd):
cmd_list_check(cmd_list)
text = read_cmd_list_file(cmd_list)
if cmd is None or cmd == '':
return
if cmd in text:
print_on_screen('****** Already exists, Not added again! *******')
return
if text is None or text == '':
text = cmd
else:
text = text + '\n' + cmd
with open(cmd_list_path(cmd_list), 'w', encoding='utf-8') as fw:
fw.write(text)
def eshell_dir_check():
dir_path = eshell_dir()
if os.path.exists(dir_path) is False:
os.system('mkdir ' + dir_path)
dir_path = eshell_dir()
if os.path.exists(dir_path + '/bak') is False:
os.system('mkdir ' + dir_path+'/bak')
return dir_path
def cmd_list_check(cmd_list):
eshell_dir_check()
file_path = cmd_list_path(cmd_list)
if os.path.exists(file_path) is False:
os.system('touch ' + file_path)
return file_path
def bash_profile_add_list(cmd_list):
# 配置bash_profile
with open(bash_sh_path(), 'r', encoding=_ENCODING) as fr:
bash_profile = fr.read()
new_sh = '\nsource ' + cmd_list_path(cmd_list)
if new_sh not in bash_profile:
finish_sh = bash_profile + new_sh
with open(bash_sh_path(), 'w', encoding=_ENCODING) as fw:
fw.write(finish_sh)
def bash_profile_rm_list(cmd_list):
# 配置bash_profile
with open(bash_sh_path(), 'r', encoding=_ENCODING) as fr:
bash_profile = fr.read()
new_sh = '\nsource ' + cmd_list_path(cmd_list)
finish_sh = bash_profile.replace(new_sh, '')
with open(bash_sh_path(), 'w', encoding=_ENCODING) as fw:
fw.write(finish_sh)
def all_list()->(str, set):
files = os.listdir(eshell_dir())
list = set()
list_text = ''
for file in files:
if file != 'init' and file != 'eshell.py' \
and '.swp' not in file and os.path.isfile(eshell_dir() + os.path.sep + file):
list_text = list_text + file + ' '
list.add(file)
return list_text, list
def add_cmd_list_command(cmd_list, command):
if cmd_list is None:
print_on_screen('eshell add [cmd_list] [command]')
return
write_cmd_list_file(cmd_list, command)
bash_profile_add_list(cmd_list)
source_bash_shell()
print_on_screen('Please run: source {0}, or exec: reshell, or active this in a new tab.'.format(bash_sh_path()))
if __name__ == '__main__':
_PYTHON = sys.executable
try:
# show_all_cmd()
arg1 = None
arg2 = None
arg3 = None
arg4 = None
try:
arg0 = sys.argv[0]
arg1 = sys.argv[1]
arg2 = sys.argv[2]
arg3 = sys.argv[3]
arg4 = sys.argv[4]
except BaseException as e:
a = 1
log = Log(arg1 == "main.py")
if arg1 == 'install':
py_path = os.path.abspath(__file__)
print_on_screen('current path:' + py_path)
dest_dir = eshell_dir_check()
dest_path = dest_dir + '/' + 'eshell.py'
print_on_screen('copy to: ' + dest_path)
os.system('cp {0} {1}'.format(py_path, dest_path))
edit_file_replace(dest_path, '_PYTHON = \'default\'', "_PYTHON = " + "'" + _PYTHON + "'")
add_cmd_list_command('init', 'alias eshell=\'{0}\''.format(_PYTHON + ' ' + dest_path))
add_cmd_list_command('init', 'alias reshell=' + "'" + 'source ' + bash_sh_path() + "'")
if arg1 is None or arg1 == '--help':
print_on_screen('eshell add list command : ## add new shell command')
print_on_screen('eshell list : ## show all list')
print_on_screen('eshell list list : ## show all commands in one list')
print_on_screen('eshell rm list1 list2 list3... : ## delete list')
if arg1 == 'list':
list = arg2
if list is None:
list_text, list = all_list()
print_on_screen(list_text)
else:
text = read_cmd_list_file(list)
print_on_screen(text)
if arg1 == 'add':
add_cmd_list_command(arg2, arg3)
source_bash_shell()
if arg1 == 'vim':
list_text, list = all_list()
if arg2 in list:
# log.debug(list)
os.system('vim ' + cmd_list_path(arg2))
else:
# log.debug(list)
decision = input("No This list, Add new[Y/n]：")
if decision.lower() == 'y':
os.system('vim ' + cmd_list_path(arg2))
else:
print_on_screen('canceled!!!')
source_bash_shell()
if arg1 == 'rm':
decision = input("Can't recover, really delete?[Y/n]：")
if decision.lower() == 'y':
for i in range(len(sys.argv)):
if i >= 2:
back_up(sys.argv[i])
os.system('rm ' + cmd_list_path(sys.argv[i]))
bash_profile_rm_list(sys.argv[i])
source_bash_shell()
else:
print_on_screen('canceled!!!')
except BaseException as e:
print_on_screen(e)
网友评论