前言

      我们都知道利用python实现mysql的操作是件很简单的事情,只需要熟练使用MySQLdb模块就能实现mysql的增删改查操作。

      为了更好地整合mysql的操作,使用python的类讲mysql的操作整合到一起,是个不错的思路。这里我编写了一个简单的class,来实现对mysql的操作与查询。


操作

      本例中,我们准备在mysql的iceny中创建了一张测试表t1,字段为id和timestamp,主要存储系统的时间戳,并在该表中进行增、删、改、查的操作:

      当前mysql的状态:

      image.png

      

      MySQLdb为python的第三方模块,使用之前需提前安装该模块,这里推荐pip安装:

pip install MySQL-python

     注:python3.4之后Mysql-python已经不被支持了,可以换成mysqlclient模块。


       编写mysql的class类:

#!/usr/local/env python3.6 
# -*- coding: UTF-8 -*-

import MySQLdb

class Mysql(object):
    def __init__(self,host,port,user,passwd,db,charset='utf8'):
        """初始化mysql连接"""
        try:
            self.conn = MySQLdb.connect(host,user,passwd,db,int(port))
        except MySQLdb.Error as e:
            errormsg = 'Cannot connect to server\nERROR(%s):%s' % (e.args[0],e.args[1])
            print(errormsg)
            exit(2)
        self.cursor = self.conn.cursor()

    def exec(self,sql):
        """执行dml,ddl语句"""
        try:
           self.cursor.execute(sql)
           self.conn.commit()
        except:
           self.conn.rollback()

    def query(self,sql):
        """查询数据"""
        self.cursor.execute(sql)
        return self.cursor.fetchall()

    def __del__(self):
        """ 关闭mysql连接 """
        self.conn.close()
        self.cursor.close()

      创建mysql对象:

mysql_test = Mysql('192.168.232.128','3306','root','123456','iceny')

      创建表t1:

mysql_test.exec('create table t1 (id int auto_increment primary key,timestamp TIMESTAMP)')

      image.png

      往t1插入一条数据:

mysql_test.exec('insert into t1 (id,timestamp) value (NULL,CURRENT_TIMESTAMP)')


      image.png

      更新id为1的数据时间戳,改为执行当前的系统时间:

      image.png

      再插入一条数据,查询该表数据:

mysql_test.exec('insert into t1 (id,timestamp) value (NULL,CURRENT_TIMESTAMP)')  
result = mysql_test.query('select * from t1')
print(result)

      image.png

      可以看到查询出来的结果是存放在一个元祖中。

      删除表中id = 1的数据:

mysql_test.exec('delete from t1 where id = 1')

      image.png


      以上就是通过python编写简单的class类操作mysql增删改查的简单实现,这已经能够应付日常工作中大量的mysql操作了。