shell脚本实例-菜单样例

1.9.1 实例需求

用户在进行Linux系统管理的过程中，经常需要用到查看进程的信息、用户的信息等常用的功能。本例针对这一需求，使用shell编程实现基本的系统管理
功能。通过本程序，可以按照要求实现查看进程信息、查看当前登录系统的用户信息等功能。程序主界面颜色可以更改，也可以显示帮助信息。
1.9.2 系统设计
根据需求，首先考虑到界面的美观性，将主界面设计为菜单形式。个系统管理功能作为其中的一个菜单项。当用户选择某项功能后，则执行相应的代码段完成功能并输出。输出结束后再返回主菜单，供用户继续选择。流程图如图1-1所示。



[align=center] 图1-1 程序流程图[/align]

1.9.3 程序代码
按照上述设计思想，程序可以划分为几个模块：界面修改模块、主菜单模块、查看进程信息模块、查看用户信息模块和帮助模块。程序代码如下：

#############################################################
#界面修改模块：color函数

#根据输入颜色信息修改界面颜色
color()
{
case $1 in
black_green) #黑底绿字
echo -e "\033[40;32m"
;;
black_yellow) #黑底黄字
echo -e "\033[40;33m"
;;
black_white) #黑底白字
echo -e "\033[40;37m"
;;
black_blue) #黑底蓝字
echo -e "\033[40;34m"
;;
*) #默认为黑底白字
echo -e "\033[30;37m"
;;
esac
}

#############################################################
#主菜单模块

#清屏
echo -e "\033[2J"
#设置信号处理
trap "" 1 2 3
#日期信息
date=`date +%d/%m/%y`
#机器名信息
host=`hostname`
#当前用户信息
user=`whoami`

#循环显示主菜单
while :
do
#显示主菜单，menu作为cat命令的结束符
cat <<menu
---------------------------------------------------
$user $host $date
---------------------------------------------------
1：改变字体颜色
2：查看进程信息
3：查看用户信息
h：帮助
q：退出
---------------------------------------------------
menu
#cat命令结束

#提示用户输入选项
echo -e -n "\t请输入您的选择【1,2,3,h,q】："
#读取用户输入的选择项
read choice
#判断choice值
case $choice in
1) #用户输入1
#while循环，接收用户输入的颜色选择
while :
do
#显示颜色选择菜单
cat <<menu_color
---------------------------------------------------
$user $host $date
---------------------------------------------------
1：黑绿 2：黑黄 3：黑白 4：黑蓝 0：返回
---------------------------------------------------
menu_color
#cat命令结束
#提示用户输入颜色选项
echo -e -n "\t请输入选择的颜色【1，2，3，4，0】："
#读取用户的输入
read choice_color
#判断用户输入的颜色选项
case $choice_color in
1) #用户输入1
#调用color函数设置颜色
color black_green
;;
2) #用户输入2
#调用color函数设置颜色
color black_yellow
;;
3) #用户输入3
#调用color函数设置颜色
color black_white
;;
4) #用户输入4
#调用color函数设置颜色
color black_blue
;;
0) #用户输入0
#退出当前菜单
break
;;
*) #用户输入其它任意值
#输入错误，清屏继续等待输入
echo -e "\033[2J"
echo -e "\tInput error! Please input again!"
continue
;;
#case结束
esac

clear
#while结束
done
;;

#############################################################
#查看进程信息模块

2) #用户输入2
#调用ps命令查看进程信息
ps aux|sort -m|head -10
;;

#############################################################
#查看用户信息模块

3) #用户输入3
#调用who命令查看用户信息
who
;;

#############################################################
#帮助模块

H|h) #用户输入H或h
#调用cat显示帮助信息
cat <<menu_help
选择“改变字体颜色“可以修改当前界面的字体颜色
选择“查看进程信息“可以获取当前占用资源最高的前10个进程
选择“查看用户信息“可以获取当前登录系统的用户信息
menu_help
#cat命令结束
;;

Q|q) #用户输入Q或q
#推出主程序
exit 0
;;
*) #用户输入其他值
#输入错误，清屏继续等待输入
echo -e "\033[2J"
echo -e "\tInput error! Please input again!"
continue
;;
#case语句结束
esac
#执行完命令后的提示
echo -e -n "\t按任意键继续..."
read anykey
clear
#while循环结束
done

[root@250-shiyan prog]# cat menu
#!/bin/bash

