近来做作业(老男孩那个9.9元的训练营)我想写一个装逼点的密文输入密码,类似于:
这个东西我先前实现过,忘了获取一个字节的方法是什么,于是去网上找,发现网上的实现方式大部分都有问题。
一、网上(百度)的三种实现方式
网上的实现方式不外乎三种:
- 直接明文输入(这是扯淡)
- 使用getpass模块 覆盖输入,无法看到位数,没有退格。(不好用)
- 通过msvcrt模块的getch和putch实现(有问题)
我们重点研究第三种。
二、第三种实现方式的问题何在?
先看代码:
1 def inputPassword():# 密码输入 2 try: 3 li = [] 4 while True: 5 ch = msvcrt.getch() 6 if ch == b'\r': #回车键,确认输入 7 return ''.join(li) # 返回密码字符串 8 elif ch == b'\x08': #退格键,删除最后一个输出 9 del li[-1] #删除密码列表最后一位 10 msvcrt.putch(b'\b') 11 msvcrt.putch(b' ') 12 msvcrt.putch(b'\b')# 这相当于刷新 13 elif ch == b'\x1a': # ctrl+z 退出键,抛出异常 14 raise EOFError 15 else: #否则,就是密码 16 li.append(ch.decode()) # 加入列表 17 msvcrt.putch(b'*') # 遮盖符 18 except EOFError: 19 pass
看似没有问题,然而当实际运行的时候,获取到的密码会变成
"1 2 3 1 2 3"之类
而且可以发现,按下一个键后会出现两个*
这是为什么呢?
原来,当按下一个键时,实际上捕获到两个ch,一个是实际按下的键,另一个是b'\x00',这个东西造成了密码输入的不正确
怎么改?
只需要加入一个判断 处理b'\x00'就行了
正常的代码:
1 def inputPassword():# 密码输入 2 try: 3 li = [] 4 while True: 5 ch = msvcrt.getch() 6 if ch == b'\r': #回车键,确认输入 7 return ''.join(li) # 返回密码字符串 8 elif ch == b'\x08': #退格键,删除最后一个输出 9 del li[-1] #删除密码列表最后一位 10 msvcrt.putch(b'\b') 11 msvcrt.putch(b' ') 12 msvcrt.putch(b'\b')# 这相当于刷新 13 elif ch == b'\x00': #去除空字符 14 pass 15 elif ch == b'\x1a': # ctrl+z 退出键,抛出异常 16 raise EOFError 17 else: #否则,就是密码 18 li.append(ch.decode()) # 加入列表 19 msvcrt.putch(b'*') # 遮盖符 20 except EOFError: 21 pass
你只需要注意13行就可以。