diff --git a/01.基础语法/01.python基础.md b/01.基础语法/01.python基础.md index 90018fa..abee077 100644 --- a/01.基础语法/01.python基础.md +++ b/01.基础语法/01.python基础.md @@ -1222,51 +1222,3 @@ while True: print("\033[41;37m输入错误!\033[0m") ``` -### 登录注册文件数据库版本 - -```python -while True: - # 获取文件中保存的用户名和密码信息 - dic = {} - with open("db", "a+", encoding="utf-8") as f: - f.seek(0) - temp_data = f.readlines() - for i in temp_data: - data = i.strip() - if len(data) == 0: - continue - temp_db = data.split("✨") - dic[temp_db[0]] = temp_db[-1] - - # 登录注册的逻辑 - print("登录".center(30,"=")) - print("请选择:\n1.登录\n2.注册") - choice = input(">>>") - if choice == "1": - username = input("输入用户名:") - if username in dic.keys(): - password = input("输入密码:") - if dic[username] == password: - print("\033[42;30m 登录成功! \033[0m") - else: - print("\033[41;37m 密码错误 \033[0m") - else: - print("\033[41;37m用户名不存在!\033[0m") - elif choice == "2": - username = input("输入用户名") - if username in dic.keys(): - print("\033[34;41m 用户名已存在!\033[0m") - continue - password = input("请输入密码") - dic[username] = password - print("\033[42;30m 注册成功! \033[0m") - else: - print("\033[41;37m输入错误!\033[0m") - - # 将数据写入文件 - with open("db", "w+", encoding="utf-8") as f: - for k,v in dic.items(): - data = f"{k}✨{v}\n" - f.write(data) -``` - diff --git a/01.基础语法/02.python文件操作.md b/01.基础语法/02.python文件操作.md index b79d82b..d7a08d3 100644 --- a/01.基础语法/02.python文件操作.md +++ b/01.基础语法/02.python文件操作.md @@ -306,4 +306,53 @@ for i in list: price += int(i['price']) * int(i['amount']) print(price) -``` \ No newline at end of file +``` + +### 登录注册文件数据库版本 + +```python +while True: + # 获取文件中保存的用户名和密码信息 + dic = {} + with open("db", "a+", encoding="utf-8") as f: + f.seek(0) + temp_data = f.readlines() + for i in temp_data: + data = i.strip() + if len(data) == 0: + continue + temp_db = data.split("✨") + dic[temp_db[0]] = temp_db[-1] + + # 登录注册的逻辑 + print("登录".center(30,"=")) + print("请选择:\n1.登录\n2.注册") + choice = input(">>>") + if choice == "1": + username = input("输入用户名:") + if username in dic.keys(): + password = input("输入密码:") + if dic[username] == password: + print("\033[42;30m 登录成功! \033[0m") + else: + print("\033[41;37m 密码错误 \033[0m") + else: + print("\033[41;37m用户名不存在!\033[0m") + elif choice == "2": + username = input("输入用户名") + if username in dic.keys(): + print("\033[34;41m 用户名已存在!\033[0m") + continue + password = input("请输入密码") + dic[username] = password + print("\033[42;30m 注册成功! \033[0m") + else: + print("\033[41;37m输入错误!\033[0m") + + # 将数据写入文件 + with open("db", "w+", encoding="utf-8") as f: + for k,v in dic.items(): + data = f"{k}✨{v}\n" + f.write(data) +``` +