本文实例讲述了flask框架实现连接sqlite3数据库的方法。分享给大家供大家参考,具体如下:
1. 在flask文件夹中新建一个models.py文件用来定义模型,内部代码如下
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
import sqlite3 #导入sqlite3包 def get_conn(): #定义该函数用来连接数据库 return sqlite3.connect( "test.db" ) class User( object ): def __init__( self , id ,name): self . id = id self .name = name def save( self ): sql = "insert into user VALUES (?,?)" #sql语句 conn = get_conn() #连接数据库 cursor = conn.cursor() #定义一个游标 cursor.execute(sql,( self . id , self .name)) #执行sql语句 conn.commit() #提交数据库改动 cursor.close() #关闭游标 conn.close() #关闭数据库连接 ''' staticmethod相当于一个定义在类里面的函数,所以如果一个方法既不跟实例 相关也不跟特定的类相关,推荐将其定义为一个staticmethod,这样不仅使代 码一目了然,而且似的利于维护代码。 ''' @staticmethod def query(): sql = "select * from user" conn = get_conn() cursor = conn.cursor() rows = cursor.execute(sql) users = [] for row in rows: user = User(row[ 0 ],row[ 1 ]) users.append(user) conn.commit() cursor.close() conn.close() return users def __str__( self ): return 'id:{}--name:{}' . format ( self . id , self .name) #注此处的是点不是逗号 |
2. 在flask文件夹中新建一个manage.py文件
(1)导入flask_script
包,导入sqlite3包,导入models.py中定义的模型,导入flask中的应用,所以最终头部代码如下
1
2
3
4
|
from flask_script import Manager from learn_flask_script import app import sqlite3 from models import User |
(2)创建数据库,代码如下
1
2
3
4
5
6
7
8
9
|
@manager .command def init_db(): sql = "create table user (id INT,name TEXT)" conn = sqlite3.connect( "test.db" ) cursor = conn.cursor() cursor.execute(sql) conn.commit() cursor.close() conn.close() |
(3)保存数据,代码如下
1
2
3
4
|
@manager .command def save(): user = User( 1 , 'csdn' ) user.save() |
(4)查询数据,代码如下
1
2
3
4
5
|
@manager .command def query_all(): users = User.query() for user in users: print (user) |
整体代码如下
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
from flask_script import Manager from learn_flask_script import app import sqlite3 from models import User manager = Manager(app) @manager .command def init_db(): sql = "create table user (id INT,name TEXT)" conn = sqlite3.connect( "test.db" ) cursor = conn.cursor() cursor.execute(sql) conn.commit() cursor.close() conn.close() @manager .command def save(): user = User( 1 , 'zhangjia' ) user.save() @manager .command def query_all(): users = User.query() for user in users: print (user) if __name__ = = "__main__" : manager.run() |
3. 在终端中使用命令即可运行数据库的创建,添加以及查询
希望本文所述对大家Python程序设计有所帮助。
原文链接:https://blog.csdn.net/qq_38542085/article/details/79317287