解耦Flask-SQLAlchemy：实现应用外部数据库查询与模型复用

app/
    app.py        # 主Flask应用
    models.py     # ORM模型定义
    scheduled_tasks/
        remove_old_tokens.py # 外部脚本
instance/
    db.sqlite

# app/database.py
from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

# app/models.py
import uuid
from sqlalchemy import func # 确保导入func用于server_default
from .database import db # 从新的database.py导入db


def uuid_str():
    return str(uuid.uuid4())


class TokenBlocklist(db.Model):
    id = db.Column(
        db.String(36),
        primary_key=True,
        nullable=False,
        index=True,
        default=uuid_str
    )
    jti = db.Column(
        db.String(36),
        nullable=False,
        index=True
    )
    type = db.Column(
        db.String(10),
        nullable=False
    )
    created_at = db.Column(
        db.DateTime,
        nullable=False,
        server_default=func.now(),
        index=True
    )

# app/app.py
from flask import Flask
from app.database import db # 从app.database导入db

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite'
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False

db.init_app(app) # 将db实例与app绑定

with app.app_context():
    db.create_all() # 在应用上下文内创建所有表

# app/scheduled_tasks/remove_old_tokens.py
import sys
import os
from datetime import datetime, timedelta
from flask import Flask

# 将项目根目录添加到Python模块搜索路径，以便进行绝对导入
# 假设脚本位于 `app/scheduled_tasks/`，项目根目录在 `../../`
sys.path.append(os.path.abspath('../../'))

from app.database import db # 从app.database导入db
from app.models import TokenBlocklist # 从app.models导入ORM模型

# app/scheduled_tasks/remove_old_tokens.py (接上文)

def remove_old_tokens():
    forty_days = timedelta(days=40)
    forty_days_ago = datetime.now() - forty_days
    # 使用ORM模型进行查询和删除操作
    query = TokenBlocklist.__table__.delete().where(
        TokenBlocklist.created_at < forty_days_ago
    )
    db.session.execute(query)
    db.session.commit()
    print('旧令牌已删除')


# 在脚本中创建一个临时的Flask应用实例
app = Flask(__name__)
# 配置数据库URI，注意路径需要根据脚本位置调整
# 假设db.sqlite在instance/目录下，相对于脚本是../../instance/db.sqlite
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' \
    + os.path.abspath('../../instance/db.sqlite')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db.init_app(app) # 将db实例与这个临时的app绑定

# 所有数据库操作必须在应用上下文中执行
with app.app_context():
    db.create_all() # 确保表存在 (在生产环境可能由迁移工具处理)
    remove_old_tokens() # 调用数据库操作函数