mysql过程 设置 Python3_[接口测试-实战]01 python3操作mysql

前言

最近跟大婶的一个课程(可以到千聊平台搜索“颠覆你的python接口自动化测试”)，主要讲通过py+httplib+mysql+Django来搭建接口测试的一个实战，课程使用的是环境和我的环境有点区别，课程使用的是py2+httplib，而我使用的是py3+requests。主要跟的其实还是思想，而且我也收获良多，这里我会将学习的结果输出到简书，供应大家一起学习，当然我自认为写代码的能力和经验有限，所以可能会出一些低级的问题，希望各位指点，我们一起学习，进步。

这里我记录一下，我在跟第四节课《python操作mysql》的成果记录以及遇到的一些坑。

文章目录

课程内容简介

实践代码

坑

课程内容简介

大婶的第四次分享《python操作mysql》主要包括三个内容，“py操作mysql实例”，“本章涉及的语法讲解”，“pycharm使用基础”，本文的主要内容还是“py操作mysql实例”，其他的部分，在后面的文章中陆续更新。

那么下面我们对实例部分做一个简单的介绍，课程主要对mysqldb进行一个简单再封装的操作，提供几个框架中会用到的一些基本功能，功能比较简单(代码中有注释)，而mysqldb暂时不支持py3，所以实战中我使用的是pymysql这个模块。(不要问我为什么知道，我不告诉你)

实战代码

实战涉及到的新知识(这里只是记录一些关键字，后面会有相关的文章输出)：logging模块，pymysql模块，单例设置模式

实战中我封装了两个文件，opmysql.py(数据库操作的封装)，mylogging.py(logging日志的封装)

在看代码之前，我先抛出几个问题，大家可以带着这些问题看代码，印象可能会更深刻，当然很多问题都可以在代码中找到答案，毕竟，我还是很喜欢写注释的。

课程中每次使用logging的时候都需要设置输出格式信息，怎么封装

希望整个系统公用一个logging实例，怎么做

除了捕获到Exception的时候需要输入log，还有什么时候需要输出log，才能帮助我在出现bug的时候定位问题

凡是有sql的地方都有sql注入的问题，所以，在封装的过程中，我们该怎么处理sql注入的问题

# mylogging.py

#!/usr/bin/env python3

# -*- coding:utf-8 -*-

import logging

#在实际开发中，不会直接用logging，一般会将logging封装一下，再用

class MyLogging:

'''

logging的初始化操作，以类封装的形式进行

'''

def __init__(self):

filename = "app.log" #日志文件的地址

self.logger = logging.getLogger() #定义对应的程序模块名name，默认为root

self.logger.setLevel(logging.INFO) #必须设置，这里如果不显示设置，默认过滤掉warning之前的所有级别的信息

sh = logging.StreamHandler() #日志输出到屏幕控制台

sh.setLevel(logging.ERROR) #设置日志等级

fh = logging.FileHandler(filename=filename) #向文件filename输出日志信息

fh.setLevel(logging.INFO) #设置日志等级

# 设置格式对象

formatter = logging.Formatter("%(asctime)s %(filename)s[line:%(lineno)d]%(levelname)s - %(message)s") #定义日志输出格式

# 设置handler的格式对象

sh.setFormatter(formatter)

fh.setFormatter(formatter)

# 将handler增加到logger中

self.logger.addHandler(sh)

self.logger.addHandler(fh)

# python的模块是天然的单例模式，因为模块在第一次导入的时候，会生成.pyc文件，当第二次导入时，就会直接加载.pyc文件，而不会再次执行模块代码

# 因此，我们只需要把相关的文件和数据定义在一个模块中，就可以获得一个单例对象了

# 日志操作对象logger，这里设置为单例，则凡是需要使用logging对象的地方都共同这个logger，节省内存开销

mylogger = MyLogging().logger

if __name__=="__main__":

#test

mylogger.debug("debug")

mylogger.info("info")

mylogger.warning("warning")

mylogger.error("error")

mylogger.critical("critical")

# opmysql.py

#!/usr/bin/env python3

# -*- coding:utf-8 -*-

'''

定义对mysql数据库基本操作的封装

1.单条数据的操作：insert，update，delete

2.查询数据表的一条数据

3.查询数据表的所有数据

'''

import pymysql

from mylogging import mylogger

class OpMysql:

'''mysql操作类，基于pymysql的封装'''

def __init__(self,host="127.0.0.1",username="root",password="",databasename="dsn"):

'''

初始化opmysql对象

:param host:数据库服务器地址

:param username:数据库用户名

:param password:用户的密码

:param databasename:要操作的库名称

'''

try:

self.conn = pymysql.connect(

host = host,

user = username,

password = password,

database = databasename,

charset = 'utf8', # 如果sql语句中存在中文字符的时候，需要在这里指定charset的参数，否则中文显示乱码

cursorclass = pymysql.cursors.DictCursor # pymysql默认select获取的数据是元祖类型，如果想要字典类型的数据，可以在这里统一设置

)

