计划 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)

它期望删除用户记录的实例，并且不知道如何仅使用模型删除整个表。

查看有关如何删除表中的所有行的答案。有几种方法可以做到这一点，但这对你来说可能是最小的变化：

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

在本例中，您将添加将用户映射到表中的所有记录SELECT，然后删除它们的步骤。

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

db.session.commit()

Answer 2

代码片段：

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