python 数据库操作类

#安装PyMySQL：pip3 install PyMySQL  
#!/usr/bin/python3  
#coding=utf-8  
#数据库操作类

   
from  datetime  import  * 
import pymysql  
import hashlib
import time

class SingletonModel:
        #数据库连接对象
        __db = None
        #游标对象
        __cursor = None
        def __new__(self, *args, **kwargs):
                if not hasattr(self, '_instance'):
                        self._instance = super().__new__(self)
                        #主机
                        host = 'host' in kwargs and kwargs['host'] or 'localhost'
                        #端口
                        port = 'port' in kwargs and kwargs['port'] or '3306'
                        #用户名
                        user = 'user' in kwargs and kwargs['user'] or 'root'
                        #密码
                        passwd = 'passwd' in kwargs and kwargs['passwd'] or '123456'
                        #数据库
                        db = 'db' in kwargs and kwargs['db'] or 'mypython'
                        #编码
                        charset = 'charset' in kwargs and kwargs['charset'] or 'utf8'
                        # 打开数据库连接  
                        print('连接数据库')
                        self.__db = pymysql.connect(host=host,port=int(port),user=user,passwd=passwd,db=db,charset=charset)
                        #创建一个游标对象 cursor
                        #self.__cursor = self.__db.cursor()
                        self.__cursor = self.__db.cursor(cursor=pymysql.cursors.DictCursor)
                return self._instance

        #返回执行execute()方法后影响的行数 
        def execute(self,sql):
                self.__cursor.execute(sql)
                rowcount = self.__cursor.rowcount
                return rowcount

        #增->返回新增ID
        def insert(self,**kwargs):
                table = kwargs['table']
                del kwargs['table']
                sql = 'insert into %s set '%table
                for k,v in kwargs.items():
                        sql += "`%s`='%s',"%(k,v)
                sql = sql.rstrip(',')
                print(sql)
                try:
                        # 执行SQL语句
                        self.__cursor.execute(sql)
                        # 提交到数据库执行
                        self.__db.commit()
                        #获取自增id
                        res = self.__cursor.lastrowid  
                except:
                        # 发生错误时回滚
                        self.__db.rollback()
                return res

        #删->返回影响的行数
        def delete(self,**kwargs):
                table = kwargs['table']
                where = kwargs['where']
                sql = 'DELETE FROM %s where %s'%(table,where)
                print(sql)
                try:
                        # 执行SQL语句
                        self.__cursor.execute(sql)
                        # 提交到数据库执行
                        self.__db.commit()
                        #影响的行数
                        rowcount = self.__cursor.rowcount
                except:
                        # 发生错误时回滚
                        self.__db.rollback()
                return rowcount

        #改->返回影响的行数
        def update(self,**kwargs):
                table = kwargs['table']
                #del kwargs['table']
                kwargs.pop('table')

                where = kwargs['where'] 
                kwargs.pop('where')

                sql = 'update %s set '%table
                for k,v in kwargs.items():
                        sql += "`%s`='%s',"%(k,v)
                sql = sql.rstrip(',')
                sql += ' where %s'%where 
                print(sql)
                try:
                        # 执行SQL语句
                        self.__cursor.execute(sql)
                        # 提交到数据库执行
                        self.__db.commit()
                        #影响的行数
                        rowcount = self.__cursor.rowcount
                except:
                        # 发生错误时回滚
                        self.__db.rollback()
                return rowcount

        #查->单条数据
        def fetchone(self,**kwargs):
                table = kwargs['table']
                #字段
                field = 'field' in kwargs and kwargs['field'] or '*'
                #where
                where = 'where' in kwargs and 'where '+kwargs['where'] or ''
                #order
                order = 'order' in kwargs and 'order by '+ kwargs['order'] or ''

                sql = 'select %s from %s %s %s limit 1'%(field,table,where,order) 
                print(sql)
                try:
                        # 执行SQL语句
                        self.__cursor.execute(sql)
                        # 使用 fetchone() 方法获取单条数据.
                        data = self.__cursor.fetchone()
                except:
                        # 发生错误时回滚
                        self.__db.rollback()
                return data

        #查->多条数据
        def fetchall(self,**kwargs):
                table = kwargs['table']
                #字段
                field = 'field' in kwargs and kwargs['field'] or '*'
                #where
                where = 'where' in kwargs and 'where '+kwargs['where'] or ''
                #order
                order = 'order' in kwargs and 'order by '+ kwargs['order'] or ''
                #limit
                limit = 'limit' in kwargs and 'limit '+ kwargs['limit'] or ''
                sql = 'select %s from %s %s %s %s'%(field,table,where,order,limit) 
                print(sql)
                try:
                        # 执行SQL语句
                        self.__cursor.execute(sql)
                        # 使用 fetchone() 方法获取单条数据.
                        data = self.__cursor.fetchall()
                except:
                        # 发生错误时回滚
                        self.__db.rollback()
                return data

        #析构函数，释放对象时使用
        def __del__(self):
                # 关闭数据库连接
                self.__db.close()
                print('关闭数据库连接')

