一日一技：用Python如何正确开发命令行交互程序

青南

发布于 2020-02-19 15:51:31

4.1K0

发布于 2020-02-19 15:51:31

我要写一个命令行交互程序，当用户输入不同命令时执行不同操作。

相信看本文的同学，你可能会这样写：

def query_by_id(user_id):
    print(f'查询 id 为：{user_id} 对应的用户')

def query_by_name(user_name):
    print(f'查询用户名为：{user_name} 的用户')

def main():
    while True:
        option = input('请输入选项前面的数字：1. 根据 id 查询；2.根据用户名查询>')
        if option == '1':
            user_id = input('请输入用户 id>')
            query_by_id(user_id)
        elif option == '2':
            user_name = input('请输入用户名>')
            query_by_name(user_name)
        elif option == 'exit':
            print('退出！')
            break
        else:
            print('输入无效，请重新输入')


if __name__ == '__main__':
    main()

运行效果如下图左边所示：

这种方式确实可以达到目的。但 Python 实际上有专门用来实现这个目的的模块，叫做cmd。这是 Python 自带的。

我们来改造一下上面的例子：

import cmd


class Query(cmd.Cmd):
    intro = '用户信息查询系统，输入 help 或者?查看帮助。\n'
    prompt = 'query>'

    def do_query_by_id(self, arg):
        '根据用户 id 查询用户'
        self.query_by_id(arg)

    def do_query_by_name(self, arg):
        '根据用户名查询'
        self.query_by_name(arg)

    def do_exit(self, _):
        '退出'
        exit(0)

    def query_by_id(self, user_id):
        print(f'查询 id 为：{user_id} 对应的用户')

    def query_by_name(self, user_name):
        print(f'查询用户名为：{user_name} 的用户')


if __name__ == '__main__':
    Query().cmdloop()

运行效果如下图所示：