소스 검색

Update insert_or_get to match postgres' approach

jherve 1 년 전
부모
커밋
f1a4b45338
1개의 변경된 파일2개의 추가작업 그리고 3개의 파일을 삭제
  1. 2 3
      src/de_quoi_parle_le_monde/storage.py

+ 2 - 3
src/de_quoi_parle_le_monde/storage.py

@@ -609,10 +609,9 @@ class Storage:
         select_stmt: str,
         select_args: list[Any],
     ) -> int:
-        (id_,) = await conn.execute_insert(insert_stmt, insert_args)
+        await conn.execute_insert(insert_stmt, *insert_args)
 
-        if id_ == 0:
-            [(id_,)] = await conn.execute_fetchall(select_stmt, select_args)
+        [(id_,)] = await conn.execute_fetchall(select_stmt, *select_args)
 
         return id_