3737from google .cloud .sql .connector .lazy import LazyRefreshCache
3838from google .cloud .sql .connector .monitored_cache import MonitoredCache
3939import google .cloud .sql .connector .pg8000 as pg8000
40+ import google .cloud .sql .connector .psycopg as psycopg
4041import google .cloud .sql .connector .pymysql as pymysql
4142import google .cloud .sql .connector .pytds as pytds
4243from google .cloud .sql .connector .resolver import DefaultResolver
@@ -230,7 +231,7 @@ def connect(
230231 Example: "my-project:us-central1:my-instance"
231232
232233 driver (str): A string representing the database driver to connect
233- with. Supported drivers are pymysql, pg8000, and pytds.
234+ with. Supported drivers are pymysql, pg8000, psycopg, and pytds.
234235
235236 **kwargs: Any driver-specific arguments to pass to the underlying
236237 driver .connect call.
@@ -266,7 +267,8 @@ async def connect_async(
266267 Example: "my-project:us-central1:my-instance"
267268
268269 driver (str): A string representing the database driver to connect
269- with. Supported drivers are pymysql, asyncpg, pg8000, and pytds.
270+ with. Supported drivers are pymysql, asyncpg, pg8000, psycopg, and
271+ pytds.
270272
271273 **kwargs: Any driver-specific arguments to pass to the underlying
272274 driver .connect call.
@@ -278,7 +280,7 @@ async def connect_async(
278280 ValueError: Connection attempt with built-in database authentication
279281 and then subsequent attempt with IAM database authentication.
280282 KeyError: Unsupported database driver Must be one of pymysql, asyncpg,
281- pg8000, and pytds.
283+ pg8000, psycopg, and pytds.
282284 """
283285 if self ._keys is None :
284286 self ._keys = asyncio .create_task (generate_keys ())
@@ -332,6 +334,7 @@ async def connect_async(
332334 connect_func = {
333335 "pymysql" : pymysql .connect ,
334336 "pg8000" : pg8000 .connect ,
337+ "psycopg" : psycopg .connect ,
335338 "asyncpg" : asyncpg .connect ,
336339 "pytds" : pytds .connect ,
337340 }
0 commit comments