python - 如何处理来自 raw_input 的整数和字符串？

Question

仍在努力理解 python。它与 php 非常不同。

我将选择设置为整数，问题出在我的菜单上，我也需要使用字母。

如何同时使用整数和字符串？
为什么我不能设置为字符串而不是整数？

def main(): # Display the main menu
    while True:
        print
        print "  Draw a Shape"
        print "  ============"
        print
        print "  1 - Draw a triangle"
        print "  2 - Draw a square"
        print "  3 - Draw a rectangle"
        print "  4 - Draw a pentagon"
        print "  5 - Draw a hexagon"
        print "  6 - Draw an octagon"
        print "  7 - Draw a circle"
        print
        print "  D - Display what was drawn"
        print "  X - Exit"
        print

        choice = raw_input('  Enter your choice: ')

        if (choice == 'x') or (choice == 'X'):
            break

        elif (choice == 'd') or (choice == 'D'):
            log.show_log()

        try:
            choice = int(choice)
            if (1 <= choice <= 7):

                my_shape_num = h_m.how_many()
                if ( my_shape_num is None): 
                    continue

                # draw in the middle of screen if there is 1 shape to draw
                if (my_shape_num == 1):
                    d_s.start_point(0, 0)
                else:
                    d_s.start_point()
                #
                if choice == 1: 
                    d_s.draw_triangle(my_shape_num) 
                elif choice == 2: 
                    d_s.draw_square(my_shape_num) 
                elif choice == 3:             
                    d_s.draw_rectangle(my_shape_num) 
                elif choice == 4:             
                    d_s.draw_pentagon(my_shape_num) 
                elif choice == 5:             
                    d_s.draw_hexagon(my_shape_num) 
                elif choice == 6:             
                    d_s.draw_octagon(my_shape_num) 
                elif choice == 7: 
                    d_s.draw_circle(my_shape_num)

                d_s.t.end_fill() # shape fill color --draw_shape.py-- def start_point

            else:
                print
                print '  Number must be from 1 to 7!'
                print

        except ValueError:
            print
            print '  Try again'
            print

Answer 1

让我用另一个问题来回答你的问题:
字母和数字真的有必要混用吗？
它们不能都是字符串吗？

好吧，让我们走远一点，看看程序在做什么:

1. 显示主菜单
2. 询问/接收用户输入
   • 如果有效:ok
   • 如果不是:打印错误信息并重复
3. 现在我们有了一个有效的输入
   • 如果是一封信:做一个特殊的任务
   • 如果是数字:调用正确的绘图函数

第 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 0

def 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，它现在不再是一个 list，而是一个 dict，所以如果您只是简单地迭代它像 if choice in options 你的迭代在 keys:['1', '2', 'D', 'X'] ，但是如果你执行 options['X'] 你会得到退出函数(是不是很美妙!)。

现在再改进一下，因为维护主菜单消息和options字典不太好，一年后我可能忘记改一个或另一个，我不会得到我想要的想要而且我很懒，我不想做同样的事情两次，等等......
那么为什么不将 options 字典传递给 display_manu 并让 display_menu 使用 __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

对于options，我们需要OrderedDict而不是dict，因为顾名思义，OrderedDict保留了它的项目顺序(看看 official doc )。所以我们有:

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()

请注意，您必须设计 Action ，使它们都具有相同的签名(无论如何它们都应该是这样的，它们都是 Action !)。您可能希望将可调用对象用作操作:实现了 __call__ 的类实例。创建一个 Action 基类并从中继承将是完美的选择。