python emo_utils_python接口自动化（4）操作mysql

操作mysql

安装pymysql:pip install pymysql

常用对象：

Connection：代表一个与MySQL Server的socket连接，使用connect方法来创建一个连接实例。

Cursor：代表一个与MySQL数据库交互对象，使用Connection.Cursor()在当前socket连接上的交互对象。

Connection对象常用的API:

connect() 创建一个数据库连接实例

begin() 开始一个事务

close() 发送一个退出消息，并关闭连接

commit() 提交修改至数据库

cursor() 创建一个cursor(游标)实例

ping() 检测服务器是否在运行

rollback() 回滚当前事务

select_db(db) 设置当前db

show_warnings() 显示警告信息

Cursor对象常用API：

close() 关闭当前cursor

execute() 执行一个sql语句

executemany() 执行批量sql语句

fetchall() 取所有数据

fetchone() 取一条数据

插入、查询、删除操作

新建database：

create database pymysql;

use pymysql;

create table users(

id int primary key,

name varchar(10)

)

代码：

解释：

cursor.fetchone()，获取一行数据，为元组类型

cursor.fetchall()，获取所有行的数据，为元组类型，元组里面包含元组，一个元组代表一样数据

import pymysql

import random

class MysqlDemo:

def __init__(self):

pass

def get_conn(self):

conn=pymysql.connect(

host="192.168.239.137",

port=3306,

user="root",

password="krystal",

db="pymysql",

charset="utf8"

)

return conn

if __name__=='__main__':

#获取cursor游标

conn=MysqlDemo().get_conn()

cursor=conn.cursor()

#插入数据：

sql1='insert into `users`(id,name) values(1,"jimmy"),(2,"krystal")'

cursor.execute(sql1)

#插入多条数据：

sql2='insert into `users`(id,name) values(%s,%s)'

sql2_data=[]

for i in range(3,10):

name="".join(random.sample('asdfghjklzxcvbnmqwertuyiop',5))

id=i

sql2_data.append((i,name)) # 封装为元组，并追加到列表中

print(sql2_data)

cursor.executemany(sql2,sql2_data)

#查询数据

sql3='select * from users limit 3'

cursor.execute(sql3)

data1=cursor.fetchone() #获取一条数据,为元组类型

data2=cursor.fetchall() #获取所有数据，为元组类型，元组包含元组

print("name:{},id:{}".format(data1[0],data1[1]))

for row in data2:

name=row[0]

id=row[1]

print("name:{},id:{}".format(name, id))

conn.commit()

cursor.close()

conn.close()

别人的mysql工具类封装

import pymysql

from warnings import filterwarnings

# 忽略Mysql告警信息

filterwarnings("ignore", category=pymysql.Warning)

class MysqlDb:

def __init__(self):

# 建立数据库连接

self.conn = pymysql.connect("192.168.239.137", "root", "krystal", "pymysql")

# 使用 cursor 方法获取操作游标，得到一个可以执行sql语句，并且操作结果作为字典返回的游标

self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

def __del__(self):

# 关闭游标

self.cur.close()

# 关闭连接

self.conn.close()

def query(self, sql, state="all"):

"""

查询

:param sql:

:param state: all是默认查询全部

:return:

"""

self.cur.execute(sql)

if state == "all":

# 查询全部

data = self.cur.fetchall()

else:

# 查询单条

data = self.cur.fetchone()

return data

def execute(self, sql):

"""

更新、删除、新增

:param sql:

:return:

"""

try:

# 使用execute操作sql

rows = self.cur.execute(sql)

# 提交事务

self.conn.commit()

return rows

except Exception as e:

print("数据库操作异常 {0}".format(e))

# 回滚修改

self.conn.rollback()

if __name__ == '__main__':

mydb = MysqlDb()

r = mydb.query("select * from `users`","all")

print(r)

mydb.execute("insert into `users` (`id`,`name`) values(0,'bob')")

print(r)