Flask：操作SQLite3（0.1）

1 DROP TABLE IF EXISTS post;
2 CREATE TABLE post (
3     id INTEGER PRIMARY KEY AUTOINCREMENT,
4     body VARCHAR(500) NOT NULL,
5     created TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
6 );

 1 from flask import * # 导入flask模块下的所有元素
 2 import sqlite3 # 导入sqlite3模块
 3 
 4 app = Flask('Posts') # 建立Flask应用
 5 
 6 DATABASE = 'db/posts.db' # 数据库文件地址
 7 DATABASE_INIT_FILE = 'db/init.sql' # 数据库建模文件地址
 8 
 9 # init_db()
10 def init_db(): # 此函数要使用 数据库建模文件 初始化数据库：建立post表（Initial Schemas）。此函数在命令行中使用，仅一次，再次使用会删除数据库中已有数据。
11     with app.app_context(): # 官网文档说了原因：不是在Web应用中使用，而是在Python Shell中使用时需要此语句（Connect on Demand）。
12         db = get_db()
13         with app.open_resource(DATABASE_INIT_FILE, mode='r') as f: # with语句用法！
14             db.cursor().executescript(f.read()) # 执行建模文件中的脚本
15         db.commit() # 提交事务
16 
17 # make_dicts()
18 def make_dicts(cursor, row): # 将查询返回的数据的转换为字典类型，这样会跟方便使用。此函数会在get_db()函数中用到，赋值给db.row_factory。
19     return dict((cursor.description[idx][0], value)
20                 for idx,value in enumerate(row))
21 
22 # get_db()
23 def get_db(): # 获取数据库连接
24     db = getattr(g, '_database', None) # g对象时一个Flask应用的公共对象（和request、session一样），用于存储用户的数据——整个应用共享！
25     if db is None:
26         db = g._database = sqlite3.connect(DATABASE) # 建立数据库连接
27         db.row_factory = make_dicts # 转换默认的查询数据类型为字典类型，也可以使用sqlite3.Row
28     return db # 返回数据库连接，可能返回为None
29 
30 # close_connection()
31 @app.teardown_appcontext # 这个装饰器用于实现在请求的最后自动关闭数据库连接的功能
32 def close_connection(exception): # 关闭数据库连接
33     db = getattr(g, '_database', None)
34     if db is not None:
35         db.close()

1 @app.route('/testdb')
2 def testdb():
3     get_db()
4     return "After test"

1 # path: /
2 # show posts
3 @app.route('/')
4 def home():
5     return render_template('temp.html')

1 <form action="pureadd" method="post">
2     <textarea name="newpost" style="width:200px;height:100px;" maxlength="500">textarea><br />
3     <input type="submit" value="添加" /> 
4 form>
5 <a href="/pureshow">展示帖子a><br />

 1 # pure add page for test
 2 @app.route('/pureadd', methods=['POST'])
 3 def pureadd():
 4     # step 1. get the new post and check the data
 5     newp = ''
 6     try:
 7         newp = request.form['newpost']
 8     except:
 9         return 'ERROR: Invalid form parameters!'
10     
11     print('newp = "', newp, '"')
12     
13     newp2 = newp.strip() # 清理post两遍的空格，然后赋值给新变量——此时旧变量没有改变
14     
15     if newp2 == '':
16         return 'Warning: New post is empty!
' \
17             '继续添加
' \
18             '展示帖子
'
19     
20     # step 2.write the new post into database
21     sqlmode = 'INSERT INTO post(body) VALUES(?)' # 添加数据的SQL语句，占位符使用问号（?）。需要注意的是，如果是MySQL，占位符是百分号（%）。
22     try:
23         db = get_db()
24         cursor = db.cursor()
25         cursor.execute(sqlmode, (newp2,))
26         cursor.close() # 关闭cursor。Q：是否一定要关闭呢？不关闭有什么影响？
27         db.commit() # 需要commit，否则，数据不会更新到数据库 28     except Exception as e:
29         return 'INFO: New post added failed 
 %s
' \
30             '继续添加
' \
31             '展示帖子
' % str(e)
32     else:
33         return 'INFO: New post added 
[%s]

' \
34             '继续添加
' \
35             '展示帖子
' % newp

 1 # pure show page for test
 2 @app.route('/pureshow')
 3 def pureshow():
 4     sqlmode = "SELECT * FROM post ORDER BY created DESC"
 5     rv = []
 6     try:
 7         db = get_db()
 8         cursor = db.execute(sqlmode)
 9         rv = cursor.fetchall()
10         cursor.close()
11     except Exception as e:
12         print(e)
13         abort(500)
14     else:
15         return render_template('showall.html', posts=rv)

1 <div id="postslist">
2     {% for post in posts %}
3         <div class="post_item" id="post{{ post.id }}">
4             {{ post.body }}<br/>
5             {{ post.created }}
6         div>
7     {% endfor %}
8 div>
9 <a href="/">继续添加a>