让我用另一个问题回答您的问题:
是否真的需要混合字母和数字?
难道他们不都是字符串吗?
好吧,让我们走很长一段路,看看程序在做什么:
- 显示主菜单
- 询问/接收用户输入
- 如果有效:确定
- 如果不是,请打印错误消息并重复
- 现在我们有一个有效的输入
- 如果是字母:请执行特殊任务
- 如果是数字:调用正确的绘图功能
要点1. 让我们为此做一个功能:
def display_menu(): menu_text = """ Draw a Shape ============ 1 - Draw a triangle 2 - Draw a square D - Display what was drawn X - Exit""" print menu_text
display_menu非常简单,因此无需解释其作用,但是稍后我们将看到将这段代码放入单独的函数中的优势。
第2点。 这将通过循环完成:
options = ['1', '2', 'D', 'X']while 1: choice = raw_input(' Enter your choice: ') if choice in options: break else: print 'Try Again!'
第3点。 好了,想一想也许特殊任务不是那么特殊,所以我们也将它们放入函数中:
def exit(): """Exit""" # this is a docstring we'll use it later return 0def display_drawn(): """Display what was drawn""" print 'display what was drawn'def draw_triangle(): """Draw a triangle""" print 'triangle'def draw_square(): """Draw a square""" print 'square'
现在,我们将它们放在一起:
def main(): options = {'1': draw_triangle, '2': draw_square, 'D': display_drawn, 'X': exit} display_menu() while 1: choice = raw_input(' Enter your choice: ').upper() if choice in options: break else: print 'Try Again!' action = options[choice] # here we get the right function action() # here we call that function
切换键的关键在于
options现在不再是
lista
dict,而是a ,因此,如果您像
if choice in options迭代在
key :上那样简单地对其进行迭代
['1', '2', 'D','X'],但是如果您这样做,
options['X']则会得到exit函数(不是那么好!)。
现在,让我们再次进行改进,因为维护主菜单消息和
options字典并不是太好,一年以后,我可能会忘记更改其中一个,而我将无法获得想要的东西,而且我很懒惰,但我没有想要做两次相同的事情,等等。。。
为什么不将
options字典传递给
display_manu,而是
display_menu使用doc字符串
__doc__生成菜单来完成所有工作:
def display_menu(opt): header = """ Draw a Shape ============""" menu = 'n'.join('{} - {}'.format(k,func.__doc__) for k,func in opt.items()) print header + menu
我们需要
OrderedDict代替
dictfor
options,因为
OrderedDict顾名思义,请保持其项的顺序(请看官方文档)。因此,我们有:
def main(): options = OrderedDict((('1', draw_triangle), ('2', draw_square), ('D', display_drawn), ('X', exit))) display_menu(options) while 1: choice = raw_input(' Enter your choice: ').upper() if choice in options: break else: print 'Try Again!' action = options[choice] action()
请注意,您必须设计动作,使它们都具有相同的签名(无论如何,它们都应该是这样,它们都是动作!)。您可能希望将可调用对象用作 *** 作:具有已
__call__实现的类的实例。创建基
Action类并从中继承将是完美的选择。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)