Python中的MySQL操作

Posted 2021-03-23 yuandian43

1.Python中导入MySQL库

在Python2与Python3中导入的MySQLdb库是不一样的，在Python2安装的是mysql-python，Python3安装的是mysqlclient，分别安装后就可以import MySQLdb。

在Python3中安装MySQLdb，可参考以下方式：

• sudo apt-get install python-dev libmysqlclient-dev
• sudo apt-get install python3-dev
• pip install mysqlclient

如果环境中已有python-dev python3-dev libmysqlclient-dev，需要在脚本中视mysqlclient是否存来安装mysqlclient，可以在py脚本中用以下方式：

#!/usr/bin/env python3
# -*- coding: utf8 -*-
import os
try:
    import MySQLdb
except ImportError:
    os.system(‘pip --proxy=http://xxxxxxxx:8080 install mysqlclient‘)
    import MySQLdb

2.链接MySQL，执行MySQL命令

整个过程大致分为链接数据库，获取游标，执行命令，获取返回结果，关闭数据库几部分，如下所示：

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import MySQLdb
db = MySQLdb.connect("localhost", "dbUser", "dbPasswd", "dbName", charset=‘utf8‘ )  # 打开数据库连接
cursor = db.cursor()  # 使用cursor()方法获取操作游标 
cursor.execute("SELECT VERSION()")  # 使用execute方法执行SQL语句
data = cursor.fetchone()  # 使用 fetchone() 方法获取一条数据
print "Database version : %s " % data
db.close() # 关闭数据库连接

对于不同类型的待执行sql命令，中间步骤有些许不同

1)创建表或删除表，需要执行cursor.execute(sql)命令，没有啥返回数据

2)插入数据，需要执行cursor.execute(sql)命令，然后还需要db.commit()将执行提交到数据库，如下所示：

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import MySQLdb
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset=‘utf8‘ )  # 打开数据库连接
cursor = db.cursor()  # 使用cursor()方法获取操作游标 
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES (‘Mac‘, ‘Mohan‘, 20, ‘M‘, 2000)"""   # SQL 插入语句
try:
   cursor.execute(sql)    # 执行sql语句
   db.commit()    # 提交到数据库执行
except:
   db.rollback()    # 一旦发生错误回滚数据
db.close()  # 关闭数据库连接

3)查询/更新/删除数据，需要执行cursor.execute(sql)命令，然后还需要使用cursor.fetchall()方法获取查询到的数据，如下所示：

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import MySQLdb
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset=‘utf8‘ )  # 打开数据库连接
cursor = db.cursor()  # 使用cursor()方法获取操作游标 
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES (‘Mac‘, ‘Mohan‘, 20, ‘M‘, 2000)"""   # SQL 插入语句
try:
   cursor.execute(sql)    # 执行sql语句
   db.commit()    # 提交到数据库执行
except:
   db.rollback()    # 一旦发生错误回滚数据
db.close()  # 关闭数据库连接 

也可以通过cursor.rowcount获取执行execute()方法后影响的行数。

获取查询mysql的函数有fetchone() fetchall() fetchmany()，区别如下：

cur.fetchone():返回单个的元组，也就是一条记录(row)，如果没有结果则返回 None

cur.execute("select host,user,password from user where user=‘%s‘" %acc)
jilu = cur.fetchone()  ##通过 jilu[0],jilu[1],jilu[2]可以依次访问host,user,password 

cur.fetchall()：返回二维元组，即返回多个记录(rows),如果没有结果则返回 ()

curl.fetchmany(n):获取前n行数据row_2=cur.fetchmany(n)，返回值为二维数组，如果没有结果则返回()。

举栗：

如果select本身取的时候有多条数据时：cur.execute("select id,title from user")

cursor.fetchone()：将只取最上面的第一条结果，返回单个元组如(‘id‘,‘title‘)，然后多次使用cursor.fetchone()，依次取得下一条结果，直到为空。

cursor.fetchall() :将返回所有结果，返回二维元组，如((‘id‘,‘title‘),(‘id‘,‘title‘))。

如果select本身取的时候只有一条数据时：cur.execute("select id,title from user limit 1")

cursor.fetchone()：将只返回一条结果，返回单个元组如(‘id‘,‘title‘)。

cursor.fetchall() :也将返回所有结果，返回二维元组，如((‘id‘,‘title‘),),

综合起来讲，fetchall返回二维元组（元组中含有元组），fetchone只返回一维元组。

需要注明：在MySQL中是NULL，而在Python中则是None

参考：https://www.runoob.com/python/python-mysql.html