Python的flask如何链接数据库
时间: 2024-05-03 13:22:26 浏览: 132
Flask可以使用多种方式链接数据库,以下是其中两种常用的方式:
1. 使用SQLAlchemy库链接数据库
首先需要安装SQLAlchemy库,可以使用以下命令安装:
```
pip install sqlalchemy
```
然后在Flask应用中添加以下代码:
```python
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
db = SQLAlchemy(app)
```
其中,`SQLALCHEMY_DATABASE_URI`是数据库的链接地址,这里使用SQLite数据库作为例子。如果使用其他数据库,需要将链接地址改为对应的格式。
2. 使用Flask自带的SQLite库链接数据库
Flask自带了一个SQLite库,可以直接在Flask应用中使用。以下是链接SQLite数据库的代码:
```python
import sqlite3
from flask import Flask, g
app = Flask(__name__)
DATABASE = 'database.db'
def get_db():
db = getattr(g, '_database', None)
if db is None:
db = g._database = sqlite3.connect(DATABASE)
return db
@app.teardown_appcontext
def close_connection(exception):
db = getattr(g, '_database', None)
if db is not None:
db.close()
```
在上述代码中,`get_db()`函数用于获取数据库连接,`close_connection()`函数用于关闭数据库连接。在Flask应用中,可以通过以下方式使用数据库:
```python
@app.route('/')
def index():
cur = get_db().cursor()
cur.execute('SELECT * FROM users')
users = cur.fetchall()
return str(users)
```
以上代码执行一个SELECT语句,并返回查询结果。可以根据需要修改SELECT语句或者执行其他SQL语句。
阅读全文