python操作数据库，实现用户名密码登录数据库，首次登录自行设置密码，并返回工资表明细。

Posted 2021-01-16 cooldsj

python操作数据库，实现用户名、密码登录数据库，首次登录自行设置密码，并返回工资表明细。

 1 #!/usr/bin/env python3
 2 # -*- coding: utf-8 -*-
 3 
 4 # 导入依赖包
 5 import psycopg2
 6 
 7 print("营口烟草工资查询系统。授权辽宁省烟草公司营口市公司人事科使用。")
 8 print("注意！！！，本系统仅供查询当月工资，每月8日更新数据，8日后可查询当月工资。")
 9 print("开发者：CoolDSJ")
10 print()
11 
12 
13 def showdate(username):
14     """读取数据库工资条信息，根据用户名读取不同的记录。"""
15     # conn = psycopg2.connect(database="shop", user="postgres", password="*******", host="localhost", port="5432")
16     conn = psycopg2.connect(database="postgres", user="postgres", password="********", host="********9", port="5432")
17     cur = conn.cursor()
18     cur.execute("select * from gongzi where ygbh=‘%s‘" % username)
19     results = cur.fetchone()
20     cur.close()
21     conn.close()
22     return results
23 
24 
25 def gaimm(new_mima,username):
26     """第一次使用的用户，提醒更改密码，将新密码存入数据库的用户名对应密码列中。"""
27     # conn = psycopg2.connect(database="shop", user="postgres", password="*******", host="localhost", port="5432")
28     conn = psycopg2.connect(database="postgres", user="postgres", password="*******", host="********", port="5432")
29     cur = conn.cursor()
30     cur.execute("update gongzi set mm =‘%s‘ where ygbh=‘%s‘"%(new_mima,username))  #新密码写入数据库对应列。
31     conn.commit()  #修改数据库后，必须用该语句提交修改。
32     cur.close()
33     conn.close()
34 
35 
36 def val():
37     name = input(‘请输入用户名，并按回车键。>>>>>>‘)  # 输入用户名
38     results = showdate(name)  # 获取用户名对应的数据库资料
39     if results:
40         mima = input(‘请输入密码，并按回车键。>>>>>>‘)  # 输入用户名
41         if str(results[11]) == ‘nnnn‘:
42             mima = input(‘您是第一次使用系统，请设定一个4位数字密码，并按回车键。>>>>>>‘)
43             if len(mima) != 4:
44                 print("请输入4位数字！")
45                 mima = input(‘请设定一个4位数字密码，并按回车键。>>>>>>‘)
46                 if len(mima) == 4 :
47                     xiugai = gaimm(mima, name)  #调用写入数据库密码函数。
48                     print("密码修改成功。")
49                     val()
50                 else:
51                     print("未能成功修改密码,请重新操作！！！。")
52                     val()
53             else:
54                 xiugai = gaimm(mima, name)   #调用写入数据库密码函数。
55                 print("密码修改成功。")
56                 val()
57         elif results[11] == mima:
58             print()
59             print(results[1], "同志，您好：")
60             print("    您2018年11月份工资明细如下：")
61             print()
62             print("  项目名称    项目内容")
63             print(" ----------  ------------")
64             print("‖姓    名‖", results[0])
65             print("‖姓    名‖", results[0])
66             print("‖岗位工资‖", results[1])
67             print("‖绩效工资‖", results[2])
68             print("‖绩效考评‖", results[3])
69             print("‖工资合计‖", results[4])
70             print("‖公 积 金‖", results[5])
71             print("‖养老保险‖", results[6])
72             print("‖医疗保险‖", results[7])
73             print("‖失业保险‖", results[8])
74             print("‖所 得 税‖", results[9])
75             print("‖企业年金‖", results[10])
76             print("‖实发工资‖", results[11])
77             print()
78             print("以上工资，如与实际不符，或有其它疑问，请及时与市局（公司）人事科部沟通。")
79             print("祝工作顺利，身体健康。")
80         else:
81             print("密码错误")
82     else:
83         print("用户名错误")
84 
85 
86 val()
87 input()