python新人一月工资_python【项目】：工资管理(简易版）

1 #!/usr/bin/env python

2 #coding=utf-8

3 __author__ = 'yinjia'

4

5 importhashlib6 importsys7 from project.day2 importPayrollSystem8

9 #存储文件

10 filename = 'log.txt'

11 #账号信息表

12 result ={}13

14 defmd5(arg):15 """

16 密码进行md5加密17 :param arg: 用户的密码18 :return: 返回进行加密后的密码19 """

20 result = hashlib.md5(bytes("k1:1a2 k2:688",encoding='utf-8'))21 result.update(bytes(arg,encoding='utf-8'))22 returnresult.hexdigest()23

24

25 deflogin(username,password):26 """

27 用于验证用户密码28 :param username:用户名29 :param password:密码30 :return: True,用户名验证成功；False，用户验证失败31 """

32

33 with open(filename,"r") as f:34 for line inf:35 u,p = line.strip().split(":")36 if username == u and md5(password) ==p:37 returnTrue38 returnFalse39

40 defregister(username,password):41 """

42 用户注册43 :param username: 用户名44 :param password: 密码45 :return: True，用户注册成功；46 """

47

48 with open(filename,"a") as f:49 temp = "\n" + username + ":" +md5(password)50 f.write(temp)51 returnTrue52

53 defuser_exist(username):54 """

55 检查用户名是否存在56 :param username: 用户名57 :return: True，用户名存在； False，用户名不存在58 """

59 with open(filename,"r") as f:60 for line inf:61 line =line.strip()62 line_list = line.split(":")63 if username ==line_list[0]:64 returnTrue65 returnFalse66

67 defdelete(username):68 """

69 删除用户名70 :param username: 用户名71 :return: True,删除成功； False,删除失败72 """

73 info()74 delresult[username]75 with open(filename,'w') as f:76 for key,value inresult.items():77 f.write("%s%s%s\n" % (key,":",value))78 returnTrue79

80 defedit(username,old_password,new_password):81 """

82 修改密码83 :param username:用户名84 :param old_password:旧密码85 :param new_password:新密码86 :return:True,修改成功；False,修改失败87 """

88 info()89 ret =result[username]90 if ret ==md5(old_password):91 result[username] =md5(new_password)92 with open(filename, 'w') as f:93 for key, value inresult.items():94 f.write("%s%s%s\n" % (key, ":", value))95 returnTrue96

97 definfo():98 """

99 提取账号信息100 :return: 返回账号信息表101 """

102 with open(filename,'r') as f:103 for line inf:104 line =line.strip()105 result[line.split(":")[0]] = line.split(":")[1]106 returnresult107

108 defmain():109 info_msg = """

110 1、登录系统111 2、注册账号112 3、删除账号113 4、修改密码114 """

115

116 print('\033[32;1m')117 print('*' * 50)118 print('\t\t\t\t欢迎登录工资管理系统')119 print('*' * 50)120 print('\033[0m')121 print(info_msg)122 pingtai = str(input("\033[1;34;1m请选择操作(退出请输入q|Q)：\033[0m")).strip()123 if pingtai == "1":124 user = input("请输入用户名：")125 pwd = input("请输入密码：")126 ret =login(user,pwd)127 ifret:128 print("\033[1;31;1m<<<<<<<<登录成功>>>>>>>>>\033[0m")129 info =PayrollSystem.salary()130 info.comm()131 else:132 print("\033[1;31;1m<<<<<<<<登录失败>>>>>>>>>\033[0m")133 elif pingtai == "2":134 user = input("请输入用户名：")135 pwd = input("请输入密码：")136 ret =user_exist(user)137 ifret:138 print("\033[1;31;1m用户名已存在，不能注册！\033[0m")139 else:140 add_user =register(user,pwd)141 ifadd_user:142 print("注册成功")143 else:144 print("注册失败")145 elif pingtai == "3":146 user = input("请输入用户名：")147 del_user1 =user_exist(user)148 ifdel_user1:149 del_user2 =delete(user)150 ifdel_user2:151 print("删除成功！")152 else:153 print("删除失败！")154 else:155 print("\033[1;31;1m用户名不存在，无法删除！\033[0m")156

157 elif pingtai == "4":158 user = input("请输入用户名：")159 pas1 = input('请输入旧密码：')160 pas2 = input('请输入新密码：')161 del_user =user_exist(user)162 ifdel_user:163 edit_user =edit(user,pas1,pas2)164 ifedit_user:165 print("修改密码成功！")166 else:167 print("\033[1;31;1m旧密码错误，修改密码失败！\033[0m")168 else:169 print("\033[1;31;1m用户名不存在，无法修改！\033[0m")170 elif str(pingtai).lower() == "q":171 print("GoodBye!")172 sys.exit()173 else:174 print("\033[1;31;1m非法输入！\033[0m")175

176 if __name__ == "__main__":177 main()