首页 > 技术文章 > MySQL与Python交互

xinyangsdut 2017-10-18 17:08 原文

关于MySQL推荐一本书MySQL必知必会

首先安装第三方模块(ubuntu下Python2)

sudo apt-get install python-mysql

假设有一数据库test1,里面有一张产品信息表products,向其中插入一条产品信息,程序如下:

# -*- coding: utf-8 -*-
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    count=cs1.execute("insert into products(prod_name) values('iphone')")
    print count
    conn.commit()
    cs1.close()
    conn.close()
except Exception,e:
    print e.message

Connection对象:用于建立与数据库的连接
        创建对象:调用connect()方法
conn=connect(参数列表)
    参数host:连接的mysql主机,如果本机是'localhost'
    参数port:连接的mysql主机的端口,默认是3306
    参数db:数据库的名称
    参数user:连接的用户名
    参数password:连接的密码
    参数charset:通信采用的编码方式,默认是'gb2312',要求与数据库创建时指定的编码一致,否则中文会乱码

对象的方法
    close()关闭连接
    commit()事务,所以需要提交才会生效
    rollback()事务,放弃之前的操作
    cursor()返回Cursor对象,用于执行sql语句并获得结果

Cursor对象:执行sql语句
    创建对象:调用Connection对象的cursor()方法
    cursor1=conn.cursor()

对象的方法
    close()关闭
    execute(operation [, parameters ])执行语句,返回受影响的行数
    fetchone()执行查询语句时,获取查询结果集的第一个行数据,返回一个元组
    next()执行查询语句时,获取当前行的下一行
    fetchall()执行查询时,获取结果集的所有行,一行构成一个元组,再将这些元组装入一个元组返回
    scroll(value[,mode])将行指针移动到某个位置
        mode表示移动的方式
        mode的默认值为relative,表示基于当前行移动到value,value为正则向下移动,value为负则向上移动
        mode的值为absolute,表示基于第一条数据的位置,第一条数据的位置为0

修改/删除:

# -*- coding: utf-8 -*-
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    # 修改
    count=cs1.execute("update products set prod_name='xiaomi' where id=6")
   # 删除
  count=cs1.execute("delete from products where id=6")
  print count
     conn.commit()
     cs1.close()
     conn.close()
except Exception,e:
    print e.message

参数化:插入一条数据

# -*- coding: utf-8 -*-
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    prod_name=raw_input("请输入产品名称:")
    params=[prod_name]
    count=cs1.execute('insert into products(sname) values(%s)',params)
    print count
    conn.commit()
    cs1.close()
    conn.close()
except Exception,e:
    print e.message

查询一条

# -*- coding: utf-8 -*-
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    cur.execute('select * from products where id=2')
    result=cur.fetchone()
    print result
    conn.commit() 
    cs1.close() 
    conn.close() 
except Exception,e: 
    print e.message

查询多条

# -*- coding: utf-8 -*-
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    cur.execute('select * from prod_name')
    result=cur.fetchall()
    print result
    conn.commit() 
    cs1.close() 
    conn.close() 
except Exception,e: 
    print e.message

封装:观察前面的程序发现,除了sql语句及参数不同,其它语句都是一样的,可以进行封装然后调用

# -*- coding: utf-8 -*-
import MySQLdb

class MysqlHelper():
    def __init__(self,host,port,db,user,passwd,charset='utf8'):
        self.host=host
        self.port=port
        self.db=db
        self.user=user
        self.passwd=passwd
        self.charset=charset

    def connect(self):
        self.conn=MySQLdb.connect(host=self.host,port=self.port,db=self.db,user=self.user,passwd=self.passwd,charset=self.charset)
        self.cursor=self.conn.cursor()

    def close(self):
        self.cursor.close()
        self.conn.close()

    def get_one(self,sql,params=()):
        result=None
        try:
            self.connect()
            self.cursor.execute(sql, params)
            result = self.cursor.fetchone()
            self.close()
        except Exception, e:
            print e.message
        return result

    def get_all(self,sql,params=()):
        list=()
        try:
            self.connect()
            self.cursor.execute(sql,params)
            list=self.cursor.fetchall()
            self.close()
        except Exception,e:
            print e.message
        return list

    def insert(self,sql,params=()):
        return self.__edit(sql,params)

    def update(self, sql, params=()):
        return self.__edit(sql, params)

    def delete(self, sql, params=()):
        return self.__edit(sql, params)

    def __edit(self,sql,params):
        count=0
        try:
            self.connect()
            count=self.cursor.execute(sql,params)
            self.conn.commit()
            self.close()
        except Exception,e:
            print e.message
        return count

保存为MysqlHelper.py文件。

调用类添加

# -*- coding: utf-8 -*-
from MysqlHelper import *

sql='insert intoproducts(prod_name,price) values(%s,%s)'
prod_name=raw_input("请输入产品名称:")
price=raw_input("请输入单价:")
params=[prod_name,price]

mysqlHelper=MysqlHelper('localhost',3306,'test1','root','mysql')
count=mysqlHelper.insert(sql,params)
if count==1:
    print 'ok'
else:
    print 'error'

调用类查询查询

# -*- coding: utf-8 -*-
from MysqlHelper import *

sql='select prod_name,price from products order by id '

helper=MysqlHelper('localhost',3306,'test1','root','mysql')
one=helper.get_one(sql)
print one

 

推荐阅读