如何用
Python
写一段用户登录代码
本篇文章小编主要是想带大家一起来看一下如何用
Python
写一段用户登录的程序代码,对
此感兴趣的小伙伴可以随着小编一起来了解下。
如下所示:
#!/usr/bin/env python
#coding: utf8
import getpass
db = {}
def newUser():
username = raw_input('username: ')
if username in db:
#
添加打印颜色
print "\033[32;1m%s already exists![0m" % username
else:
#
屏幕不显示密码,调用
getpass.getpass()
password = getpass.getpass()
db[username] = password #
字典
k-v
赋值
def oldUser():
username = raw_input('username: ')
password = getpass.getpass()
if username in db:
if db.get(username) == password:#
判断输入的用户名和密码是否和字典的
k-v
匹配
print '\033[32;1mlogin successful!\033[0m'
else:
print '\033[32;1mpassword not match username\033[0m'
else:
print '\033[32;1musername does not exist\033[0m'
CMDs = {'n':newUser,'o':oldUser}
def showMenu():
prompt = """(N)ew user
(O)ld user
(Q)uit
input your choice: """
while True:
try:#
捕获
ctrl+c ctrl+d
的异常
choice = raw_input(prompt).strip().lower()[0]