Browse Source

Feat: return newly inserted rowid for db insertion

dk 1 year ago
parent
commit
ef84023db2
3 changed files with 9 additions and 1 deletions
  1. 4 0
      backend/db/models/test.py
  2. 4 0
      backend/db/models/train.py
  3. 1 1
      backend/settings/config.py

+ 4 - 0
backend/db/models/test.py

@@ -25,3 +25,7 @@ def create_test(conn, test_form):
                         model_path=test_form['model_path'])
         )
         s.commit()
+        test_id = conn.query('select rowid from test where start_time = :start_time and owner_name =:owner_name', 
+                            params={'start_time': test_form['start_time'], 
+                                    'owner_name': test_form['owner_name']})
+        return test_id

+ 4 - 0
backend/db/models/train.py

@@ -27,3 +27,7 @@ def create_train(conn, train_form):
                         model_path=train_form['model_path'])
         )
         s.commit()
+    train_id = conn.query('select rowid from train where start_time = :start_time and owner_name =:owner_name', 
+                            params={'start_time': train_form['start_time'], 
+                                    'owner_name': train_form['owner_name']})
+    return train_id

+ 1 - 1
backend/settings/config.py

@@ -29,7 +29,7 @@ class Settings:
         ]
     }
     PROJECT_VERSION: str = '0.0.1'
-    DATA_PATH = './db/data'
+    DATA_PATH = './data'
 
     def __init__(self):
         self.config = {"lang": "zh"}