flask连接sqlite3数据库

#1.在flask文件夹中新建一个models.py文件用来定义模型，内部代码如下

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中的应用，所以最终头部代码如下

from flask_script import Manager
from learn_flask_script import app
import sqlite3
from models import User

---

###（2）创建数据库，代码如下

@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）保存数据，代码如下

@manager.command
def save():
    user = User(1,'csdn')
    user.save()

---

###（4）查询数据，代码如下

@manager.command
def query_all():
    users = User.query()
    for user in users:
        print(user)

---

###整体代码如下

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.在终端中使用命令即可运行数据库的创建，添加以及查询