python第一次连接mysql数据库

Posted 2021-01-09 dave555

文件结构图

 

源码

jing_dong.py

 1 # coding:utf-8
 2 from mysql_connect import MySql
 3 
 4 
 5 class JD(object):
 6     def __init__(self):
 7         self.mysql = MySql(‘localhost‘, 3306, ‘root‘, ‘‘, ‘ksoa_tdyy_test‘, ‘utf8‘)
 8 
 9     def show_items_info(self):
10         """查询商品信息"""
11         item_name = input("请输入要查询的商品名称：")
12         sql = """select * from spkfk where spmch=%s or tongym=%s"""
13         for temp in self.mysql.sql_select_all_data(sql, item_name, item_name):
14             print(temp)
15 
16     def show_order_info(self):
17         """查询订单信息"""
18         item_name = input("请输入要查询的订单号：")
19         sql = """select * from pf_djhz where djbh=%s"""
20         for temp in self.mysql.sql_select_all_data(sql, item_name):
21             print(temp)
22 
23     def show_customer_info(self):
24         """查询顾客信息"""
25         item_name = input("请输入要查询的顾客名称：")
26         sql = """select * from mchk where dwmch=%s"""
27         for temp in self.mysql.sql_select_all_data(sql, item_name):
28             print(temp)
29 
30     def add_items(self):
31         pass
32 
33     # 创建静态方法
34     @staticmethod
35     def print_menu():
36         """打印菜单"""
37         print("------京东商城--------")
38         print("1:查询商品信息")
39         print("2:查询订单信息")
40         print("3:查询顾客信息")
41         print("4.添加商品信息")
42         print("0:退出")
43         return input("请输入功能对应的序号：")
44 
45     def run(self):
46         while True:
47             user_option = self.print_menu()
48             if user_option == "1":
49                 # 查询商品信息
50                 self.show_items_info()
51             elif user_option == "2":
52                 # 查询订单信息
53                 self.show_order_info()
54             elif user_option == "3":
55                 # 查询顾客信息
56                 self.show_customer_info()
57             elif user_option == "4":
58                 # 添加商品信息
59                 self.add_items()
60             elif user_option == "0":
61                 # 退出
62                 print("谢谢使用")
63                 break
64             else:
65                 print("输入有误，请重新输入")
66                 continue
67 
68 
69 def main():
70     # 1.创建一个京东商城对象
71     jd = JD()
72 
73     # 2.调用这个对象的run()方法，让其运行
74     jd.run()
75 
76 
77 if __name__ == "__main__":
78     main()

 

mysql_connect.py

 1 # coding:utf-8
 2 from pymysql import connect
 3 
 4 
 5 class MySql(object):
 6     not_select_data_info = ("提示,", "没有查询到数据")
 7 
 8     def __init__(self, host, port, user, password, database, charset):
 9         # 创建connect连接
10         self.conn = connect(host=host, port=port, user=user, password=password, database=database, charset=charset)
11         # 获得Cursor对象
12         self.cursor = self.conn.cursor()
13 
14     def __del__(self):
15         # 关闭Cursor对象
16         self.cursor.close()
17         # 关闭connect对象
18         self.conn.close()
19 
20     def sql_select_all_data(self, sql_str, *args):
21         """查询所有数据，返回元组"""
22         """
23         如果sql里面需要有用户输入的参数传入，这时不要自己手工拼接字符串
24         以元组的方式将多个参数(*args)和sql语句传给execute，让其自己拼接
25         用来防止SQL注入
26         """
27         if self.cursor.execute(sql_str, args) != 0:
28             # 返回全部数据
29             return self.cursor.fetchall()
30         # 返回提示信息
31         return self.not_select_data_info
32 
33     def sql_select_one_data(self, sql_str, *args):
34         """查询一行数据，返回元组"""
35         if self.cursor.execute(sql_str, args) != 0:
36             # 返回全部数据
37             return self.cursor.fetchone()
38         # 返回提示信息
39         return self.not_select_data_info
40 
41     def sql_change_data(self, sql_str, *args):
42         """数据的增、删、改操作，返回影响的行数"""
43         change_count = self.cursor.execute(sql_str, args)
44         if change_count != 0:
45             # 返回影响的行数
46             return change_count
47         # 返回0
48         return 0

 

运行效果图

未对查询出来的数据做显示优化处理。