阿里云-云小站(无限量代金券发放中)
【腾讯云】云服务器、云数据库、COS、CDN、短信等热卖云产品特惠抢购

Python3操作MySQL数据库

209次阅读
没有评论

共计 4090 个字符,预计需要花费 11 分钟才能阅读完成。

这是 P ython3 下 的 MySQL 基本操作。其他类型的数据库用法基本一样。就是库的名字不同。因为 Python 官方很早之前就规定了数据库第三方库的借口,来避免 API 混乱的情况。

安装与准备

这是 python3 的库,所以 windows 下安装不会像 python2 那样各种奇葩 VC 错误。是比较方便的傻瓜安装。

  • Windows 平台下: py -3 -m pip install PyMySQL
  • Linux:python3 pip install PyMySQL

当然,引入的时候:import pymysql


数据库连接对象 connection

Function 描述
connection 创建 connection 对象
cursor() 使用该链接创建 + 返回游标
commit() 提交当前事务
rollback() 回滚当前十五
close() 关闭连接

介绍一下 connection 的参数

  1. host mysql 服务器地址
  2. port 数字类型 端口
  3. user 用户名
  4. passwd 密码
  5. db 数据库名称
  6. charset 连接编码,需要显式指明编码方式

上一段代码示例:

conn = pymysql.Connect(host='127.0.0.1',port=3306,user='root',passwd='dyx240030',db='imooc',charset='utf8')
cursor = conn.cursor()
print(conn)
print(cursor)
cursor.close()
conn.close()
OUT:
<pymysql.connections.Connection object at 0x00000051C15BFDA0>
<pymysql.cursors.Cursor object at 0x00000051C15BFD68>

数据库游标对象 cursor

Function 描述
execute(op[,args]) 执行一个数据库查询和命令
fetchone() 取得结果集下一行
fetchmany(size) 取得结果集 size 行
fetchall() 取得结果集剩下所有行
rowcount 最近一次 execute 返回数据的行数或影响行数
close() 关闭 cursor

代码实现:

conn = pymysql.Connect(host='127.0.0.1',port=3306,user='root',passwd='dyx240030',db='imooc',charset='utf8')
cursor = conn.cursor()
sql = "select * from user"
cursor.execute(sql)
print("cursor.excute:",cursor.rowcount)

rs = cursor.fetchone()
print("rs:",rs)

for each in cursor.fetchmany(2):
    print(each)
print()
for each in cursor.fetchall():
    print(each)
OUT:
cursor.excute: 4
rs: ('1', 'name1')
('2', 'name2')
('3', 'name3')

('4', 'name4')

更新数据库 insert/update/delete

不同于 select 操作,这三个操作修改了数据库内容,所以需要commit(),否则数据库没有做相应的更改,但是也不会报错。

按照一般的思路,一般是以下套路:

  1. 关闭自动 commit:conn.autocommit(False)
  2. 出现:conn.rowback()回滚
  3. 未出现:conn.commit()

下面这段脚本,实现 insert/update/delete 操作。其实这种检错模式不对,这里只做简单 raise,后面有更好的方法。

conn = pymysql.Connect(host='127.0.0.1',port=3306,user='root',passwd='dyx240030',db='imooc',charset='utf8')
conn.autocommit(False)
cursor = conn.cursor()

sqlInsert = "insert into user(userid,username) values('6','name6')"
sqlUpdate = "update user set username='name41'where userd='4'"
sqlDelete = "delete from user where userid='1'"
try:
    cursor.execute(sqlInsert)
    print(cursor.rowcount)
    cursor.execute(sqlUpdate)
    print(cursor.rowcount)
    cursor.execute(sqlDelete)
    print(cursor.rowcount)
    
    conn.commit()
except Exception as e:
    print("Reason:",e)
    conn.rollback()
    
cursor.close()
cursor.close()
[OUT]:
1
Reason: (1054, "Unknown column'userd'in'where clause'")

实例 银行转账

可以看一下类思想的 SQL 操作,其中之前提到过的高级报错模式用到了之前看似无用的 rowcount 函数,通过查看操作对于数据库的影响来检错。

import os
import sys
import pymysql

class transferMoney(object):
    def__init__(self,conn):
        self.conn = conn
    def transfer(self,sourceID,targetID,money):
        #   其他函数中若是有错会抛出异常而被检测到。
        try:
            self.checkIdAvailable(sourceID)
            self.checkIdAvailable(targetID)
            self.ifEnoughMoney(sourceID,money)
            self.reduceMoney(sourceID,money)
            self.addMoney(targetID,money)
            self.conn.commit()
        except Exception as e:
            self.conn.rollback()
            raise e
    def checkIdAvailable(self,ID):
        cursor = self.conn.cursor()
        try:
            sql = "select * from account where id = %d" % ID #select 语句判断可以用 len(rs)
            cursor.execute(sql)
            rs=  cursor.fetchall()
            if len(rs) != 1:#   数据库类思想的报错模式,检查操作对数据库的影响条目。没有达到目标,抛出异常
                raise Exception("账号 %d 不存在"%ID)
        finally:
            cursor.close()
        
    def ifEnoughMoney(self,ID,money):
        cursor = self.conn.cursor()
        try:
            sql = "select * from account where id = %d and money>=%d" % (ID,money)
            cursor.execute(sql)
            rs=  cursor.fetchall()
            if len(rs) != 1:
                raise Exception("账号 %d 不存在 %d Yuan"%(ID,money))
        finally:
            cursor.close()
    
    def reduceMoney(self,ID,money):
        cursor = self.conn.cursor()
        try:
            sql = "update account set money = money-%d where id=%d"%(money,ID)
            cursor.execute(sql)
            if cursor.rowcount != 1:
                raise Exception("失败减钱")
        finally:
            cursor.close()
    
    def addMoney(self,ID,money):
        cursor = self.conn.cursor()
        try:
            sql = "update account set money = money+%d where id=%d"%(money,ID)
            cursor.execute(sql)
            if cursor.rowcount != 1:
                raise Exception("失败加款")
        finally:
            cursor.close()
    

if __name__=="__main__":
    if len(sys.argv)>=2:
        sourceID = int(sys.argv[1])
        targetID = int(sys.argv[2])
        money = int(sys.argv[3])

        conn = pymysql.Connect(host='127.0.0.1',port=3306,user='root',passwd='dyx240030',db='imooc',charset='utf8')
        trMoney = transferMoney(conn)

        try:
            trMoney.transfer(sourceID,targetID,money)
        except  Exception as e:
            print("出现问题"+str(e))
        finally:
            conn.close()

我踩过的坑。。。(这里是 Python3 哦)

‘NoneType’ object has no attribute ‘encoding’,之前指明的 charset 必须是 ”UTF8″, 不是 ”utf-8″/”UTF-8″
MySQL 语句后面必须有 ’;’,否则不会报错,也难以发现
数据库 insert/update/delete 操作需要 commit()
在构造命令的时候,注意用 ” 包裹起来,因为 SQL 语句字符串需要 ‘ 包裹。所以,” 比较简单的避免歧义。

本文永久更新链接地址:http://www.linuxidc.com/Linux/2017-06/145155.htm

正文完
星哥玩云-微信公众号
post-qrcode
 0
星锅
版权声明:本站原创文章,由 星锅 于2022-01-22发表,共计4090字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
【腾讯云】推广者专属福利,新客户无门槛领取总价值高达2860元代金券,每种代金券限量500张,先到先得。
阿里云-最新活动爆款每日限量供应
评论(没有评论)
验证码
【腾讯云】云服务器、云数据库、COS、CDN、短信等云产品特惠热卖中