####
color()
{
case $1 in
black_green)
echo -e "\033[40;32m"
;;
black_yellow)
echo -e "\033[40;33m"
;;
black_white)
echo -e "\033[40;37m"
;;
black_blue)
echo -e "\033[40;34m"
;;
*)
echo -e "\033[30;37m"
;;
esac
}
####

echo -e "\033[2j"
trap "" 1 2 3
date=`date +%d/%m/%y`
host=`hostname`
user=`whoami`

####
while :
do
cat <<menu
-------------------------------------------------
$user          $host           $date
-------------------------------------------------
1:change font color
2:view process info
3:view user info
h:help
q:quit
-------------------------------------------------
menu
echo -en "please input choice [1,2,3,h,q]:"
read choice
case $choice in
1)
####
while :
do
cat <<menu_color
*************************************************
$user          $host           $date
*************************************************
1:black_green
2:balck_yellow
3:black_white
4:black_blue
0:back
*************************************************
menu_color
echo -ne "please input color [1,2,3,4,0]:"
read color
case $color in
1)
color black_green;;
2)
color black_yellow;;
3)
color black_white;;
4)
color black_blue;;
0)
break;;
*)
echo -e "\033[2j"
echo -e "input error! please input again!"
continue
;;
esac
clear
done
;;
####
2)
ps aux|sort -m|head -10
;;
3)
who
;;
H|h)
cat <<menu_help
select1 "change font color" can alert current UI's color
select1 "view ps info" can view ps info
select1 "view user info" can view user info
menu_help
;;
Q|q)
exit 0
;;
*)
echo -e "\033[2j"
echo -e "input error! please input again!"
continue
;;
esac
echo -ne "please press any key to continue"
read anykey
clear
done
[root@250-shiyan prog]#

shell菜单样例

#!/bin/sh
#menu
#set the date,user and hostname up
MYDATE=`date +%d/%m/%Y`
THIS_HOST=`hostname -s`
USER=`whoami`
PWD=`pwd`

trap "" 2 3 15

####colour function
colour ()
{
case $1 in
black_green)
echo -e '\033[40;32m'
;;
black_yellow)
echo -e '\033[40;33m'
;;
black_white)
echo -e '\033[40;37m'
;;
black_cyan)
echo -e '\033[40;36m'
;;
red_yellow)
echo -e '\033[41;33m'
;;
esac
}
#### single key function
get_char()
{
#get_char
#save current stty setting
SAVEDSTTY=`stty -g`
#如果你的机器上不认stty raw那么把函数中两处出现的raw换成cbreak
stty cbreak
dd if=/dev/tty bs=1 count=1 2>/dev/null
stty -cbreak
#restore stty
stty $SAVEDSTTY
}
#### while1 function
while1()
{
while [ 1 -gt 0 ]
do
sleep 3
echo used
echo "`free |awk '/Mem/{print $3}'`"
done
}
#### for1 function
for1(){
for i in {1,2,3,4}
do
sleep 3
echo used
echo "0:`free |awk '/Mem/{print $3}'`:freeeeeeeee"
done
}
#### while2 function
while2 () {
min=1
max=100
while [ $min -le $max ]
do
echo $min
min=`expr $min + 1`
done
}
#### while3 function
while3 () {
i=1
while(($i<100))
do
if(($i%4==0))
then
echo $i
fi
i=$(($i+1))
done
}

#loop forever !
while :
do
# clear the screen
tput clear
colour black_green
# here documents starts here
cat <<MAYDAY
---------------------------------------------------------------------------
User:$USER  Host:$THIS_HOST  Date:$MYDATE  Pwd:$PWD
---------------------------------------------------------------------------
1 : while1                            2 : while2
3 : while3                            4 : for1
H : Help screen
Q : Exit Menu
---------------------------------------------------------------------------
MAYDAY
# here document finished
colour black_cyan
echo -e -n "\tYour Choice [1,2,3,4,H,Q] >"
read CHOICE
case $CHOICE in
1) while1 ;;
2) while2 ;;
3) while3 ;;
4)
echo -e  "\e[31mTHIS is source\e[0m"
cat <<MAYDAY
for1(){
for i in {1,2,3,4}
do
sleep 3
echo used
echo "0:`free |awk '/Mem/{print $3}'`:freeeeeeeee"
done
}
MAYDAY
echo -e "\e[31mTHIS is result\e[0m"
for1 ;;
H|h)
cat <<MAYDAY
this is the help screen,nothing here yet to help you !
MAYDAY
;;
Q|q) exit 0
;;
*) echo -e "\t\041Unknown user response"
;;
esac
echo -e -n "\tHit the any key to continue"
get_char
done