self.cur = self.conn.cursor() #创建游标

except pymysql.Error as e:

mylogger.info("[connection_message]-host:%s;user:%s;password:%s;database:%s"%(host,username,password,databasename))

mylogger.exception(e)

def op_sql(self,query,params=None):

'''

单条数据的操作，insert，update，delete

:param query:包含%s的sql字符串，当params=None的时候，不包含%s

:param params:一个元祖，默认为None

:return:如果执行过程没有crash，返回True，反之返回False

'''

try:

self.cur.execute(query,params)

self.conn.commit()

return True

except BaseException as e:

self.conn.rollback() #如果这里是执行的执行存储过程的sql命令，那么可能会存在rollback的情况，所以这里应该考虑到

mylogger.info("[sql_str_message]-%s"%self.cur.mogrify(query,params))

mylogger.exception(e)

return False

def select_one(self,query,params=None):

'''

查询数据表的单条数据

:param query: 包含%s的sql字符串，当params=None的时候，不包含%s

:param params: 一个元祖，默认为None

:return: 如果执行未crash，并以包含dict的列表的方式返回select的结果，否则返回错误代码001

'''

try:

self.cur.execute(query,params)

self.cur.scroll(0,"absolute") #光标回到初始位置，感觉自己的这句有点多余

return self.cur.fetchone()

except BaseException as e:

mylogger.info("[sql_str_message]-%s" % self.cur.mogrify(query, params))

mylogger.exception(e)

return "001" #错误代码001

def select_all(self,query,params=None):

'''

查询数据表的单条数据

:param query:包含%s的sql字符串，当params=None的时候，不包含%s

:param params:一个元祖，默认为None

:return:如果执行未crash，并以包含dict的列表的方式返回select的结果，否则返回错误代码001

'''

try:

self.cur.execute(query,params)

self.cur.scroll(0,"absolute") #光标回到初始位置，感觉这里得这句有点多余

return self.cur.fetchall()

except BaseException as e:

mylogger.info("[sql_str_message]-%s" % self.cur.mogrify(query, params))

mylogger.exception(e)

return "001" #错误代码001

def insert_many(self,query,params):

'''

向数据表中插入多条数据

:param query:包含%s的sql字符串，当params=None的时候，不包含%s

:param params:一个内容为元祖的列表

:return:如果执行过程没有crash，返回True，反之返回False

'''

try:

self.cur.executemany(query,params)

self.conn.commit()

return True

except BaseException as e:

self.conn.rollback()

mylogger.info("[sql_str_message]-%s" % self.cur.mogrify(query, params))

mylogger.exception(e)

return False

def __del__(self):

'''

当该对象的引用计数为0的时候，python解释器会自动执行__dell__方法，自动释放游标和链接

'''

self.cur.close()

self.conn.close()

if __name__ == "__main__":

# test

opmysql = OpMysql()

print(opmysql.select_one("SELECT * FROM user WHERE id = %s;",(1,)))

opmysql.select_one("SELECT * FROM user WHERE idd = %s;",("12",))

填坑指南

怎么样，看完上面的代码，你找到前面问题的答案没有，如果没有，继续看下去，我会给你参考链接，拿走学习吧。

【问】课程中每次使用logging的时候都需要设置输出格式信息，怎么封装

【答案】代码中有答案，同时可以参考这里

【问】希望整个系统公用一个logging实例，怎么做

【答案】单例模式(Singleton Pattern)，是一种常用的软件设计模式，该模式的主要目的是确保某一个类只有一个实例存在，当你希望在整个系统中，某个类只能出现一个实例的时候，单例对象就能派上用处，比如我们实践代码中的mylogging类就可以采用单例模式的方式来处理，节约内存资源。python中实现单例模式的方式有很多种，这里使用模块的方式来获得一个单例对象，具体实现方式，代码中有，自己找，当然你也可以参考这里

【问】除了捕获到Exception的时候需要输入log，还有什么时候需要输出log，才能帮助我在出现bug的时候定位问题

【答案】在错误日志里面除了有exception的具体内容而外，还应该打印sql信息，函数入参等，在报错的时候，除了提供的出错模块，行数，类型等信息之外，如果能够知道是在传递入什么内容之后执行出错的，可以更好的定位问题并分析原因。

【问】凡是有sql的地方都有sql注入的问题，所以，在封装的过程中，我们该怎么处理sql注入的问题 **

【答案】pymysql模块的execute/callproc方法的参数化语句自动防注入，所以我们在进行sql拼接的时候尽量使用pymysql的execute和callproc方法的参数化语句，具体可以查看我之前的笔记做了解《python3使用pymysql操作mysql》

这些都是课程里面没有覆盖到的地方，也是我自己在实践中的疑问和解决结果

好啦，暂时分享到这里，自己文笔真的太有限了，看来在未来，要好好研究一下写作了，不过，只要你们看得懂就行。