python连接数据库

一、python连接数据库
pyton连接数据库需要先安装pymysql模块：pip install pymysql
安装完成后导入pymysql模块：import pymysql
python连接数据库主要分五个步骤：
step1：连接数据库
step2：创建游标对象
step3：对数据库进行增删改查
step4：关闭游标
step5：关闭连接

1. # 1. 连接数据库，
2. conn = pymysql.connect(
3. host='localhost',
4. user='root',
5. password='redhat',
6. db='helloTest',
7. charset='utf8',
8. # autocommit=True, # 如果插入数据，， 是否自动提交? 和conn.commit()功能一致。
9. )
10. # ****python, 必须有一个游标对象， 用来给数据库发送sql语句， 并执行的.
11. # 2. 创建游标对象，
12. cur = conn.cursor()
13. # 3. 对于数据库进行增删改查
14. # 1). ************************创建数据表**********************************
15. try:
16. create_sqli = "create table hello (id int, name varchar(30));"
17. cur.execute(create_sqli)
18. except Exception as e:
19. print("创建数据表失败:", e)
20. else:
21. print("创建数据表成功;")
22. ## 2). *********************插入数据****************************
23. try:
24. insert_sqli = "insert into hello values(2, 'fensi');"
25. cur.execute(insert_sqli)
26. except Exception as e:
27. print("插入数据失败:", e)
28. else:
29. # 如果是插入数据， 一定要提交数据， 不然数据库中找不到要插入的数据;
30. conn.commit()
31. print("插入数据成功;")
32. # 3). *********************插入多条数据****************************
33. try:
34. info = [(i, "westos%s" %(i)) for i in range(100)]
36. # *********************第一种方式********************
37. # %s必须手动添加一个字符串， 否则就是一个变量名， 会报错.
38. insert_sqli = "insert into hello values(%d, '%s');"
39. for item in info:
40. print('insert语句:', insert_sqli %item)
41. cur.execute(insert_sqli %item)
43. # *********************第二种方式********************
44. insert_sqli = "insert into hello values(%s, %s);"
45. cur.executemany(insert_sqli, info )
46. except Exception as e:
47. print("插入多条数据失败:", e)
48. else:
49. # 如果是插入数据， 一定要提交数据， 不然数据库中找不到要插入的数据;
50. conn.commit()
51. print("插入多条数据成功;")
52. # 4). **************************数据库查询*****************************
53. sqli = "select * from hello;"
54. result = cur.execute(sqli) # 默认不返回查询结果集， 返回数据记录数。
55. print(result)
56. print(cur.fetchone()) # 1). 获取下一个查询结果集;
57. print(cur.fetchone())
58. print(cur.fetchone())
59. print(cur.fetchmany(4)) # 2). 获取制定个数个查询结果集；
60. info = cur.fetchall() # 3). 获取所有的查询结果
61. print(info)
62. print(len(info))
63. print(cur.rowcount) # 4). 返回执行sql语句影响的行数
64. # 5). 移动游标指针
65. print(cur.fetchmany(3))
66. print("正在移动指针到最开始......")
67. cur.scroll(0, 'absolute')
68. print(cur.fetchmany(3))
69. print("正在移动指针到倒数第2个......")
70. print(cur.fetchall()) # 移动到最后
71. cur.scroll(-2, mode='relative')
72. print(cur.fetchall())
73. # 4. 关闭游标
74. cur.close()
75. # 5. 关闭连接
76. conn.close()

复制代码

操作运行结果截图：
1.创建数据表：

2.插入数据;

3.插入多条数据（第一种方法）;

3.插入多条数据（第二种方法）

4.数据库查询;

5.移动游标指针

二、获取表的字段名和信息
首先导入所需要的模块

1. import time
2. import pymysql

复制代码

然后进入主程序部分：此部分用with语句编写（with语句实现的效果是: with语句执行结束， 如果成功， 则提交改变的数据， 如果不成功， 则回滚.）

