你这段代码有几个问题导致注册成功之后直接退出了循环。让我给你详细解释一下:
1. **变量名冲突**:在 `register` 函数中,`count` 被重新赋值为一个字符串,而不是最初的字典。这会导致 `login` 函数无法正常工作,因为它期望 `count` 是一个字典。
2. **函数调用结束后没有返回到主循环**:在 `register` 和 `login` 函数中,一旦执行完成,程序没有回到主循环(`get_int` 函数)继续执行,而是直接退出。
我们需要对代码进行一些修改,使其能够正确工作:
import hashlib
print('欢迎来到鱼C论坛')
print('====================')
count = {}
def register():
user = input('请输入用户名:')
if user in count:
print('该用户已注册!')
return
else:
keys = input('请输入密码:')
result = hashlib.md5(bytes(keys, 'utf-8'))
count[user] = result.hexdigest() # 保存用户名和密码的哈希值到字典中
print('恭喜,注册成功!')
print('====================')
def login():
user = input('请输入用户名:')
while True:
if user not in count:
print('该用户名不存在。')
user = input('请重新输入用户名:')
continue
else:
break
keys = input('请输入密码:')
result = hashlib.md5(bytes(keys, 'utf-8'))
while True:
if result.hexdigest() == count[user]: # 比较输入密码的哈希值与保存的哈希值
print('恭喜,登陆成功!')
print('====================')
break
else:
print('密码错误!')
keys = input('请重新输入密码:')
result = hashlib.md5(bytes(keys, 'utf-8'))
continue
def get_int():
while True:
print('1.注册')
print('2.登录')
print('3.退出')
command = input('请输入指令:')
print('====================')
if int(command) == 1:
register()
elif int(command) == 2:
login()
elif int(command) == 3:
break
get_int()
### 修改要点:
1. **保留 `count` 作为字典**:在 `register` 函数中,将用户名和密码的哈希值存入 `count` 字典中。
2. **不改变 `count` 的类型**:确保 `count` 始终是一个字典,避免后续的 `login` 函数出错。
3. **返回到主循环**:在 `register` 和 `login` 函数执行完毕后,返回到主循环,让用户可以继续选择操作。
这样修改后,程序在注册成功或登录成功后会返回到主循环,允许用户继续选择下一步操作。
以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。 |