python实现按任意键继续执行程序

Python  /  管理员 发布于 7年前   192

在windows下写bat的时候，通过pause命令，可以暂停程序运行，例如经常见的程序会在终端提示”按任意键继续……”,用户在终端回车后程序可以接着运行，这个功能有多大用途今天暂且不说，但我觉得应该有很多人也想在python下实现这个功能，这样当自己写的python程序运行时， 突然给出这么一个提示，然后自己再来个帅气的回车，我觉得肯定给人一种非常专业的感觉，最起码自己肯定被自己迷倒了，所以今天我们就学习下这段代码，这里定义了一个函数，所以你可以嵌入到你的程序中，在任何你希望调用的地方调用它即可，使用非常的方便，代码如下：

#!/usr/bin/env python# -*- coding:utf-8 -*-import osimport sysimport termiosdef press_any_key_exit(msg):# 获取标准输入的描述符fd = sys.stdin.fileno()# 获取标准输入(终端)的设置old_ttyinfo = termios.tcgetattr(fd)# 配置终端new_ttyinfo = old_ttyinfo[:]# 使用非规范模式(索引3是c_lflag 也就是本地模式)new_ttyinfo[3] &= ~termios.ICANON# 关闭回显(输入不会被显示)new_ttyinfo[3] &= ~termios.ECHO# 输出信息sys.stdout.write(msg)sys.stdout.flush()# 使设置生效termios.tcsetattr(fd, termios.TCSANOW, new_ttyinfo)# 从终端读取os.read(fd, 7)# 还原终端设置termios.tcsetattr(fd, termios.TCSANOW, old_ttyinfo)if __name__ == “__main__”:press_any_key_exit(“按任意键继续…”)print ‘\n'

代码不多解释，看注释，这里要说一下termios模块，这个模块提供了一个接口可以控制tty终端的Io，它所有函数第一个参数需要一个文件描述符，可以是整型的文件描述符，也可以是个文件对象，因为它可以控制在终端的显示设置，常用场景就是用户在终端输入密码时不要显示出来，就跟我们用root登陆系统时系统给出输入密码提示符一样， 代码实现如下：

def getpass(prompt=”Password: “):import termios, sysfd = sys.stdin.fileno()old = termios.tcgetattr(fd)new = termios.tcgetattr(fd)new[3] = new[3] & ~termios.ECHOtry:termios.tcsetattr(fd, termios.TCSADRAIN, new)passwd = raw_input(prompt)finally:termios.tcsetattr(fd, termios.TCSADRAIN, old)return passwdpasswd = getpass()print passwd

这个脚本运行会提示你输入密码，输入后会打印出刚输入的密码，给这个2个例子也是说明termios的简单用法，大家可以自己运行程序去体会下。