pymysql操作数据库优化
|
<h1 id="autoid-0-0-0">pymysql操作数据库优化 我们之前使用pymysql操作数据库的操作都是写死在视图函数中的，并且很多都是重复的代码。 我们可以优化一下，把重复的代码提取出来，写成函数： <span style="color: #008000;">#<span style="color: #008000;"> 定义一个数据库相关的配置项
DB_CONFIG =<span style="color: #000000;"> { <span style="color: #800000;">"<span style="color: #800000;">host<span style="color: #800000;">": <span style="color: #800000;">"<span style="color: #800000;">127.0.0.1<span style="color: #800000;">"<span style="color: #000000;">,<span style="color: #800000;">"<span style="color: #800000;">port<span style="color: #800000;">": 3306<span style="color: #000000;">,<span style="color: #800000;">"<span style="color: #800000;">user<span style="color: #800000;">": <span style="color: #800000;">"<span style="color: #800000;">root<span style="color: #800000;">"<span style="color: #000000;">,<span style="color: #800000;">"<span style="color: #800000;">passwd<span style="color: #800000;">": <span style="color: #800000;">"<span style="color: #800000;">root1234<span style="color: #800000;">"<span style="color: #000000;">,<span style="color: #800000;">"<span style="color: #800000;">db<span style="color: #800000;">": <span style="color: #800000;">"<span style="color: #800000;">mysite<span style="color: #800000;">"<span style="color: #000000;">,<span style="color: #800000;">"<span style="color: #800000;">charset<span style="color: #800000;">": <span style="color: #800000;">"<span style="color: #800000;">utf8<span style="color: #800000;">"<span style="color: #000000;"> } <span style="color: #008000;">#<span style="color: #008000;"> 查询多条数据函数 <span style="color: #008000;">#<span style="color: #008000;"> 查询单挑数据函数 <span style="color: #008000;">#<span style="color: #008000;"> 修改记录 <span style="color: #008000;">#<span style="color: #008000;"> 创建记录 这样只要在需要连接数据库做操作的时候，只需要调用我们上面定义好的函数就可以了。 但是这样还是有问题，当我要大批量创建数据的时候，就需要多次调用create方法了，相当于多次连接多次提交。 可以继续优化下，把数据库的连接重用，做到只需一次连接就可执行多次操作。 </span><span style="color: #008000;">#</span><span style="color: #008000;"> 初始化实例方法</span>
<span style="color: #0000ff;">def</span> <span style="color: #800080;">__init__</span><span style="color: #000000;">(self):
self.conn </span>=<span style="color: #000000;"> None
self.cursor </span>=<span style="color: #000000;"> None
self.connect()
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 连接数据库</span>
<span style="color: #0000ff;">def</span><span style="color: #000000;"> connect(self):
self.conn </span>=<span style="color: #000000;"> pymysql.connect(
host</span>=DB_CONFIG[<span style="color: #800000;">"</span><span style="color: #800000;">host</span><span style="color: #800000;">"</span><span style="color: #000000;">],charset</span>=DB_CONFIG[<span style="color: #800000;">"</span><span style="color: #800000;">charset</span><span style="color: #800000;">"</span><span style="color: #000000;">]
)
self.cursor </span>= self.conn.cursor(cursor=<span style="color: #000000;">pymysql.cursors.DictCursor)
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 查询多条数据</span>
<span style="color: #0000ff;">def</span> get_list(self,sql,args=<span style="color: #000000;">None):
self.cursor.execute(sql,args)
result </span>=<span style="color: #000000;"> self.cursor.fetchall()
</span><span style="color: #0000ff;">return</span><span style="color: #000000;"> result
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 查询单条数据</span>
<span style="color: #0000ff;">def</span> get_one(self,args)
result </span>=<span style="color: #000000;"> self.cursor.fetchone()
</span><span style="color: #0000ff;">return</span><span style="color: #000000;"> result
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 执行单条SQL语句</span>
<span style="color: #0000ff;">def</span> moddify(self,args)
self.conn.commit()
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 创建单条记录的语句</span>
<span style="color: #0000ff;">def</span> create(self,args)
self.conn.commit()
last_id </span>=<span style="color: #000000;"> self.cursor.lastrowid
</span><span style="color: #0000ff;">return</span><span style="color: #000000;"> last_id
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 关闭数据库cursor和连接</span>
<span style="color: #0000ff;">def</span><span style="color: #000000;"> close(self):
self.cursor.close()
self.conn.close()</span></pre>
我们把我们数据库的相关操作都封装成一个类，在用到的时候，只需要生成一个实例，并对实例调用相应的操作方法就可以了。 db == db.get_list(= db.get_list(
但是，我如果要批量执行多个创建操作，虽然只建立了一次数据库连接但是还是会多次提交，可不可以改成一次连接，一次提交呢？ 可以，只需要用上pymysql的executemany()方法就可以了。 给我们的 SQLManager类添加一个批量执行的?multi_modify()方法就可以了。
multi_modify(self,args=
最后，我们每次操作完数据库之后都要手动关闭，可不可以写成自动关闭的呢？
联想到我们之前学过的文件操作，使用with语句可以实现缩进结束自动关闭文件句柄的例子。 我们来把我们的数据库连接类SQLManager类再优化下，使其支持with语句操作。 </span><span style="color: #008000;">#</span><span style="color: #008000;"> 初始化实例方法</span>
<span style="color: #0000ff;">def</span> <span style="color: #800080;">__init__</span><span style="color: #000000;">(self):
self.conn </span>=<span style="color: #000000;"> None
self.cursor </span>=<span style="color: #000000;"> None
self.connect()
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 连接数据库</span>
<span style="color: #0000ff;">def</span><span style="color: #000000;"> connect(self):
self.conn </span>=<span style="color: #000000;"> pymysql.connect(
host</span>=DB_CONFIG[<span style="color: #800000;">"</span><span style="color: #800000;">host</span><span style="color: #800000;">"</span><span style="color: #000000;">],args)
self.conn.commit()
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 执行多条SQL语句</span>
<span style="color: #0000ff;">def</span> multi_modify(self,args=<span style="color: #000000;">None):
self.cursor.executemany(sql,args)
self.conn.commit()
last_id </span>=<span style="color: #000000;"> self.cursor.lastrowid
</span><span style="color: #0000ff;">return</span><span style="color: #000000;"> last_id
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 关闭数据库cursor和连接</span>
<span style="color: #0000ff;">def</span><span style="color: #000000;"> close(self):
self.cursor.close()
self.conn.close()
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 进入with语句自动执行</span>
<span style="color: #0000ff;">def</span> <span style="color: #800080;">__enter__</span><span style="color: #000000;">(self):
</span><span style="color: #0000ff;">return</span><span style="color: #000000;"> self
</span><span style="color: #008000;">#</span><span style="color: #008000;"> 退出with语句块自动执行</span>
<span style="color: #0000ff;">def</span> <span style="color: #800080;">__exit__</span><span style="color: #000000;">(self,exc_type,exc_val,exc_tb):
self.close()</span></pre>
现阶段，我们只需要优化到这一步就可以，后面的项目实战中会继续优化。如使用数据库连接池等。 （编辑：李大同） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |