diff --git a/push_to_db.py b/push_to_db.py index 3e40f8c6026b357e65ac9a83a9785afe11f15191..52b7c94e5586d45a910194bf90904000b292c267 100644 --- a/push_to_db.py +++ b/push_to_db.py @@ -62,7 +62,10 @@ def create_new_one(_dict, part_type, _cursor): values = [value for _, value in _obj] sql = """WITH rows AS (INSERT INTO parts (name, type) VALUES (%s, %s) RETURNING id) -INSERT INTO part_info_{type} (part_id, {columns}) (SELECT rows.id, {value_entries} FROM rows) RETURNING part_id;""".format(type=_part_type_enum, columns=columns, value_entries=", ".join(["%s"] * len(values))) +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()