ps = "123456"
count = 1
while count <= 3:
password = input("请输入密码：")
#转化布尔值，以此判断输入是否为空，下行同
password_b = bool(password)
count = count +1
if password == ps :
print("登陆成功")
def add(): # 添加数据函数
# 接收用户收入
id = input('请输入职工号：')
name = input('请输入姓名：')
sex = input('请输入性别：')
age = input('请输入年龄：')
grade = input('请输入等级（1-5）：')
money = input('请输入工资：')
# 向列表中添加数据
data.append([id, name, sex, age, grade, money])
print('添加成功')
# 调用保存函数 保存数据
save()
def show(): # 显示函数
# 打印标题
print('职工号 姓名 性别 年龄 等级 工资')
# 遍历数据列表 然后打印数据
for i in data:
for j in i:
print(j, end=' ')
print()
def quety(): # 查询函数
# 选择查询目标
id = input('输入职工编号')
# 职工编号列表
ids = [i[0] for i in data]
# 判断输入的编号是否存在
if id not in ids:
print('您查询的职工不存在')
return
else:
print('职工号 姓名 性别 年龄 学历 住址 电话 工资')
# 打印该编号的信息
for i in data[ids.index(id)]:
print(i, end=' ')
print()
def modify(): # 修改函数
# 原理同上
id = input('请输入你要修改的职工编号')
ids = [i[0] for i in data]
if id not in ids:
print('您查询的职工不存在')
return
else:
# 输入要修改的数据
name = input('请输入姓名：')
sex = input('请输入性别：')
age = input('请输入年龄：')
grade = input('请输入等级：')
money = input('请输入工资：')
# 修改数据
data[ids.index(id)] = [id, name, sex, age, grade, money]
print('修改成功')
save()
def delete(): # 删除函数
# 原理同上
id = input('请输入你要删除的职工编号')
ids = [i[0] for i in data]
if id not in ids:
print('您输入的职工不存在')
return
else:
del data[ids.index(id)]
print('删除成功')
save()
def save(): # 保存函数
# 打开文件，写入数据
with open('数据.csv','w') as j:
for i in data:
j.write(','.join(i)+'\n')
j.close()
def load(): # 读取函数
# 读取文件
with open('数据.csv','r') as j:
# 读取每行数据
for i in j.readlines():
# 清洗掉换行符 然后以逗号为间隔符分割
data.append(i.replace('\n','').split(','))
j.close()
if __name__ == '__main__':
data = [] # 数据保存列表
# 读取文件 如果文件不存在 报错跳过 无视
try:
load()
except FileNotFoundError:
pass
while True:
# 根据玩家的输入 选择相应的功能
choice = int(input('1.添加职工数据\n2.显示职工数据\n3.查询职工数据\n4.修改职工数据\n5.删除职工数据\n6.保存职工数据\n7.退出'))
if choice == 1:
add()
elif choice == 2:
show()
elif choice == 3:
quety()
elif choice == 4:
modify()
elif choice == 5:
delete()
elif choice == 6:
save()
elif choice ==7:
print('退出程序')
break
elif count == 4: #判断输入次数
print("超过三次，系统锁定")
elif password_b == False:
print("用户名或密码不能为空，请输入用户名及密码。")
else:
print("用户名及密码错误，请重新登陆！")
只是修改密码，还是修改密码后，要重新登录？