在Python程序中操作MySQL的基本方法
Python操作Mysql 最近在学习python,这种脚本语言毫无疑问的会跟数据库产生关联,因此这里介绍一下如何使用python操作mysql数据库。我python也是零基础学起,所以本篇博客针对的是python初学者,大牛可以选择绕道。 另外,本篇基于的环境是Ubuntu13.10,使用的python版本是2.7.5。 MYSQL是一个全球领先的开源数据库管理系统。它是一个支持多用户、多线程的数据库管理系统,与Apache、PHP、Linux共同组成LAMP平台,在web应用中广泛使用,例如Wikipedia和YouTube。MYSQL包含两个版本:服务器系统和嵌入式系统。 在我们开始语法学习之前,还需要按装mysql和python对mysql操作的模块。 安装mysql: sudo apt-get install mysql-server 安装过程中会提示你输入root帐号的密码,符合密码规范即可。 接下来,需要安装python对mysql的操作模块: sudo apt-get install python-mysqldb 这里需要注意:安装完python-mysqldb之后,我们默认安装了两个python操作模块,分别是支持C语言API的_mysql和支持Python API的MYSQLdb。稍后会重点讲解MYSQLdb模块的使用。 接下来,我们进入MYSQL,创建一个测试数据库叫testdb。创建命令为: create database testdb;
create user 'testuser'@'127.0.0.1' identified by 'test123'; grant all privileges on testdb.* to 'testuser'@'127.0.0.1'; _mysql module _mysql模块直接封装了MYSQL的C语言API函数,它与python标准的数据库API接口是不兼容的。我更推荐大家使用面向对象的MYSQLdb模块才操作mysql,这里只给出一个使用_mysql模块的例子,这个模块不是我们学习的重点,我们只需要了解有这个模块就好了。 #!/usr/bin/python # -*- coding: utf-8 -*- import _mysql import sys try: con = _mysql.connect('127.0.0.1','testuser','test123','testdb') con.query("SELECT VERSION()") result = con.use_result() print "MYSQL version : %s " % result.fetch_row()[0] except _mysql.Error,e: print "Error %d: %s %s" % (e.args[0],e.args[1]) sys.exit(1) finally: if con: con.close() 这个代码主要是获取当前mysql的版本,大家可以模拟敲一下这部分代码然后运行一下。 MYSQLdb是在_mysql模块的基础上进一步进行封装,并且与python标准数据库API接口兼容,这使得代码更容易被移植。Python更推荐使用这个MYSQLdb模块来进行MYSQL操作。 #!/usr/bin/python # -*- coding: utf-8 -*- import MySQLdb as mysql try: conn = mysql.connect('127.0.0.1','testdb') cur = conn.cursor() cur.execute("SELECT VERSION()") version = cur.fetchone() print "Database version : %s" % version except mysql.Error,e: print "Error %d:%s" % (e.args[0],e.args[1]) exit(1) finally: if conn: conn.close()
接下来,我们通过MySQLdb模块创建一个表,并在其中填充部分数据。实现代码如下: #!/usr/bin/python # -*- coding: utf-8 -*- import MySQLdb as mysql conn = mysql.connect('127.0.0.1','testdb'); with conn: cur = conn.cursor() cur.execute("DROP TABLE IF EXISTS writers"); cur.execute("CREATE TABLE writers(id INT PRIMARY KEY AUTO_INCREMENT,name varchar(25))") cur.execute("insert into writers(name) values('wangzhengyi')") cur.execute("insert into writers(name) values('bululu')") cur.execute("insert into writers(name) values('chenshan')") 这里使用了with语句。with语句会执行conn对象的enter()和__exit()方法,省去了自己写try/catch/finally了。 执行完成后,我们可以通过mysql-client客户端查看是否插入成功,查询语句: select * from writers; 查询结果如下: 查询数据 刚才往表里插入了部分数据,接下来,我们从表中取出插入的数据,代码如下: #!/usr/bin/python import MySQLdb as mysql conn = mysql.connect('127.0.0.1','testdb'); with conn: cursor = conn.cursor() cursor.execute("select * from writers") rows = cursor.fetchall() for row in rows: print row
(1L,'wangzhengyi') (2L,'bululu') (3L,'chenshan') dictionary cursor 我们刚才不论是创建数据库还是查询数据库,都用到了cursor。在MySQLdb模块有许多种cursor类型,默认的cursor是以元组的元组形式返回数据的。当我们使用dictionary cursor时,数据是以python字典形式返回的。这样我们就可以通过列名获取查询数据了。 还是刚才查询数据的代码,改为dictionary cursor只需要修改一行代码即可,如下所示: #!/usr/bin/python import MySQLdb as mysql conn = mysql.connect('127.0.0.1','testdb'); with conn: cursor = conn.cursor(mysql.cursors.DictCursor) cursor.execute("select * from writers") rows = cursor.fetchall() for row in rows: print "id is %s,name is %s" % (row['id'],row['name']) 使用dictionary cursor,查询结果如下: id is 1,name is wangzhengyi id is 2,name is bululu id is 3,name is chenshan
之前写过php的同学应该对预编译很了解,预编译可以帮助我们防止sql注入等web攻击还能帮助提高性能。当然,python肯定也是支持预编译的。预编译的实现也比较简单,就是用%等占位符来替换真正的变量。例如查询id为3的用户的信息,使用预编译的代码如下: #!/usr/bin/python import MySQLdb as mysql conn = mysql.connect('127.0.0.1','testdb'); with conn: cursor = conn.cursor(mysql.cursors.DictCursor) cursor.execute("select * from writers where id = %s","3") rows = cursor.fetchone() print "id is %d,name is %s" % (rows['id'],rows['name']) 我这里使用了一个%s的占位符来替换“3”,代表需要传入的是一个字符串类型。如果传入的不是string类型,则会运行报错。 事务是指在一个或者多个数据库中对数据的原子操作。在一个事务中,所有的SQL语句的影响要不就全部提交到数据库,要不就全部都回滚。 对于支持事务机制的数据库,python接口在创建cursor的时候就开始了一个事务。可以通过cursor对象的commit()方法来提交所有的改动,也可以使用cursor对象的rollback方法来回滚所有的改动。 我这里写一个代码,对不存在的表进行插入操作,当抛出异常的时候,调用rollback进行回滚,实现代码如下: #!/usr/bin/python # -*- coding: utf-8 -*- import MySQLdb as mysql try: conn = mysql.connect('127.0.0.1','testdb'); cur = conn.cursor() cur.execute("insert into writers(name) values('wangzhengyi4')") cur.execute("insert into writers(name) values('bululu5')") cur.execute("insert into writerss(name) values('chenshan6')") conn.commit() except mysql.Error,e: if conn: conn.rollback() print "Error happens,rollback is call" finally: if conn: conn.close() 执行结果如下: Error happens,rollback is call 因为前两条数据是正确的插入操作,但是因为整体回滚,所以数据库里也没有wangzhengyi4和bululu5这两个数据的存在。 (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |