常用函数
Python DB API 2.0 对事务提供了两个方法:
commit()
提交rollback()
回滚
cursor用来执行命令的方法:
callproc(self, procname, args)
用来执行存储过程,接收的参数为存储过程名和参数列表,返回值为受影响的行数execute(self, query, args)
执行单条sql语句,接收的参数为sql语句本身和使用的参数列表,返回值为受影响的行数executemany(self, query, args)
执行单挑sql语句,但是重复执行参数列表里的参数,返回值为受影响的行数nextset(self)
移动到下一个结果集
cursor用来接收返回值的方法:
fetchall(self)
接收全部的返回结果行.fetchmany(self, size=None)
接收size条返回结果行.如果size的值大于返回的结果行的数量,则会返回cursor.arraysize条数据.fetchone(self)
返回一条结果行.rowcount
这是一个只读属性,并返回执行execute() 方法后影响的行数。scroll(self, value, mode='relative')
移动指针到某一行; 如果mode='relative',则表示从当前所在行移动value条,如果 mode='absolute',则表示从结果集的第一行移动value条.
实例
#!/usr/bin/env python # -*- coding: utf-8 -*- import MySQLdb as mdb # 连接数据库 conn = mdb.connect('localhost', 'root', 'root') # 也可以使用关键字参数 conn = mdb.connect(host='127.0.0.1', port=3306, user='root', passwd='root', db='test', charset='utf8') # 也可以使用字典进行连接参数的管理 config = { 'host': '127.0.0.1', 'port': 3306, 'user': 'root', 'passwd': 'root', 'db': 'test', 'charset': 'utf8' } conn = mdb.connect(**config) # 如果使用事务引擎,可以设置自动提交事务,或者在每次操作完成后手动提交事务conn.commit() conn.autocommit(1) # conn.autocommit(True) # 使用cursor()方法获取操作游标 cursor = conn.cursor() # 因该模块底层其实是调用CAPI的,所以,需要先得到当前指向数据库的指针。 try: # 创建数据库 DB_NAME = 'test' cursor.execute('DROP DATABASE IF EXISTS %s' %DB_NAME) cursor.execute('CREATE DATABASE IF NOT EXISTS %s' %DB_NAME) conn.select_db(DB_NAME) #创建表 TABLE_NAME = 'user' cursor.execute('CREATE TABLE %s(id int primary key,name varchar(30))' %TABLE_NAME) # 插入单条数据 sql = 'INSERT INTO user values("%d","%s")' %(1,"jack") # 不建议直接拼接sql,占位符方面可能会出问题,execute提供了直接传值 value = [2,'John'] cursor.execute('INSERT INTO test values(%s,%s)',value) # 批量插入数据 values = [] for i in range(3, 20): values.append((i,'kk'+str(i))) cursor.executemany('INSERT INTO user values(%s,%s)',values) # 查询数据条目 count = cursor.execute('SELECT * FROM %s' %TABLE_NAME) print 'total records: %d' %count print 'total records:', cursor.rowcount # 获取表名信息 desc = cursor.description print "%s %3s" % (desc[0][0], desc[1][0]) # 查询一条记录 print 'fetch one record:' result = cursor.fetchone() print result print 'id: %s,name: %s' %(result[0],result[1]) # 查询多条记录 print 'fetch five record:' results = cursor.fetchmany(5) for r in results: print r # 查询所有记录 # 重置游标位置,偏移量:大于0向后移动;小于0向前移动,mode默认是relative # relative:表示从当前所在的行开始移动; absolute:表示从第一行开始移动 cursor.scroll(0,mode='absolute') results = cursor.fetchall() for r in results: print r cursor.scroll(-2) results = cursor.fetchall() for r in results: print r # 更新记录 cursor.execute('UPDATE %s SET name = "%s" WHERE id = %s' %(TABLE_NAME,'Jack',1)) # 删除记录 cursor.execute('DELETE FROM %s WHERE id = %s' %(TABLE_NAME,2)) # 如果没有设置自动提交事务,则这里需要手动提交一次 conn.commit() except: import traceback traceback.print_exc() # 发生错误时会滚 conn.rollback() finally: # 关闭游标连接 cursor.close() # 关闭数据库连接 conn.close()
查询时返回字典结构
MySQLdb默认查询结果都是返回tuple,通过使用不同的游标可以改变输出格式,这里传递一个cursors.DictCursor参数。
import MySQLdb.cursors conn = MySQLdb.connect(host='localhost', user='root', passwd='root', db='test', cursorclass=MySQLdb.cursors.DictCursor) cursor = conn.cursor() cursor.execute('select * from user') r = cursor.fetchall() print r # 当使用位置参数或字典管理参数时,必须导入MySQLdb.cursors模块 # 也可以用下面的写法 import MySQLdb as mdb conn = mdb.connect('localhost', 'root', 'root', 'test') cursor = conn.cursor(cursorclass=mdb.cursors.DictCursor) cursor.execute('select * from user') r = cursor.fetchall() print r
MySQLdb取回大结果集的技巧
普通的操作无论是fetchall()还是fetchone()都是先将数据载入到本地再进行计算,大量的数据会导致内存资源消耗光。解决办法是使用SSCurosr光标来处理。
然而,在python3下,MySQLdb模块不再提供支持,此时可以使用另一个模块PyMySQL,它支持python2和python3。
2. PyMySQL
PyMySQL是一个纯Python写的MySQL客户端,它的目标是替代MySQLdb,可以在CPython、PyPy、IronPython和Jython环境下运行。PyMySQL在MIT许可下发布。
PyMySQL的性能和MySQLdb几乎相当,如果对性能要求
不是特别的强,使用PyMySQL将更加方便。
PyMySQL的使用方法和MySQLdb几乎一样。
安装
pip install pymysql
实例
#!/usr/bin/env python # -*- coding: utf-8 -*- import pymysql config = { 'host': '127.0.0.1', 'port': 3306, 'user': 'root', 'passwd': 'root', 'charset':'utf8mb4', 'cursorclass':pymysql.cursors.DictCursor } conn = pymysql.connect(**config) conn.autocommit(1) cursor = conn.cursor() try: # 创建数据库 DB_NAME = 'test' cursor.execute('DROP DATABASE IF EXISTS %s' %DB_NAME) cursor.execute('CREATE DATABASE IF NOT EXISTS %s' %DB_NAME) conn.select_db(DB_NAME) #创建表 TABLE_NAME = 'user' cursor.execute('CREATE TABLE %s(id int primary key,name varchar(30))' %TABLE_NAME) # 批量插入纪录 values = [] for i in range(20): values.append((i,'kk'+str(i))) cursor.executemany('INSERT INTO user values(%s,%s)',values) # 查询数据条目 count = cursor.execute('SELECT * FROM %s' %TABLE_NAME) print 'total records:', cursor.rowcount # 获取表名信息 desc = cursor.description print "%s %3s" % (desc[0][0], desc[1][0]) cursor.scroll(10,mode='absolute') results = cursor.fetchall() for result in results: print result except: import traceback traceback.print_exc() # 发生错误时回滚 conn.rollback() finally: # 关闭游标连接 cursor.close() # 关闭数据库连接 conn.close()#coding = utf-8#3.6版本使用pymysql,2.7版本使用MySQLdb
import pymysql#连接数据库使用关键字参数的方式conn = pymysql.connect(host='localhost', port = 3306, user = 'root', passwd = 'password', db = 'py_test',)
#连接数据库,使用字典进行连接参数的管理conn = pymysql.connect ( host = 'localhost', port = 3306, user = 'root', passwd = 'password', db = 'py_test', )#通过获取到的数据连接conn下的cursor()方法来创建游标cur = conn.cursor()#通过游标cur操作execute()方法写入纯sql语句。通过execute()方法中写入sql语句来对数据库进行操作。#创建数据库cur.execute("create database py_test")#创建数据库表cur.execute("create table student(id int ,name varchar(20),class varchar(30),age varchar(10))"#添加数据cur.execute("insert into student values('3','Xue','2 year 1 class','10')")#删除数据cur.execute("delete from student where age='9'")#修改查询条件的数据cur.execute("update student set class='3 year 1 class' where name = 'Amy'")#查询表中有多少条数据aa = cur.execute("select * from student")print(aa)#打印表中的数据info = cur.fetchmany(aa)for ii in info: print(ii)info = cur.fetchmany(cur.execute("select * from student"))for ii in info: print(ii)#插入数据的优化方法sqli="insert into student values(%s,%s,%s,%s)"cur.execute(sqli,('3','Huhu','2 year 1 class','7'))cur.execute(sqli,('2','Fufu','4 year 5 class','6'))cur.executemany(sqli, [('3','Qiqi','2 year 1 class','7'), ('2','Bobo','4 year 5 class','6'), ('4','Anan','5 year 2 class','2'), ])#异常try: cur.execute(sqli) conn.commit()except: conn.rollback() #发生错误时回滚#关闭游标cur.close()#conn.commit()提交数据库执行,再向数据库插入一条数据时必须要有这个方法,否则数据不会被真正的插入conn.commit()#关闭数据库连接conn.close()
来源:https://www.cnblogs.com/qfdxxdr/p/7040219.html