From a85a9648199e3813e30be2bfd2f48a37b45bbb91 Mon Sep 17 00:00:00 2001
From: Ruben Fiszel <ruben@rubenfiszel.com>
Date: Fri, 13 Jan 2023 07:54:05 +0100
Subject: [PATCH] add psycopg3 to the list of support imports

---
 backend/parsers/windmill-parser-py/src/lib.rs | 1 +
 lsp/Pipfile                                   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/backend/parsers/windmill-parser-py/src/lib.rs b/backend/parsers/windmill-parser-py/src/lib.rs
index fac1c78810..a7e692c098 100644
--- a/backend/parsers/windmill-parser-py/src/lib.rs
+++ b/backend/parsers/windmill-parser-py/src/lib.rs
@@ -174,6 +174,7 @@ fn constant_to_value(c: &Constant) -> serde_json::Value {
 
 static PYTHON_IMPORTS_REPLACEMENT: phf::Map<&'static str, &'static str> = phf_map! {
     "psycopg2" => "psycopg2-binary",
+    "psycopg" => "psycopg[binary]",
     "yaml" => "pyyaml",
     "git" => "GitPython",
     "u" => "requests",
diff --git a/lsp/Pipfile b/lsp/Pipfile
index 680732a931..eade2c4281 100644
--- a/lsp/Pipfile
+++ b/lsp/Pipfile
@@ -22,6 +22,7 @@ numpy = "*"
 pyyaml = "*"
 toml = "*"
 psycopg2-binary = "*"
+psycopg = "*"
 awscli = "*"
 python-dateutil = "*"
 pytz = "*"