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

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

脚本内容如下：

#!/usr/bin/python

import os
import sys
import tty, termios
import random

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