1. conn = pymysql.connect(
2. host='localhost',
3. user='root',
4. password='redhat',
5. db='helloTest',
6. charset='utf8',
7. # autocommit=True, # 如果插入数据，， 是否自动提交? 和conn.commit()功能一致。
8. ) #连接数据库
9. with conn: #主程序with语句部分
10. # ****** 判断是否连接?
11. print(conn.open) # True
12. # 2. 创建游标对象，
13. cur = conn.cursor()
14. # 3).
15. sqli = "select * from hello;"
16. result = cur.execute(sqli) # 默认不返回查询结果集， 返回数据记录数。
17. # 显示每列的详细信息
18. des = cur.description
19. print("表的描述:", des)
20. # 获取表头
21. print("表头:", ",".join([item[0] for item in des]))
22. cur.close() #关闭游标，with语句内部
23. conn.close() #关闭指针，with语句外部
24. print("with语句之外:", conn.open) # 正确情况下会是False

复制代码

操作运行结果;

三、基于mysql数据库银行转账功能实现
银行转账系统考虑因素主要包括：
1). 判断两个银行卡号是否存在?
2). 判断source_id是否有足够的钱？
3). source_id扣钱
4). target_id加钱
代码如下：

1. import pymysql
2. class TransferMoney(object):
3. # 构造方法
4. def __init__(self, conn):
5. self.conn = conn
6. self.cur = conn.cursor()
7. def transfer(self, source_id, target_id, money):
8. if not self.check_account_avaialbe(source_id):
9. raise Exception("账户不存在")
10. if not self.check_account_avaialbe(target_id):
11. raise Exception("账户不存在")
12. if self.has_enough_money(source_id, money):
13. try:
14. self.reduce_money(source_id, money)
15. self.add_money(target_id, money)
16. except Exception as e:
17. print("转账失败:", e)
18. self.conn.rollback()
19. else:
20. self.conn.commit()
21. print("%s给%s转账%s金额成功" % (source_id, target_id, money))
23. def check_account_avaialbe(self, acc_id):
24. """判断帐号是否存在， 传递的参数是银行卡号的id"""
25. select_sqli = "select * from bankData where id=%d;" % (acc_id)
26. print("execute sql:", select_sqli)
27. res_count = self.cur.execute(select_sqli)
28. if res_count == 1:
29. return True
30. else:
31. # raise Exception("账户%s不存在" %(acc_id))
32. return False
34. def has_enough_money(self, acc_id, money):
35. """判断acc_id账户上金额> money"""
36. # 查找acc_id存储金额?
37. select_sqli = "select money from bankData where id=%d;" % (acc_id)
38. print("execute sql:", select_sqli)
39. self.cur.execute(select_sqli) # ((1, 500), )
40. # 获取查询到的金额钱数;
41. acc_money = self.cur.fetchone()[0]
42. # 判断
43. if acc_money >= money:
44. return True
45. else:
46. return False
48. def add_money(self, acc_id, money):
49. update_sqli = "update bankData set money=money+%d where id=%d" % (money, acc_id)
50. print("add money:", update_sqli)
51. self.cur.execute(update_sqli)
53. def reduce_money(self, acc_id, money):
54. update_sqli = "update bankData set money=money-%d where id=%d" % (money, acc_id)
55. print("reduce money:", update_sqli)
56. self.cur.execute(update_sqli)
57. # 析构方法
58. def __del__(self):
59. self.cur.close()
60. self.conn.close()
62. if __name__ == '__main__':
63. # 1. 连接数据库，
64. conn = pymysql.connect(
65. host='localhost',
66. user='root',
67. password='redhat',
68. db='helloTest',
69. charset='utf8',
70. autocommit=True, # 如果插入数据，， 是否自动提交? 和conn.commit()功能一致。
71. )
72. trans = TransferMoney(conn)
73. trans.transfer(13997, 13998, 200) #账户13997向13998转账200

复制代码

测试时，首先在数据库中构造数据表bankData,在数据表中添加用户信息（包括用户帐号和账户内金额）
如下：

运行代码后数据库内金额发生变化，由账户13997向账户13998转账200元：