fix db session

2023-10-18 18:30:09 +08:00
parent 68e16d18fe
commit 2e4536edb6
10 changed files with 48 additions and 48 deletions
--- a/app/db/models/site.py
+++ b/app/db/models/site.py
@ -3,7 +3,7 @@ from datetime import datetime
 from sqlalchemy import Boolean, Column, Integer, String, Sequence
 from sqlalchemy.orm import Session

-from app.db.models import Base
+from app.db.models import Base, db_persist


 class Site(Base):
@ -59,6 +59,6 @@ class Site(Base):
        return db.query(Site).order_by(Site.pri).all()

    @staticmethod
+    @db_persist
    def reset(db: Session):
        db.query(Site).delete()
-        Base.commit(db)