Add environment variable setup for CI workflow based on branch

Author: khavnekar-y

.github/workflows/snowpark-ci-cd.yml

@@ -20,6 +20,18 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
+          - name: Set environment variables
+          run: |
+            echo "AWS_ACCESS_KEY=${{ secrets.AWS_ACCESS_KEY }}" >> $GITHUB_ENV
+            echo "AWS_SECRET_KEY=${{ secrets.AWS_SECRET_KEY }}" >> $GITHUB_ENV
+            if [[ "${{ github.ref }}" == "refs/heads/main" ]]; then
+              echo "ENVIRONMENT=prod" >> $GITHUB_ENV
+              echo "DATABASE_NAME=CO2_DB_PROD" >> $GITHUB_ENV
+              echo "CONN_PROFILE=prod" >> $GITHUB_ENV
+            else
+              echo "ENVIRONMENT=dev" >> $GITHUB_ENV
+              echo "DATABASE_NAME=CO2_DB_DEV" >> $GITHUB_ENV
+              echo "CONN_PROFILE=dev" >> $GITHUB_ENV          
       - name: Debug Environment Variables
         run: |
           pytest tests/testing_snowflake_connection.py