Write azd env

Author: pamelafox

.env.azure

@@ -1,4 +1,5 @@
-# Use these values to connect to the Azure database from within the devcontainer
+# Set these values to connect to the Azure database
+# Use write_azure_env.sh or write_azure_env.ps1 to set these values
 POSTGRES_DATABASE="db"
 POSTGRES_HOST="YOUR-SERVER-NAME.postgres.database.azure.com"
 POSTGRES_SSL="require"

README.md

@@ -100,10 +100,10 @@ Follow these steps to deploy a PostgreSQL Flexible Server to Azure with the pgve
 
  This will create a new resource group, and create the PostgreSQL Flexible server inside that group.
 
-1. The example Python scripts look for configuration variables from a `.env` file located in the directory from where you invoke the scripts. You can easily create a file with the correct variables for your PostgreSQL server by running this command that copies the `azd` environment variables into your local `.env`:
+1. The example Python scripts look for configuration variables from a `.env` file located in the directory from where you invoke the scripts. You can easily create a file with the correct variables for your PostgreSQL server by running this script that copies the necessary `azd` environment variables into your local `.env`:
 
  ```shell
- azd env get-values > .env
+ ./write_azure_env.sh
  ```
 
 1. Now you may run the Python scripts in order to interact with the PostgreSQL server.

examples/asyncpg_items.py

@@ -23,6 +23,9 @@ async def async_main():
  POSTGRES_PASSWORD = os.environ["POSTGRES_PASSWORD"]
 
  DATABASE_URI = f"postgresql://{POSTGRES_USERNAME}:{POSTGRES_PASSWORD}@{POSTGRES_HOST}/{POSTGRES_DATABASE}"
+ # Specify SSL mode if needed
+ if POSTGRES_SSL := os.environ.get("POSTGRES_SSL"):
+ DATABASE_URI += f"?sslmode={POSTGRES_SSL}"
 
  conn = await asyncpg.connect(DATABASE_URI)
 

write_azure_env.ps1

@@ -0,0 +1,13 @@
+# Clear the contents of the .env file
+Set-Content -Path .env -Value ""
+
+# Append new values to the .env file
+$postgresDatabase = azd env get-value POSTGRES_DATABASE
+$postgresHost = azd env get-value POSTGRES_HOST
+$postgresSSL = azd env get-value POSTGRES_SSL
+$postgresUsername = azd env get-value POSTGRES_USERNAME
+
+Add-Content -Path .env -Value "POSTGRES_DATABASE=$postgresDatabase"
+Add-Content -Path .env -Value "POSTGRES_HOST=$postgresHost"
+Add-Content -Path .env -Value "POSTGRES_SSL=$postgresSSL"
+Add-Content -Path .env -Value "POSTGRES_USERNAME=$postgresUsername"

write_azure_env.sh

@@ -0,0 +1,10 @@
+#!/bin/bash
+
+# Clear the contents of the .env file
+> .env
+
+# Append new values to the .env file
+echo "POSTGRES_DATABASE=$(azd env get-value POSTGRES_DATABASE)" >> .env
+echo "POSTGRES_HOST=$(azd env get-value POSTGRES_HOST)" >> .env
+echo "POSTGRES_SSL=$(azd env get-value POSTGRES_SSL)" >> .env
+echo "POSTGRES_USERNAME=$(azd env get-value POSTGRES_USERNAME)" >> .env