Access GCP Cloud SQL from AI notebook?

user_13142245 0
Viewed 674

Looking for relevant python code to be used in the GCP AI Platform notebook which will be able to query GCP Cloud SQL (specifically Postgresql.) Unfortunately, I haven't found any relevat resources/tutorials from GCP official or even unaffiliated resources.

gcp-ai-platform-notebook google-cloud-platform postgresql python
1 Answers

I work on the Developer Relations team for Cloud SQL at Google. The quickest and simplest way to connect to a Cloud SQL instance from an AI Platform Notebook is to use our newly released Cloud SQL Python Connector. To install, run pip install cloud-sql-python-connector[pg8000]. This will also install the pg8000 driver which is used to connect to Postgres.

Then, you can use the following code to get a connection object which you can use to run queries:

conn = connector.connect(
    "project_name:region-name:instance-name", 
    "pg8000",
    user="postgres",
    password="password",
    db="db_name"
)

An alternative, which is language-independent, is to download and run the Cloud SQL Auth Proxy in a separate terminal from your notebook.

For either of these solutions, make sure to enable the Cloud SQL Admin API first.

user_117153990
Related