计划SQLAlchemy清除表中的所有行

Question

我正在尝试创建一个可以计划删除SQLAlchemy模型内所有行的函数。

我正在尝试使用apscheduler完成此任务。但是我不断收到错误消息：

sqlalchemy.orm.exc.UnmappedInstanceError: Class 'flask_sqlalchemy.model.DefaultMeta' is
 not mapped; was a class (app.models.User) supplied where an instance was required?

我想念什么吗？

这里是我的app/__init__.py：

from flask import Flask
from flask_login import LoginManager
from flask_sqlalchemy import SQLAlchemy
from config import Config

app = Flask(__name__)

db = SQLAlchemy()
login = LoginManager()

app.config.from_object(Config)
db.init_app(app)
login.init_app(app)
login.login_view = 'login'

from app import routes, models

这是我的manage.py：

from apscheduler.schedulers.background import BackgroundScheduler
from app import app, db
from flask_migrate import Migrate
from flask_script import Manager
from app.models import User

manager = Manager(app)

migrate = Migrate(app, db)

def clear_data():
    db.session.delete(User)
    print("Deleted User table!")

@manager.command
def run():
    scheduler = BackgroundScheduler()
    scheduler.add_job(clear_data, trigger='interval', seconds=5)
    scheduler.start()
    app.run(debug=True)

if __name__ == '__main__':
    manager.run()

另外，这是我的模特：

from app import db, login
from datetime import datetime
from flask_login import UserMixin
from werkzeug.security import generate_password_hash, check_password_hash

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True, unique=True)
    api_token = db.Column(db.String(50), unique=True)
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    password_hash = db.Column(db.String(128))
    todos = db.relationship('Todo', backref='owner', lazy='dynamic')

    def __repr__(self):
        return '<models.py {}>'.format(self.username)

    def set_password(self, password):
        self.password_hash = generate_password_hash(password)

    def check_password(self, password):
        return check_password_hash(self.password_hash, password)


@login.user_loader
def load_user(id):
    return User.query.get(int(id))


class Todo(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    body = db.Column(db.String(140))
    timestamp = db.Column(db.DateTime, index=True, default=datetime.utcnow)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))

    def __repr__(self):
        return '<Todo {}>'.format(self.body)

Answer 1

如您的错误所示，它期望一个对象的实例，而是您将其传递给一个类。我认为问题是clear_data函数的第一行：

db.session.delete(User)

原本希望删除User记录的实例，但不知道如何仅使用模型来删除整个表。

Check out this answer on how to delete all rows in a table.有几种方法可以执行此操作，但这对您来说可能是最小的更改：

db.session.query(User).delete()

在这种情况下，您将添加以下步骤：SELECT将用户映射到的表中的所有记录删除，然后将它们删除。

P.S .：如链接答案中所述，您需要.commit()您的会话，否则它不会卡住，并且在关闭连接后会回滚。

db.session.commit()

Answer 2

代码段：

db.session.query(model_name).delete() 
db.session.commit()