Postgres dialect

greg-finley · Apr 11, 2024 · 6678075 · 6678075
1 parent 1c5a23c
commit 6678075
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/process_file/main.py b/process_file/main.py
@@ -25,8 +25,8 @@ def run(event, context):
     )
     query = """
     INSERT INTO dropbox (desktop_path, filename, status)
-    VALUES (%s, SUBSTRING_INDEX(%s, '/', -1), 'done')
-    ON DUPLICATE KEY UPDATE status = 'done'
+    VALUES (%s, SPLIT_PART(%s, '/', -1), 'done')
+    ON CONFLICT (desktop_path) DO UPDATE SET status = 'done'
     """
     with psycopg.connect(os.environ["NEON_DATABASE_URL"]) as conn:
         with conn.cursor() as cursor:

diff --git a/queue_files/main.py b/queue_files/main.py
@@ -58,7 +58,7 @@ def run(event, context):
             print(f"Queueing {clean_name}")
             query = """
             INSERT INTO dropbox (desktop_path, filename, status)
-            VALUES (%s, SUBSTRING_INDEX(%s, '/', -1), 'pending')
+            VALUES (%s, SPLIT_PART(%s, '/', -1), 'pending')
             ON CONFLICT (desktop_path) DO NOTHING
             """
             with conn.cursor() as cursor: