利用shell + python + expect實現相似xshell功能（測試環境：ubuntu）
#各個腳本的功能介紹
#meanue.sh 主要登錄框架
#mk_key_filec python實現的將用戶信息編碼存放，最初想法是加密，這塊之後優化爲加密
#zenity ubuntu自帶的msg窗口（大概是這樣滴。。。）
#auto_ssh.sh 自動登錄遠程服務器腳本
#login_ssh 這是一個expect腳本，實現登錄功能
下面分別給出每一個腳本的代碼：
meanue.sh:
#!/bin/sh
#Name:meanue.sh
#Version:v1.3
#Author:Linux-c
#Date:2014-03-01
trap _exit 2
config=$HOME/sbin/.passwd
key_file=$HOME/sbin/.key.dat
mk_pass=$HOME/sbin/mk_key_filec
clear_file()
{
rm -f $config info.$$$$
}
_exit()
{
clear_file
exit
}
mk_key()
{
rm -f $key_file
`python $mk_pass -t encode -i $config -o $key_file`
return 0
}
un_key()
{
`python $mk_pass -t decode -i $key_file -o $config `
return 0
}
add_user()
{
echo ""
echo -n "Ip:"
read ip
echo -n "User:"
read user
echo -n "Password:"
read passwd
echo $ip $user $passwd >>$config
mk_key
return
}
add_user_gui()
{
title='Auto Login v1.0'
text='Please write ip '
wide=200
high=100
zenity --title="$title" --text="$text" --window-icon="" \
--width=$wide --height=$high --entry 1>info.$$$$ 2>/dev/null
ip="`cat info.$$$$`"
zenity --title="$title" --window-icon="" \
--width=$wide --height=$high --password --username 1>info.$$$$ 2>/dev/null
user="`cat info.$$$$|cut -d"|" -f1`"
passwd="`cat info.$$$$|cut -d"|" -f2`"
echo $ip $user $passwd >>$config
mk_key
return
}
del_user_gui()
{
title='Auto Login v1.0'
text='Please Choose Delete Ip '
wide=200
high=200
login_info="`cat $config|awk '{print $1}'`"
zenity --title="$title" --window-icon="" \
--width=$wide --height=$high --list \
--text="$text" \
--column=index $login_info \
--hide-header=1 1>info.$$$$ 2>/dev/null
option="`cat info.$$$$`"
sed -i "/$option/d" $config
mk_key
return 0
}
del_user()
{
echo -n "請選擇刪除主機編號:"
read option
case $option in
[0-9]*) id=$option
echo -n "確認刪除" `awk '{
if(NR=='"$id"')
printf("%s\n",$1)
}' $config`"的配置信息嗎？(Y/N)"
read anws
echo $anws
if [ "$anws" = "Y" -o "$anws" = "y" ]
then
sed -i "${id}d" $config
mk_key
else
return 0
fi
;;
q|Q|quit|QUIT|exit|EXIT) return 0
;;
esac
return 0
}
main_gui()
{
un_key >/dev/null
clear
if [ "$config" = "" ]
then
echo "生成主機信息失敗!"
fi
add="Add a new user!"
del="Delete a new user!"
high=300
wide=500
title='Auto Login v1.0'
text='Please Choose Login Ip From This List'
login_info="`cat $config|awk '{print $1}'`"
zenity --title="$title" --window-icon="" \
--width=$wide --height=$high --list \
--text="$text" \
--column=index $login_info "$add" "$del" \
--hide-header=1 1>info.$$$$ 2>/dev/null
option="`cat info.$$$$|awk '{print $1}'`"
case $option in
[0-9]*) #ip=`awk '{if(NR=="'$option'") print $1}' $config`
auto_ssh.sh $option
[ $? -eq 2 ]&&echo "No record!Please add new user!"&&sleep 1&&main
;;
Delete) del_user_gui&&main_gui
;;
Add) add_user_gui&&main_gui
;;
q|Q|quit|QUIT|exit|EXIT) _exit ;;
esac
}
main()
{
un_key >/dev/null
clear
if [ "$config" = "" ]
then
echo "生成主機信息失敗!"
fi
awk '{printf("\t%d:%s\n",++i,$1)}END{printf("\tn:新增主機\n\td:刪除主機信息\n")}' $config
echo -n "請選擇須要登錄的主機:"
read option
case $option in
[0-9]*) ip=`awk '{if(NR=="'$option'") print $1}' $config`
auto_ssh.sh $ip
[ $? -eq 2 ]&&echo "No record!Please add new user!"&&sleep 1&&main
;;
d) del_user&&main
;;
n) add_user&&main
;;
q|Q|quit|QUIT|exit|EXIT) _exit ;;
esac
}
chk_gui()
{
flag=`whereis zenity`
if [ ${#flag} -gt 8 ]
then
return 1
fi
return 0
}
chk_gui
result=$?
if [ $result -eq 1 ]
then
main_gui
else
main
fi
_exit
mk_key_filec 是mk_key_file腳本編譯成的
編譯代碼以下：
#!/usr/bin/python
import py_compile
import sys
py_compile.compile(sys.argv[1])
mk_key_file代碼以下：
#!/usr/bin/python
#-*- ecoding= utf-8 -*-
import os
import sys
import base64
import getopt
def usage():
print """
python %s -t encode -i inputfile -o outputfile
-t : encode make key file
decode make source file
-i :input file name
-o :out put file name
""" % sys.argv[0]
sys.exit(0)
def chk_file(file_name):
if os.path.exists(file_name):
return True
else:
return False
def encode(flag,i_file,o_file):
if chk_file(i_file):
i_fd=open(i_file,'r')
else:
usage
if not chk_file(o_file):
o_fd=open(o_file,'a')
else:
print "Warning:%s is exist,Delete it before run this programe!" % o_file
sys.exit(0)
if flag == 'encode':
encode_lines=base64.encodestring(i_fd.read()) # 對字符串編碼
o_fd.write(encode_lines)
elif flag == 'decode':
decode_lines=base64.decodestring(i_fd.read()) # 對字符串解碼
o_fd.write(decode_lines)
else:
usage
i_fd.close()
o_fd.close()
def main():
try:
flag=''
inputfile=''
outputfile=''
opts, args = getopt.getopt(sys.argv[1:], "t:i:o:h",\
["help","type=","infile=","outfile="])
for opt,arg in opts:
if opt in ("-h","--help"):
usage()
elif opt in ("-t","--type"):
flag=arg
elif opt in ("-i","--infile"):
inputfile=arg
elif opt in ("-o","--outfile"):
outputfile=arg
else:
assert False ,"unhandled option"
if flag == '' or inputfile =='' or outputfile=='':
usage()
else :
encode(flag,inputfile,outputfile)
except getopt.GetoptError:
usage()
if __name__ == '__main__':
main()
nantian@nantian-ThinkPad-Edge-E430:~/sbin$ vi mk_key_file
nantian@nantian-ThinkPad-Edge-E430:~/sbin$ cat mk_key_file
#!/usr/bin/python
#-*- ecoding= utf-8 -*-
import os
import sys
import base64
import getopt
def usage():
print """
python %s -t encode -i inputfile -o outputfile
-t : encode make key file
decode make source file
-i :input file name
-o :out put file name
""" % sys.argv[0]
sys.exit(0)
def chk_file(file_name):
if os.path.exists(file_name):
return True
else:
return False
def encode(flag,i_file,o_file):
if chk_file(i_file):
i_fd=open(i_file,'r')
else:
usage
if not chk_file(o_file):
o_fd=open(o_file,'a')
else:
print "Warning:%s is exist,Delete it before run this programe!" % o_file
sys.exit(0)
if flag == 'encode':
encode_lines=base64.encodestring(i_fd.read()) # 對字符串編碼
o_fd.write(encode_lines)
elif flag == 'decode':
decode_lines=base64.decodestring(i_fd.read()) # 對字符串解碼
o_fd.write(decode_lines)
else:
usage
i_fd.close()
o_fd.close()
def main():
try:
flag=''
inputfile=''
outputfile=''
opts, args = getopt.getopt(sys.argv[1:], "t:i:o:h",\
["help","type=","infile=","outfile="])
for opt,arg in opts:
if opt in ("-h","--help"):
usage()
elif opt in ("-t","--type"):
flag=arg
elif opt in ("-i","--infile"):
inputfile=arg
elif opt in ("-o","--outfile"):
outputfile=arg
else:
assert False ,"unhandled option"
if flag == '' or inputfile =='' or outputfile=='':
usage()
else :
encode(flag,inputfile,outputfile)
except getopt.GetoptError:
usage()
if __name__ == '__main__':
main()
auto_ssh.sh代碼以下：
#!/bin/bash -
#gconftool --set --type=string /apps/gnome-terminal/profiles/Default/encoding GB18030
#export.GB18030
#LANG="zh_CN.UTF-8"
export
config_file=$HOME/sbin/.passwd
ip="$1"
user="$2"
passwd="$3"
resualt=""
gal_flag=""
if [ ! -e "$config_file" ]
then
touch $config_file
fi
help()
{
exit 2
}
chk_config_file()
{
resualt="`grep "\<$ip\>" $config_file`"
if [ "$resualt" = "" ]
then
gal_flag=-1
else
gal_flag=0
fi
}
if [ -z $ip ]
then
help
fi
chk_config_file
login_user="`echo "$resualt"|cut -d " " -f2`"
login_passwd="`echo "$resualt"|cut -d " " -f3`"
if [ "$gal_flag" = "0" ]
then
expect $HOME/sbin/login_ssh $ip $login_user $login_passwd
elif [ "$user" != "" ]
then
expect $HOME/sbin/login_ssh $ip $user $passwd && echo "$ip $user $passwd" >>$config_file
else
help
fi
#gconftool --set --type=string /apps/gnome-terminal/profiles/Default/encoding utf-8
export
login_ssh代碼以下：
#!/usr/bin/expect
set timeout 60
if {$argc < 3} {
send "usage:login ip user passwd!\n"
exit
}
set ip [lindex $argv 0 ]
set user [lindex $argv 1 ]
set passwd [lindex $argv 2 ]
#send "user :$user\n"
#send "pwd :$passwd\n"
#send "ip :$ip\n"
spawn ssh $user@$ip
expect {
"*yes*" {
send "yes\r"
exp_continue
}
"*word*" {
send "$passwd\r"
}
}
interact
相關文章
- 1. golang實現相似var_dump()功能
- 2. 實現相似tail -f file功能
- 3. PostgreSQL相似Oracle MERGE功能的實現
- 4. shell腳原本實現相似回收站功能
- 5. svn 相似.gitignore功能實現
- 6. ava 相似jest snapshot 功能試用
- 7. 模擬測試環境實現隨機造數功能
- 8. shell相關之expect
- 9. [前端引用] 利用ajax實現相似php include require 等命令的功能
- 10. python實現相似awk的簡單功能
- 更多相關文章...
- • 使用Redis和Lua的原子性實現搶紅包功能 - 紅包項目實戰
- • C# 環境 - C#教程
- • ☆基於Java Instrument的Agent實現
- • NewSQL-TiDB相關
相關標籤/搜索
每日一句
-
每一个你不满意的现在，都有一个你没有努力的曾经。
最新文章
- 1. Duang!超快Wi-Fi來襲
- 2. 機器學習-補充03 神經網絡之**函數(Activation Function)
- 3. git上開源maven項目部署 多module maven項目（多module maven+redis+tomcat+mysql）後臺部署流程學習記錄
- 4. ecliple-tomcat部署maven項目方式之一
- 5. eclipse新導入的項目經常可以看到「XX cannot be resolved to a type」的報錯信息
- 6. Spark RDD的依賴於DAG的工作原理
- 7. VMware安裝CentOS-8教程詳解
- 8. YDOOK：Java 項目 Spring 項目導入基本四大 jar 包 導入依賴，怎樣在 IDEA 的項目結構中導入 jar 包 導入依賴
- 9. 簡單方法使得putty（windows10上）可以免密登錄樹莓派
- 10. idea怎麼用本地maven
歡迎關注本站公眾號,獲取更多信息
