python操作MySQL数据库增删改查使用MySQLdb

最新推荐文章于 2022-06-20 18:11:00 发布

CN-AllenRen

最新推荐文章于 2022-06-20 18:11:00 发布

阅读量2k

点赞数 1

文章标签： python mysql 数据库存储

本文链接：https://blog.csdn.net/rkun595/article/details/62236637

版权

在操作之前，我们来看一下 python 是怎么连接 MySQL 数据库的。
首先我们需要下载必要的包 mysql-python，下载地址为：
https://sourceforge.net/projects/mysql-python/files/ 安装的过程不在赘述。

欢迎访问我的个人网站：http://www.cjluzzl.cn

安装完成后打开终端，进入 python 环境，输入 import MySQLdb 没有报错证明安装成功。

安装好必要的包我们来介绍一下连接数据库的原理。
首先打开数据库，创建一个 user_table 表
CREATE TABLE `user_table` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(255) NOT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=10 DEFAULT CHARSET=utf8;
这里要注意设置存储引擎为 InnoDB，因为这个引擎支持事务。
然后随便添加几条数据。

然后开始测试

import MySQLdb
con = MySQLdb.connect(host='127.0.0.1',
                      port = 3306,
                      user = 'root',
                      passwd = '142857',
                      db = 'user',
                      charset='utf8')

cursor = con.cursor()

sql = 'select * from user_table'

cursor.execute(sql)

info = cursor.fetchall()

for i in info:
    print i
    
print cursor.rowcount

cursor.close()
con.close()

运行结果如下

从运行结果可以看出，python是以元组的形式返回的数据库中的记录。执行完execute()影响的行数是9条。

总结一下连接数据库的步骤

connection()创建一个连接对象，connect()函数的参数说明如下：

参数名	类型	说明
host	字符串	服务器地址
port	数字	服务器端口号
user	字符串	用户名
passwd	字符串	密码
db	字符串	数据库名称
charset	字符串	连接编码

2. cursor = con.cursor()创建一个游标对象，用于查询和获取结果，cursor的方法有：

参数名	说明
execute(sql[,args])	执行一条SQL语句
fetchone()	取结果集的下一行
fetchmany(size)	取结果集的下几行
fetchall()	获取结果集中剩下的所有行
rowcount	最近一次执行execute返回的数据行数或影响行数
commit()	提交事务
rollback()	回滚事务，恢复数据到修改之前
close()	关闭游标对象

测试增删改查操作

import MySQLdb

conn = MySQLdb.connect(
                       host = '127.0.0.1',
                       port = 3306,
                       user = 'root',
                       passwd = '142857',
                       db = 'user',
                       charset = 'utf8'
                       )

cursor = conn.cursor()
sql_insert = 'insert into user_table(`id`,`name`) values(10,"name10")'
sql_update = 'update user_table set `name`="name_07" where id =7'
sql_delete = "delete from user_table where id<3 "


try:
    cursor.execute(sql_insert)
    print cursor.rowcount
    
    cursor.execute(sql_update)
    print cursor.rowcount
    
    cursor.execute(sql_delete)
    print cursor.rowcount
    
    conn.commit()
except Exception as e:
    print e
    conn.rollback()

cursor.close()
conn.close()

对比一下运行前后的数据库状态

运行前：运行后：

这里特别要注意的是sql语句使用execute()方法执行后，要使用commit()方法提交处理，当事务失败后，要调用rollback()方法进行事务的回滚。注意尽量避免使用不支持事务的存储引擎。

使用python操作数据库模拟银行转账的功能

import sys
import MySQLdb

class TransferMoney():
    def __init__(self,conn):
        self.conn = conn
    

    def check_acct_available(self, acctid):
        cursor = self.conn.cursor()
        try:
            sql = 'select * from account where acctid=%s' % acctid
            cursor.execute(sql)
            print 'check_acct_available:' + sql
            rs = cursor.fetchall()
            if len(rs) != 1:
                raise Exception("账号%s不存在" % acctid)
        finally:
            cursor.close()
    
    def has_enough_money(self, acctid, money):
        cursor = self.conn.cursor()
        try:
            sql = 'select * from account where acctid=%s and money > %s' % (acctid,money)
            cursor.execute(sql)
            print 'has_enough_money:' + sql
            rs = cursor.fetchall()
            if len(rs) != 1:
                raise Exception("账号%s没有足够的钱" % acctid)
        finally:
            cursor.close()
    
    
    def reduce_money(self, acctid, money):
        cursor = self.conn.cursor()
        try:
            sql = 'update account set money=money-%s where acctid=%s' % (money,acctid)
            cursor.execute(sql)
            print 'reduce_money:' + sql
            if cursor.rowcount != 1:
                raise Exception("账号%s减款失败" % acctid)
        finally:
            cursor.close()
    
    
    def add_money(self, acctid, money):
        cursor = self.conn.cursor()
        try:
            sql = 'update account set money=money+%s where acctid=%s' % (money,acctid)
            cursor.execute(sql)
            print 'add_money:' + sql
            if cursor.rowcount != 1:
                raise Exception("账号%s加款失败" % acctid)
        finally:
            cursor.close()
    
    
    def transfer(self,source_acctid,target_acctid,money):
        try:
            self.check_acct_available(source_acctid)
            self.check_acct_available(target_acctid)
            self.has_enough_money(source_acctid,money)
            self.reduce_money(source_acctid,money)
            self.add_money(target_acctid,money)
            self.conn.commit()
        except Exception as e:
            self.conn.rollback()
            raise e
        
        
if __name__ == "__main__":
    source_acctid = sys.argv[1]
    target_acctid = sys.argv[2]
    money = sys.argv[3]
    print source_acctid,' ',target_acctid,' ',money
    conn = MySQLdb.connect(host='127.0.0.1',port = 3306,user = 'root',passwd='142857',db ='user',charset = 'utf8')
    tr_money = TransferMoney(conn)
    try:
        tr_money.transfer(source_acctid,target_acctid,money)
    except Exception as e:
        print u'出现问题',e
    finally:
        conn.close()