学步园

原来以为python会自带这么一相模块，毕竟这是一个很普遍的需求，但查了一些有关python库的reference没有发现有这么个现成的库，在csdn的python版发帖询问了一下：http://topic.csdn.net/u/20101227/15/22408509-7d0d-4660-8d2c-b97822761abc.html?54614，同样没有结果.

工作还得继续，于是乎自己动手，丰衣足食. 代码如下：

#!/usr/bin/env python<br /> '''<br /> Description: This is a menu manage module, it take a dictory data struct as a input and output a text menu<br /> Author: Joshua.guo <shijian.guo@xxxxx.com><br /> Date: 28/12/2010<br /> '''</p> <p>def sf():<br /> '''<br /> just a test function<br /> '''<br /> pass</p> <p>def menu_manage(menu_path_list, menu_dict):<br /> '''<br /> menu_path_list likes below:<br /> menu_path_list = ['1st', '2nd', '3rd']</p> <p> menu_dict likes below:<br /> menu_dict = {<br /> 1 : ('system', 'test.py'),<br /> 2 : ('software', 'test.py'),<br /> 3 : ('user', function),<br /> 4 : ('network', function),<br /> 5 : ('1+1 redundancy', 'test.py'),<br /> 6 : ('snmp', function),<br /> 7 : ('logout', 'test.py'),<br /> }</p> <p> The output menu likes below:</p> <p> 1st -> 2nd -> 3rd:<br /> 1) system<br /> 2) software<br /> 3) user<br /> 4) network<br /> 5) 1+1 redundancy<br /> 6) snmp<br /> 7) logout<br /> Select option number ( b>ack or l>ist ) (b) :<br /> '''</p> <p> #output menu<br /> output_menu(menu_path_list, menu_dict)<br /> ret = 0<br /> while True:<br /> #output input prompt<br /> print ''<br /> input_str = raw_input('Select option number ( b>ack or l>ist ) (b): ')<br /> if input_str in ['', 'b']:<br /> if len(menu_path_list) == 1:<br /> #it is the root menu, just output this menu again<br /> output_menu(menu_path_list, menu_dict)<br /> continue<br /> else:<br /> #back to top menu<br /> return 6</p> <p> elif input_str in ['l']:<br /> print ''<br /> output_menu(menu_path_list, menu_dict)<br /> continue</p> <p> try:<br /> option_number = int(input_str)<br /> except ValueError:<br /> print 'Error, input should be a number!'<br /> continue</p> <p> if option_number not in menu_dict:<br /> print 'Error, input option is invalid!'<br /> continue</p> <p> menu_node = menu_dict[option_number][0]<br /> if type(menu_dict[option_number][1]) == type(sf):<br /> #it is a function<br /> sub_menu_path_list = menu_path_list[:]<br /> sub_menu_path_list.append(menu_node)<br /> func = menu_dict[option_number][1]<br /> ret = func(sub_menu_path_list)<br /> else:<br /> #it is a script<br /> script_name = menu_dict[option_number][1]<br /> module_name = script_name[:-3]<br /> exec('import ' + module_name)<br /> sub_menu_path_list = menu_path_list[:]<br /> sub_menu_path_list.append(menu_node)<br /> exec "ret = %s.main(%s)" % (module_name, sub_menu_path_list)</p> <p> if ret == 6:<br /> #exit from child menu, output parent menu again<br /> output_menu(menu_path_list, menu_dict)<br /> else:<br /> #do nothing<br /> pass</p> <p> return 0</p> <p>def output_menu(menu_path_list, menu_dict):<br /> '''<br /> menu_path_list likes below:<br /> menu_path_list = ['1st', '2nd', '3rd']</p> <p> menu_dict likes below:<br /> menu_dict = {<br /> 1 : ('system', 'test.py'),<br /> 2 : ('software', 'test.py'),<br /> 3 : ('user', function),<br /> 4 : ('network', function),<br /> 5 : ('1+1 redundancy', 'test.py'),<br /> 6 : ('snmp', function),<br /> 7 : ('logout', 'test.py'),<br /> }<br /> The output menu likes below:</p> <p> 1st -> 2nd -> 3rd:<br /> 1) system<br /> 2) software<br /> 3) user<br /> 4) network<br /> 5) 1+1 redundancy<br /> 6) snmp<br /> 7) logout<br /> '''</p> <p> #output menu path<br /> menu_path = ' -> '.join(menu_path_list)<br /> print menu_path + ':'</p> <p> #output menu body<br /> for menu_key in sorted(menu_dict.keys()):<br /> print '/t%d) %s' % (menu_key, menu_dict[menu_key][0])</p> <p> return 0</p> <p>

