#安装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
关闭数据库连接