flask连接sqlite3数据库

妖狐艹你老母 2022-05-31 07:57 274阅读 0赞

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