detect unique violation for psycopg

mamba-org · Oct 28, 2022 · 7c7303b · 7c7303b
1 parent c0dec1f
commit 7c7303b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/plugins/quetz_sql_authenticator/quetz_sql_authenticator/api.py b/plugins/quetz_sql_authenticator/quetz_sql_authenticator/api.py
@@ -77,7 +77,8 @@ def _create(
         db.commit()
     except IntegrityError as e:
         db.rollback()
-        if "UNIQUE constraint failed: credentials.username" in str(e):
+        if "unique" in str(e).lower() and "INSERT INTO credentials" in str(e):
+            # This detection should work for both psycopg2 and sqlite3
             raise HTTPException(
                 status_code=status.HTTP_409_CONFLICT,
                 detail=f"""