fix: replace all dataset.Model.query to db.session.query(Model) (#19509)

2025-05-12 13:52:33 +08:00
parent 49af07f444
commit b00f94df64
21 changed files with 430 additions and 265 deletions
--- a/api/schedule/create_tidb_serverless_task.py
+++ b/api/schedule/create_tidb_serverless_task.py
@@ -19,7 +19,9 @@ def create_tidb_serverless_task():
    while True:
        try:
            # check the number of idle tidb serverless
-            idle_tidb_serverless_number = TidbAuthBinding.query.filter(TidbAuthBinding.active == False).count()
+            idle_tidb_serverless_number = (
+                db.session.query(TidbAuthBinding).filter(TidbAuthBinding.active == False).count()
+            )
            if idle_tidb_serverless_number >= tidb_serverless_number:
                break
            # create tidb serverless