#生成md5
def makeMd5(mstr):
        hmd5 = hashlib.md5()
        hmd5.update(mstr.encode("utf-8"))
        return hmd5.hexdigest()

#获取unix时间戳
def getTime():
        return round(time.time())

#时间格式化
def timeFormat(timestamp):
        #return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(timestamp))
        #return datetime.fromtimestamp(timestamp)
        return datetime.utcfromtimestamp(timestamp)

if __name__ == '__main__':

        dbObject = SingletonModel(host='localhost',port=3306,user='root',passwd='root',db='mypython',charset='utf8')

        #创建表
        print('创建表:')
        sql = "DROP TABLE IF EXISTS `user`;"
        dbObject.execute(sql)
        sql = '''
        CREATE TABLE `user` (
          `id` int(11) NOT NULL AUTO_INCREMENT,
          `name` varchar(50) NOT NULL,
          `pwd` char(32) NOT NULL,
          `insert_time` int(11) NOT NULL,
          PRIMARY KEY (`id`)
        ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8 COMMENT='用户表';
        '''
        print(sql)
        res = dbObject.execute(sql)
        print(res)

        #写入数据
        print('\n写入数据:')
        pwd = makeMd5('123456')
        insert_time = getTime()
        res = dbObject.insert(table='user',name='aaaa',pwd=pwd,insert_time=insert_time)
        print(res)

        time.sleep(1)
        pwd = makeMd5('654321')
        insert_time = getTime()
        res = dbObject.insert(table='user',name='bbbb',pwd=pwd,insert_time=insert_time)
        print(res)

        time.sleep(1)
        pwd = makeMd5('111111')
        insert_time = getTime()
        res = dbObject.insert(table='user',name='cccc',pwd=pwd,insert_time=insert_time)
        print(res)

        #查询数据-单条
        print('\n查询数据-单条:')
        res = dbObject.fetchone(table='user',where="name='cccc'")
        print(res)

        #修改数据
        print('\n修改数据:')
        res = dbObject.update(table='user',where="id=1",name='dddd')
        print(res)

        #删除数据
        print('\n删除数据:')
        res = dbObject.delete(table='user',where="id=2")
        print(res)

        #查询数据-多条
        print('\n查询数据-多条:')
        res = dbObject.fetchall(table='user',order="id desc")
        print(res,type(res))
        if res:
                for value in res:
                        print('name:%s,date:%s'%(value['name'],timeFormat(value['insert_time'])))

[chaoge@localhost db]$ python3 SingleModel.py 

连接数据库

创建表:

        CREATE TABLE `user` (

          `id` int(11) NOT NULL AUTO_INCREMENT,

          `name` varchar(50) NOT NULL,

          `pwd` char(32) NOT NULL,

          `insert_time` int(11) NOT NULL,

          PRIMARY KEY (`id`)

        ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8 COMMENT='用户表';

0

写入数据:

insert into user set `pwd`='e10adc3949ba59abbe56e057f20f883e',`name`='aaaa',`insert_time`='1517543918'

1

insert into user set `pwd`='c33367701511b4f6020ec61ded352059',`name`='bbbb',`insert_time`='1517543919'

2

insert into user set `pwd`='96e79218965eb72c92a549dd5a330112',`name`='cccc',`insert_time`='1517543920'

3

查询数据-单条:

select * from user where name='cccc'  limit 1

{'pwd': '96e79218965eb72c92a549dd5a330112', 'id': 3, 'name': 'cccc', 'insert_time': 1517543920}

修改数据:

update user set `name`='dddd' where id=1

1

删除数据:

DELETE FROM user where id=2

1

查询数据-多条:

select * from user  order by id desc 

[{'pwd': '96e79218965eb72c92a549dd5a330112', 'id': 3, 'name': 'cccc', 'insert_time': 1517543920}, {'pwd': 'e10adc3949ba59abbe56e057f20f883e', 'id': 1, 'name': 'dddd', 'insert_time': 1517543918}] <class 'list'>

name:cccc,date:2018-02-01 22:58:40

name:dddd,date:2018-02-01 22:58:38

关闭数据库连接