5第八周 - 网络编程进阶 - Python 连接 Mysql

Posted 2020-11-10 chen170615

 

Python操作MySQL主要使用两种方式：
原生模块：pymsql 其中pymsql是Python中操作MySQL的模块，其使用方法和MySQLdb几乎相同。
ORM框架：SQLAchemy

 

Mysql pymysql 基本步骤

1、pymysql 基本结构语法如下：

import pymysql

#创建连接
conn = pymysql.connect(host=\'10.10.28.5\',port=3306,user=\'chen\',passwd=\'Chen1203..\',db=\'school\',charset="utf8")

#创建游标
cursor = conn.cursor() #创建当前数据的游标

#执行SQL，并返回影响行数，常规的数据库操作命令：select、update、insert
#cursor.execute("insert into a (number) VALUES (1203)")

#执行SQL，并返回影响行数，执行多次
cursor.executemany("insert into a (number) VALUES (%s)",[("1000"),("1001"),("1002")])
cursor.execute("select * from a  ")
#output = cursor.fetchmany(3) #根据执行SQL的返回，读取前面三个数字
#output = cursor.fetchone()   #根据执行SQL的返回，读取一个数字
output = cursor.fetchall()    #根据执行SQL的返回,进行全部打印
print(output)

#提交，不然无法保存新建或者修改数据
conn.commit()
#关闭游标
cursor.close()
#关闭连接
conn.close()

输出：
( (1203,), (1203,), (1203,), (1203,), (1000,), (1001,), (1002,))

 注：存在中文的时候，连接需要添加charset=\'utf8\'，否则中文显示乱码。

2、获取新建数据的自增ID号

import pymysql

#创建连接
conn = pymysql.connect(host=\'103.66.55.234\',port=3306,user=\'chen\',passwd=\'Chen1203..\',db=\'school\',charset="utf8")

#创建游标
cursor = conn.cursor() #创建当前数据的游标

#执行SQL，并返回影响行数，常规的数据库操作命令：select、update、insert
#cursor.execute("insert into a (number) VALUES (1203)")

#执行SQL，并返回影响行数，执行多次
cursor.executemany("insert into chen (name) VALUES (%s)",[("qing4"),("qing5"),("qing6")])

#提交，不然无法保存新建或者修改数据
conn.commit()
#关闭游标
cursor.close()
#关闭连接
conn.close()

new_id = cursor.lastrowid #获取自增id
print (new_id)

 3、移动游标

操作都是靠游标，那对游标的控制也是必要的

注：在fetch数据时按照顺序进行，可以使用cursor.scroll(num,mode)来移动游标位置，如：
cursor.scroll(1,mode=\'relative\') # 相对当前位置移动
cursor.scroll(2,mode=\'absolute\') # 相对绝对位置移动

 4、数据类型

关于默认获取的数据是元祖类型，如果想要或者字典类型的数据

import pymysql

#创建连接
conn = pymysql.connect(host=\'103.66.53.98\',port=3306,user=\'chen\',passwd=\'Chen1203..\',db=\'school\',charset="utf8")

#设置游标为字典类型
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) 
cursor.execute("select * from chen")

row = cursor.fetchall()
print(row)

#提交
conn.commit()
#关闭游标
cursor.close()
#关闭连接
conn.close()

 5、调用存储过程

A、无参数存储过程

import pymysql
 
conn = pymysql.connect(host=\'127.0.0.1\', port=3306, user=\'root\', passwd=\'\', db=\'tkq1\')
#游标设置为字典类型
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
#无参数存储过程
cursor.callproc(\'p2\')  #等价于cursor.execute("call p2()")
 
row_1 = cursor.fetchone()
print row_1
 
conn.commit()
cursor.close()
conn.close()

 B、有参数的存储过程

import pymysql
 
conn = pymysql.connect(host=\'127.0.0.1\', port=3306, user=\'root\', passwd=\'\', db=\'tkq1\')
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
 
cursor.callproc(\'p1\', args=(1, 22, 3, 4))
#获取执行完存储的参数,参数@开头
cursor.execute("select @p1,@_p1_1,@_p1_2,@_p1_3")  #{u\'@_p1_1\': 22, u\'@p1\': None, u\'@_p1_2\': 103, u\'@_p1_3\': 24}
row_1 = cursor.fetchone()
print row_1

conn.commit()
cursor.close()
conn.close()

 6、使用with简化连接过程

使用上下文管理，简化连接过程

import pymysql
import contextlib
#定义上下文管理器，连接后自动关闭连接
@contextlib.contextmanager
def mysql(host=\'127.0.0.1\', port=3306, user=\'root\', passwd=\'\', db=\'tkq1\',charset=\'utf8\'):
  conn = pymysql.connect(host=host, port=port, user=user, passwd=passwd, db=db, charset=charset)
  cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
  try:
    yield cursor
  finally:
    conn.commit()
    cursor.close()
    conn.close()
 
# 执行sql
with mysql() as cursor:
  print(cursor)
  row_count = cursor.execute("select * from tb7")
  row_1 = cursor.fetchone()
  print row_count, row_1

 pymysql防注入：参考链接 https://www.cnblogs.com/wt11/p/6141225.html