使用Python编写一个简单的用户交互

# -*- coding: utf-8 -*-
# @Author: aoao
# @Date:   2022-04-28 10:28:34
# @Last Modified by:   aoao
# @Last Modified time: 2022-04-28 16:42:44

import os
import random



path = os.getcwd()  # 获取当前目录
print("程序运行路径:%s"%(path))
def resinger():
	while True:
		user = input("设置狗蛋用户名:\n")
		password = input("设置狗蛋密码:\n")
		yzm1 = str(random.randint(100000,999999))
		yzm = input("请输入验证码:%s\n"%(yzm1))
		if user == "":
			print("狗蛋用户名不能为空")
		elif password == "":
			print("狗蛋密码不能为空")
		elif yzm == "":
			print("验证码不能为空!")
		elif yzm != yzm1:
			print("验证码错误!")
		else:
			print("狗蛋账户成功!")
			print("输出目录为:%s"%(path + "\\Desktop\\" + "user.txt"))
			with open(path + "\\Desktop\\" + "user.txt","w") as f:
				f.write(user + "\n")
				f.write(password)
			f.close()
			break

def login():
	f = open(path + "\\Desktop\\" + "user.txt","r")
	user = input("请输入用户名:\n")
	password = input("请输入密码:\n")
	lines = f.readlines()

	if lines[0].replace("\n",'') != user:
		print("用户名不存在!")
	elif lines[1].replace("\n",'') != password:
		print("密码错误!")
	else:
		print("狗蛋，欢迎回家")



def forgot():
	f = open(path + "\\Desktop\\" + "user.txt","w")
	print("忘记密码了吧，狗蛋！")

	newuser = input("请输入用户名:\n")
	newpassword = input("请输入新密码:\n")
	f.write(newuser + "\n")
	f.write(newpassword)
	print("设置成功!")
	


print("=============好吃狗系统===============")
print("[1]注册用户")
print("[2]登陆系统")
print("[3]忘记密码")
print("请输入数字继续操作")
print("=====================================")
work = int(input("请输入数字:\n"))
if work == 1:
	resinger()
elif work == 2:
	login()
elif work == 3:
	forgot()
else:
	print('输入有误！')

简单易懂，代码有注释，大家可以参考我刚开始写的幼儿班代码