综合案例-登录注册-文件操作

综合案例-登录注册-文件操作

目录结构

代码

"""
 注册   登录     查询
"""
import os
def zhuce():#注册功能。一个账号是一个文件
    userID=input("请输入账号:")
    userPWD=input("请输入密码:")
    #账号密码封装,保存到字典中
    userInfo={"userID":userID,"userPWD":userPWD}
    print(f"注册的信息是：{userInfo}")
    #拼接文件名称
    fileTxt='users/'+userID+".txt"
    print(fileTxt)
    f=open(fileTxt,'w',encoding="utf-8")#打开文件
    f.write(str(userInfo))#将字典转换成字符串之后写入到文件中
    print("注册成功")
    f.close()
def denglu():#登录
    userID=input("请输入登录的用户:")
    userPWD=input("请输入登录的密码:")
    fileNames=os.listdir('users')#获取到users文件夹下的所有文件
    print(fileNames)
    flg=0#标志符 如果等于0 说明不存在  如果=1说明文件存在
    for filename in fileNames:
        if filename==(userID+".txt"):#判断文件是否存在
            flg=1
    if flg==0:
        print("用户不存在")
    elif flg==1:
        print("用户存在")
        # 拼接文件名称
        fileTxt = 'users/' + userID + ".txt"
        f=open(fileTxt,'r',encoding="utf-8")
        uuu=eval(f.read())#读取到的数据转换成字典
        if uuu['userPWD']==userPWD:
            print("登录成")
        else:
            print("登录失败")
            
def chaxun():#查询所有用户信息
    count=0#统计用户人数
    print("用户信息如下:")
    fileUsers=os.listdir("users")#读取users文件夹所有的用户文件
    print("用户\t\t密码")
    for user in fileUsers:
        # 拼接文件名称
        fileTxt = 'users/' + user
        f=open(fileTxt,encoding="utf-8")
        dictUser=eval(f.read())
        f.close()
        print(f"{dictUser['userID']}\t\t{dictUser['userPWD']}")
        count+=1
    print(f"所有用户总计:{count}")
    
def caidan():#功能菜单
    print("1.注册  2.登录    3.查询")
    num=input("请输入功能编号:")
    if num=="1":
        zhuce()
    elif num=='2':
        denglu()
    elif num=='3':
        chaxun()

if __name__ == '__main__':
   while True:
    caidan()