Flask连接数据库打怪升级之旅

前言

在初学 Flask 的时候，在数据库连接这部分也跟每个初学者一样。但是随着工作中项目接手的多了，代码写的多了，历练的多了也就有了自己的经验和技巧。在对这块儿代码不断的进行升级改造后，整理了自己在连接数据库这部分的的一个学习经验，也就是我们今天分享的连接数据库部分的打怪升级之旅。希望可以为大家在学习 Python 的路上提供一些参考。

初级阶段

首先安装 Mysql 扩展包

yum install python-dev
pip install MySQL-python

建立数据库链接

import MySQLdb as mysql

db=mysql.connect(user='root',passwd='123456',db='reboot',charset='utf8')
cur=db.cursor()

开启打怪升级之路

在日常开发中，连接数据库最多的应用场景就是，查询所有数据和查询单条数据。就以查询所有数据场景为例。

小白版本——在后端凭接表格，传到前端渲染

cur.execute('select * from host') 
str = ""

# 结果以元组嵌套的形式返回 ((id,name), (id,iname),……）
result= cur.fetchall():

# 循环父元祖，然后按照索引依次获取到每列的值，
for c in result: 
    str = '<tr><td>%s</td> \
    <td>%s</td> \
    <td>%s</td> \
    <td>%s</td> \
    <td><span class="update-btn" data-id="%s">update</span></td> \
    <td><span class="delete-btn" data-id="%s">delete</span></td> \
    </tr>' % (c[0],c[1],c[2],c[3],c[0],c[0])
    str ＋= str

cur.close()
return jsondumps(str)

进阶阶段

第一关——后端消灭 HTML 标签

后端：

# 执行sql
cur.execute('select * from user') 
result = cur.fetchall(): 
return render_template("index.html",result=result)

前端：

<table border="1px">
<tr>
    <td>ID</td>
    <td>user</td>
    <td>passwd</td>
    <td>sex</td>
    <td>phone</td>
    <td>操作</td>
</tr>
{% for user in result %}
<tr>
    <td>{{user[0]}}</td>
    <td>{{user[1]}}</td>
    <td>{{user[2]}}</td>
    <td>{{user[3]}}</td>
    <td>{{user[5]}}</td>
</tr>
{% endfor %}
</table>

第二关——让返回值更优雅

users = []
# 指定需要返回的列，存入列表
fields = ['id', 'username', 'name', 'email', 'mobile'] 

# join将列表转为字符串，传入sql
sql = "SELECT %s FROM user" % ','.join(fields) 
cur.execute(sql)

# 循环返回的元组
for row in cur.fetchall(): 
   user = {}
   # 利用enumerate遍历列表,i为索引号，k为元素
   # fields中列作为user字典的k,索引作为mysql返回列表的k,给字典赋值
   for i, k in enumerate(fields): 
        user[k] = row[i] 
        users.append(user)
        
# 返回结果时列表嵌套字典，前端渲染就更优雅了
# eg:[{'name':'wd','age':18},{'name':'pc','age':19},……]
return json.dumps({'code': 0, 'users': users})

第三关——让代码更简洁

fields = ['id','username','name','email','mobile']

result= cur.fetchall()
users＝ [dict((k, row[i]) for i,k in enumerate(fields)) for row in result]

return json.dumps({'code':0,'users':users})

一个更高效的方式——直接将返回的嵌套元祖转换为嵌套的字典,常用与只查询 ID, Username 的场景

result = cur.fetchall() 
# 最终结果 {1: 'ab', 2: 'bc'}
result = dict([(x['0'], x['1']) for x in result])

经验总结

作为一个程序员学习新的技术知识都是必须的，我们都是自己事业上无人可替的开拓者，我们都是要经历从入门到熟练再到精略的过程，过程虽然很痛苦不过收获的喜悦也是别人羡慕不来的，IT 大牛 不是那么容易就练成的。希望今天的分享能够帮助到大家。致每一位程序员