数据库编程
PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库,Python2中则使用MySQLdb。利用这个库可直接对数据库进行相应的操作。
下面是用PyMySQL对数据库进行操作的常规步骤:
PyMySQL基本操作语句:
下面我们就用Python代码实现上面的操作,首先在数据库中建立一个名为“pymysql”的数据库,我们之后都在该数据库下进行操作。
1.先创建一个名为“students”的表,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3. sql操作
# 创建数据库表的操作
create_sqli = 'create table students(id int, name varchar(20), age int, score float);'
# 执行sql语句
cursor.execute(create_sqli)
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
在数据库中可以查看新创建的表:
2.在表中添加记录,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3. 执行添加记录的操作
insert_sqli = 'insert into students(name, score) values("小E", 78);'
# 执行sql语句
cursor.execute(insert_sqli)
# 提交sql操作
conn.commit()
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
在表中查看添加的记录:
3.批量添加记录操作,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3.sql操作
import random
insert_sqli = 'insert into students(name, score) values("%s", "%s");'
for i in range(10):
name = "学生" + str(i + 1)
score = random.randint(50, 100)
sql = insert_sqli % (name, score)
print(sql)
cursor.execute(sql)
# 提交sql操作
conn.commit()
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
在表中查看批量添加的记录:
4.还有一种批量添加记录的方法,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3.sql操作
insert_sqli = 'insert into students(name, score) values(%s, %s);'
cursor.executemany(insert_sqli, [('A', 100), ('D', 90), ('E', 99)])
# 提交sql操作
conn.commit()
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
查看添加的记录:
数据库查询操作
下面用Python代码实现上面操作:
1.查询数据,若执行查询语句的话,只会返回一个值,这个值代表影响的行数,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3. 查询数据
select_sqli = 'select * from students'
affectedRows = cursor.execute(select_sqli)
print(affectedRows)
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
输出结果如下:
2.正确的查询操作,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3. 查询数据
select_sqli = 'select * from students'
affectedRows = cursor.execute(select_sqli)
print(affectedRows)
# 3-1. 获取查询的数据
print(cursor.fetchone())
# 将游标移动到记录的最开始位置
cursor.scroll(0, mode='absolute')
print(cursor.fetchmany(2))
# 将游标移动到当前记录的下一个记录的位置;
cursor.scroll(1)
print(cursor.fetchall())
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
查询到的输出结果如下:
3.还可以根据条件查询,若查询数据存在就返回1,否则返回0,代码如下:
import pymysql
# 1. 与mysql建立连接 mysql -h host/ip -u root -p westos
conn = pymysql.connect(host='172.25.254.35', user='root', password='westos', db='pymysql')
# 2. 创建游标
cursor = conn.cursor()
# 3. 1) 查询数据: 如果查询到就返回1;
select_sqli = 'select * from students where name="A"'
affectedRows = cursor.execute(select_sqli)
print(affectedRows)
# 3. 2) 查询数据: 如果没有查询到就返回0;
select_sqli = 'select * from students where name="B"'
affectedRows = cursor.execute(select_sqli)
print(affectedRows)
# 4. 关闭游标
cursor.close()
# 5.关闭连接
conn.close()
输出结果如下:
项目案例: 银行转账系统实现
我们通过一个实际的项目案例,对上面的PyMySQL操作进行一个总的应用。
项目描述:实现一个银行转账系统,使其连接数据库,当执行转账操作时,先判断两个账户是否存在,再判断要转账的账户金额是否足够,若满足上述条件,则对转账账户的金额执行减的操作,并且要保证同时对收帐账户的金额执行加操作。
下面是Python代码:
import pymysql
class TransferMoney(object):
# 构造方法
def __init__(self, conn):
self.conn = conn
self.cur = conn.cursor()
def transfer(self, source_id, target_id, money):
# 1). 判断两个银行卡号是否存在?
# 2). 判断source_id是否有足够的钱?
# 3). source_id扣钱
# 4). target_id加钱
if not self.check_account_avaialbe(source_id):
raise Exception("账户不存在")
if not self.check_account_avaialbe(target_id):
raise Exception("账户不存在")
if self.has_enough_money(source_id, money):
try:
self.reduce_money(source_id, money)
self.add_money(target_id, money)
except Exception as e:
print("转账失败:", e)
self.conn.rollback()
else:
self.conn.commit()
print("%s给%s转账%s金额成功" % (source_id, target_id, money))
else:
print("没有足够的金额")
def check_account_avaialbe(self, acc_id):
"""判断帐号是否存在, 传递的参数是银行卡号的id"""
select_sqli = "select * from bank where cardid=%s;" % (acc_id)
print("execute sql:", select_sqli)
res_count = self.cur.execute(select_sqli)
if res_count == 1:
return True
else:
# raise Exception("账户%s不存在" %(acc_id))
return False
def has_enough_money(self, acc_id, money):
"""判断acc_id账户上金额> money"""
# 查找acc_id存储金额?
select_sqli = "select money from bank where cardid=%s;" % (acc_id)
print("execute sql:", select_sqli)
self.cur.execute(select_sqli) # ((1, 500), )
# 获取查询到的金额钱数;
acc_money = self.cur.fetchone()[0]
# 判断
if acc_money >= money:
return True
else:
return False
def add_money(self, acc_id, money):
update_sqli = "update bank set money=money+%d where cardid=%s" % (money, acc_id)
print("add money:", update_sqli)
self.cur.execute(update_sqli)
def reduce_money(self, acc_id, money):
update_sqli = "update bank set money=money-%d where cardid=%s" % (money, acc_id)
print("reduce money:", update_sqli)
self.cur.execute(update_sqli)
# 析构方法
def __del__(self):
self.cur.close()
self.conn.close()
测试代码如下:
if __name__ == '__main__':
# 连接数据库,
conn = pymysql.connect(
host='172.25.254.35',
user='root',
password='westos',
db='pymysql',
charset='utf8',
autocommit=True, # 如果插入数据,, 是否自动提交? 和conn.commit()功能一致。
)
trans = TransferMoney(conn)
trans.transfer('130001', '130002', 300)
现在数据库“bank”中创建两个账户,它们初始状态如下:
程序运行之后,输出如下:
并再次对数据库进行查看,结果如下:
可见,银行转账系统基本成功实现。