From 3e29233dbdcd991d9eeda87df1f1caf4709b3de7 Mon Sep 17 00:00:00 2001 From: Eunhak Lee <lee@enak.kr> Date: Wed, 4 Dec 2024 17:39:41 +0900 Subject: [PATCH] =?UTF-8?q?chore:=20parts=20=ED=85=8C=EC=9D=B4=EB=B8=94?= =?UTF-8?q?=EC=97=90=20=EB=84=A3=EC=9D=84=20=EB=95=8C=20ON=20CONFLICT=20?= =?UTF-8?q?=EB=8F=99=EC=9E=91=20=EC=A0=95=EC=9D=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- push_to_db.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/push_to_db.py b/push_to_db.py index 52b7c94..043f72f 100644 --- a/push_to_db.py +++ b/push_to_db.py @@ -61,10 +61,9 @@ def create_new_one(_dict, part_type, _cursor): columns = ", ".join([f'"{key}"' for key, _ in _obj]) values = [value for _, value in _obj] - sql = """WITH rows AS (INSERT INTO parts (name, type) VALUES (%s, %s) RETURNING id) + sql = """WITH rows AS (INSERT INTO parts (name, type) VALUES (%s, %s) ON CONFLICT (name) DO UPDATE SET name = excluded.name RETURNING id) INSERT INTO part_info_{type} (part_id, {columns}) (SELECT rows.id, {value_entries} FROM rows) -ON CONFLICT (name) DO UPDATE name = excluded.name RETURNING part_id;""".format(type=_part_type_enum, columns=columns, value_entries=", ".join(["%s"] * len(values))) _cursor.execute(sql, (name, _part_type_enum, *values)) data = _cursor.fetchall() @@ -75,7 +74,7 @@ RETURNING part_id;""".format(type=_part_type_enum, columns=columns, value_entrie def get_user_by_email(_cursor, _email): - _cursor.execute("SELECT id FROM users WHERE email = %s;", (_email,)) + _cursor.execute("SELECT id FROM usersd WHERE email = %s;", (_email,)) data = _cursor.fetchall() if data: -- GitLab