Add psycopg2 extra, patch psycogreen in entry points.

bin/baseplate-script

@@ -12,6 +12,15 @@ from baseplate.server.monkey import patch_stdlib_queues
 patch_all()
 patch_stdlib_queues()
 
+try:
+    import psycopg2
+    from psycogreen.gevent import patch_psycopg
+
+    patch_psycopg()
+except ImportError:
+    pass
+
+
 from baseplate.server import load_and_run_script
 
 load_and_run_script()

bin/baseplate-serve

@@ -12,6 +12,15 @@ from baseplate.server.monkey import patch_stdlib_queues
 patch_all()
 patch_stdlib_queues()
 
+try:
+    import psycopg2
+    from psycogreen.gevent import patch_psycopg
+
+    patch_psycopg()
+except ImportError:
+    pass
+
+
 from baseplate.server import load_app_and_run_server
 
 load_app_and_run_server()

bin/baseplate-shell

@@ -12,6 +12,15 @@ from baseplate.server.monkey import patch_stdlib_queues
 patch_all()
 patch_stdlib_queues()
 
+try:
+    import psycopg2
+    from psycogreen.gevent import patch_psycopg
+
+    patch_psycopg()
+except ImportError:
+    pass
+
+
 from baseplate.server import load_and_run_shell
 
 load_and_run_shell()

bin/baseplate-tshell

@@ -12,6 +12,15 @@ from baseplate.server.monkey import patch_stdlib_queues
 patch_all()
 patch_stdlib_queues()
 
+try:
+    import psycopg2
+    from psycogreen.gevent import patch_psycopg
+
+    patch_psycopg()
+except ImportError:
+    pass
+
+
 from baseplate.server import load_and_run_shell
 
 print("baseplate-tshell has been renamed to baseplate-shell. Please use that in the future!")

pyproject.toml

@@ -55,6 +55,7 @@ requests = ["advocate"]
 s3fetcher = []  # Kept for backwards compatibility, these are now main requirements
 sentry = ["sentry-sdk"]
 sql = ["sqlalchemy"]
+psycopg2 = ["psycopg2", "psycogreen"]
 zookeeper = ["kazoo"]
 
 [tool.poetry.group.dev.dependencies]