下面是使用这模块的三个示例文件：

main_menu.py

#!/usr/bin/env python</p> <p>import menu_util</p> <p>def func_b(menu_path_list):<br /> '''<br /> '''<br /> print 'print in func_b()'<br /> return 0</p> <p>root_menu_dict = {<br /> 1 : ('1st_a', 'sub_a.py'),<br /> 2 : ('1st_b', func_b),<br /> } </p> <p>if __name__ == '__main__':<br /> menu_util.menu_manage(['root'], root_menu_dict)

sub_a.py:

#!/usr/bin/env python</p> <p>import menu_util</p> <p>def sub_sub_sub_a(menu_path_list):<br /> print 'in sub_sub_sub_a()'<br /> return 0</p> <p>def sub_sub_sub_b(menu_path_list):<br /> print 'in sub_sub_sub_b()'<br /> return 0</p> <p>def sub_sub_menu_a(menu_path_list):<br /> '''<br /> '''<br /> sub_sub_menu_a_dict = {<br /> 1 : ('3rd_a', sub_sub_sub_a),<br /> 2 : ('3rd_b', sub_sub_sub_b),<br /> }<br /> return menu_util.menu_manage(menu_path_list, sub_sub_menu_a_dict)</p> <p>sub_a_menu_dict = {<br /> 1 : ('2nd_a', sub_sub_menu_a),<br /> 2 : ('2nd_b', 'sub_sub_b.py'),<br /> } </p> <p>def main(menu_path_list):<br /> '''<br /> '''<br /> return menu_util.menu_manage(menu_path_list, sub_a_menu_dict)</p> <p>if __name__ == '__main__':<br /> menu_util.menu_manage(['root'], root_menu_dict)

sub_sub_b.py:

#!/usr/bin/env python</p> <p>def main(menu_path_list):<br /> '''<br /> '''<br /> print 'in sub_sub_b.main()'</p> <p> return 6

运行main_menu.py的结果如下：

[root@louis menu]# ./main_menu.py
root:
        1) 1st_a
        2) 1st_b

Select option number ( b>ack or l>ist ) (b): 1
root -> 1st_a:
        1) 2nd_a
        2) 2nd_b

Select option number ( b>ack or l>ist ) (b): 1
root -> 1st_a -> 2nd_a:
        1) 3rd_a
        2) 3rd_b

Select option number ( b>ack or l>ist ) (b): 1
in sub_sub_sub_a()

Select option number ( b>ack or l>ist ) (b): l

root -> 1st_a -> 2nd_a:
        1) 3rd_a
        2) 3rd_b

Select option number ( b>ack or l>ist ) (b): 2
in sub_sub_sub_b()

Select option number ( b>ack or l>ist ) (b): l

root -> 1st_a -> 2nd_a:
        1) 3rd_a
        2) 3rd_b

Select option number ( b>ack or l>ist ) (b): b
root -> 1st_a:
        1) 2nd_a
        2) 2nd_b

Select option number ( b>ack or l>ist ) (b): 2
in sub_sub_b.main()
root -> 1st_a:
        1) 2nd_a
        2) 2nd_b

Select option number ( b>ack or l>ist ) (b): b
root:
        1) 1st_a
        2) 1st_b

Select option number ( b>ack or l>ist ) (b): 2
print in func_b()

Select option number ( b>ack or l>ist ) (b):