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

#!/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()


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章