首页 > 代码库 > 【工作中的Python】随机点名小脚本

【工作中的Python】随机点名小脚本

背景:项目组每周的例会中,有一项固定内容就是技术分享,可以是与工作相关或无关的任何技术主题。进行技术分享讲解的同学是随机抽签的。由此做了一个Python的小脚本用于抽取姓名。

脚本内容如下:

#!/usr/bin/pythonimport osimport sysimport tty, termiosimport randomname_list = ["member_1","member_2","member_3"]input = ‘‘fd = sys.stdin.fileno()old_settings = termios.tcgetattr(fd)try:    tty.setraw(fd)    while ‘q‘ != input:        sys.stdout.write(random.choice(name_list) + "\r\n")        input = sys.stdin.read(1)finally:    termios.tcgetattr(fd, termios.TCSADRAIN, old_settings)

代码说明:

随机选择

        $ random.choice从序列中获取一个随机元素。

        $ 其函数原型为:random.choice(sequence)。参数sequence表示一个有序类型。

        $ sequence在python不是一种特定的类型,而是泛指一系列的类型。list, tuple, 字符串都属于sequence。

文件描述符

        $ 获取底层实现使用请求从操作系统的I / O操作的文件描述符:

fileObject.fileno(); 

 

        $ 脚本中获取了标准输入(终端)的文件描述符,后续用于获取输入和向屏幕输出。其实读取标准输入的话,我更喜欢用raw_input()这个函数,不过反正都实现了。输入的时候使用了sys.stdout.write()的原因是,在通过termios改变了终端的属性后,使用print输出时格式会发生变化——如果抽取多次,则每次输出的名字都会在前次输出的下一行并后延一个tab的位置,造成错位,不知道为什么。采用了sys.stdout.write()后就正常了。

Termios与tty模块

        $ (只能在Unix及其派生的系统中使用)这两个模块配合是为了实现程序实时响应用户的键盘输入。运行程序时,输入‘q‘则直接退出程序,而不需要按下回车。

        $ 首先获取当前终端配置

old_settings = termios.tcgetattr(fd)

 

        $ 设置终端为非阻塞的输入方式,按字符响应键盘输入

tty.setraw(fd)

 

        $ 还原终端配置

termios.tcgetattr(fd, termios.TCSADRAIN, old_settings)

 

【工作中的Python】随机点名小脚本