From 4d29d16b64c43bb40b8f399b90296a12e7f66426 Mon Sep 17 00:00:00 2001
From: Elizabeth Thompson <eschutho@gmail.com>
Date: Fri, 29 Jul 2022 13:03:41 -0700
Subject: [PATCH] fix: make a new entrypoint for deprecated postgres dialect
 (#20895)

* make a new entrypoint for deprecated postgres dialect

* Update setup.py

Co-authored-by: Beto Dealmeida <roberto@dealmeida.net>

Co-authored-by: Beto Dealmeida <roberto@dealmeida.net>
---
 setup.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/setup.py b/setup.py
index b62dceb47f07..5f1b7b7fd269 100644
--- a/setup.py
+++ b/setup.py
@@ -64,6 +64,11 @@ def get_git_sha() -> str:
     zip_safe=False,
     entry_points={
         "console_scripts": ["superset=superset.cli.main:superset"],
+        # the `postgres+psycopg2://` scheme was removed in SQLAlchemy 1.4, add an alias here
+        # to prevent breaking existing databases
+        "sqlalchemy.dialects": [
+            "postgres.psycopg2=sqlalchemy.dialects.postgresql:dialect"
+        ],
     },
     install_requires=[
         "backoff>=1.8.0",