python注册登录的实现--low版

mr923

#!/usr/bin/env python　　
# -*- coding:utf-8 -*-
　　
import hashlib
　　
import getpass
　　
import sys
　　
def login(name,password):
　　
    file=open("user",'r',encoding='utf-8')
　　
    for user in  file:
　　
        list=user.split(":")
　　
        if list[0] == name and list[1] == password:
　　
            file.close()
　　
            return True
　　
        else:
　　
            file.close()
　　
            return False
　　

　　

　　
def reg(name,password):
　　
    file=open("user",'a',encoding="utf-8")
　　
    info=name+":"+password+"\n"
　　
    file.write(info)
　　
    file.close()
　　

　　

　　
def main():
　　
    global user,pwd
　　
    chonice=input("-----------（选择）1登录/2注册--------：")
　　
    if chonice == "1":
　　
        obj=hashlib.md5()
　　
        user=input("输入您的用户名：")
　　
        pw=getpass.getpass("输入您的密码:")
　　
        obj.update(bytes(pw,encoding='utf-8'))
　　
        pwd=obj.hexdigest()
　　
        for times in range(3):
　　
            if login(user,pwd) == True:
　　
                print("登录成功！！")
　　
                break
　　
            else:
　　
                print("第%d次错误!!!"%(times+1))
　　
                if times == 2:
　　
                    print("已超过3次，用户已禁止登录！！")
　　
    elif chonice == "2":
　　
        reg(user,pwd)
　　
    else:
　　
        print("选择错误退出！！")
　　
        sys.exit(0)
　　
main()