Megosztás a következőn keresztül:


Rövid útmutató: Adatok csatlakoztatása és lekérdezése a Python használatával az Azure Database for PostgreSQL-ben

Ebben a rövid útmutatóban egy rugalmas Azure Database for PostgreSQL-kiszolgálópéldányhoz csatlakozhat a Python használatával. Ezután SQL-utasításokkal lekérdezheti, beszúrhatja, frissítheti és törölheti az adatbázisban lévő adatokat macOS, Ubuntu Linux és Windows platformokról.

A cikkben szereplő lépések két hitelesítési módszert tartalmaznak: a Microsoft Entra-hitelesítést és a PostgreSQL-hitelesítést. A Jelszó nélküli lapon a Microsoft Entra-hitelesítés, a Jelszó lapon pedig a PostgreSQL-hitelesítés látható.

A Microsoft Entra-hitelesítés az Azure Database for PostgreSQL-hez való csatlakozás mechanizmusa a Microsoft Entra ID-ban meghatározott identitások használatával. A Microsoft Entra-hitelesítéssel központi helyen kezelheti az adatbázis felhasználói identitásait és egyéb Microsoft-szolgáltatásait, ami leegyszerűsíti az engedélykezelést. További információ: Microsoft Entra-hitelesítés az Azure Database for PostgreSQL-lel.

A PostgreSQL-hitelesítés a PostgreSQL-ben tárolt fiókokat használja. Ha jelszavakat használ a fiókok hitelesítő adataiként, ezeket a hitelesítő adatokat a rendszer a user táblában tárolja. Mivel ezek a jelszavak a PostgreSQL-ben vannak tárolva, saját maga kell kezelnie a jelszavak rotálását.

Ez a cikk feltételezi, hogy ismeri a Python használatát, de még nem használta az Azure Database for PostgreSQL-t.

Prerequisites

Tűzfalszabályok hozzáadása az ügyfél-munkaállomáshoz

  • Ha rugalmas Azure Database for PostgreSQL-kiszolgálópéldányt hozott létre privát hozzáféréssel (virtuális hálózati integrációval), akkor a kiszolgálóhoz a kiszolgálóval azonos virtuális hálózaton belüli erőforrásból kell csatlakoznia. Létrehozhat egy virtuális gépet, és hozzáadhatja azt a rugalmas Azure Database for PostgreSQL-kiszolgálópéldányával létrehozott virtuális hálózathoz. Tekintse meg a hálózatkezelést.
  • Ha rugalmas Azure Database for PostgreSQL-kiszolgálópéldányt hozott létre nyilvános hozzáféréssel (engedélyezett IP-címekkel), hozzáadhatja a helyi IP-címet a kiszolgáló tűzfalszabályainak listájához. Tekintse meg a hálózatkezelést.

Microsoft Entra-integráció konfigurálása a kiszolgálón (csak jelszó nélküli)

Ha a jelszó nélküli hitelesítés lépéseit követi, a Microsoft Entra-hitelesítést konfigurálni kell a kiszolgálópéldányhoz, és Microsoft Entra-rendszergazdaként kell hozzárendelni a kiszolgálópéldányhoz. A Microsoft Entra-integráció konfigurálásához kövesse a Microsoft Entra-hitelesítés konfigurálását és a kiszolgálópéldány Microsoft Entra-rendszergazdaként való hozzárendelését.

A fejlesztési környezet előkészítése

Váltson olyan mappára, ahol futtatni szeretné a kódot, és hozzon létre és aktiváljon egy virtuális környezetet. A virtuális környezet egy önálló könyvtár a Python adott verziójához, valamint az alkalmazáshoz szükséges többi csomaghoz.

Futtassa a következő parancsokat egy virtuális környezet létrehozásához és aktiválásához:

py -3 -m venv .venv
.venv\Scripts\activate

A Python-kódtárak telepítése

Telepítse a példakódok futtatásához szükséges Python-kódtárakat.

Telepítse az Azure-identitástárat , amely a Microsoft Entra-jogkivonat-hitelesítést támogatja az Azure SDK-ban.

# Use the interpreter-bound pip to ensure installs go into the active venv/interpreter
python -m pip install --upgrade pip
python -m pip install azure-identity azure-keyvault-secrets

Hitelesítési kód hozzáadása

Ebben a szakaszban hitelesítési kódot ad hozzá a munkakönyvtárhoz, és végrehajtja a kiszolgálópéldány hitelesítéséhez és engedélyezéséhez szükséges további lépéseket.

A hitelesítési kód hozzáadása előtt győződjön meg arról, hogy az egyes példákhoz szükséges csomagok telepítve vannak.

Szükséges csomagok (példák ebben a cikkben):

  • Jelszó nélküli példa: azure-identity, azure-keyvault-secrets (a Key Vault használata esetén)
  • Példa jelszóra: psycopg (ajánlott: python -m pip install "psycopg[binary]")

Nem kötelező: hozzon létre egy requirements.txt ilyen bejegyzést, és telepítse a python -m pip install -r requirements.txt reprodukálható telepítéseket.

  1. Másolja a következő kódot egy szerkesztőbe, és mentse egy get_conn.py nevű fájlba.

    import urllib.parse
    import os
    
    from azure.identity import DefaultAzureCredential
    
    # IMPORTANT! This code is for demonstration purposes only. It's not suitable for use in production.
    # For example, tokens issued by Microsoft Entra ID have a limited lifetime (24 hours by default).
    # In production code, you need to implement a token refresh policy.
    
    def get_connection_uri():
    
        # Read URI parameters from the environment
        dbhost = os.environ['DBHOST']
        dbname = os.environ['DBNAME']
        dbuser = urllib.parse.quote(os.environ['DBUSER'])
        sslmode = os.environ['SSLMODE']
    
        # Use passwordless authentication via DefaultAzureCredential.
        # IMPORTANT! This code is for demonstration purposes only. DefaultAzureCredential() is invoked on every call.
        # In practice, it's better to persist the credential across calls and reuse it so you can take advantage of token
        # caching and minimize round trips to the identity provider. To learn more, see:
        # https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/identity/azure-identity/TOKEN_CACHING.md
        credential = DefaultAzureCredential()
    
        # Call get_token() to get a token from Microsft Entra ID and add it as the password in the URI.
        # Note the requested scope parameter in the call to get_token, "https://ossrdbms-aad.database.windows.net/.default".
        password = credential.get_token("https://ossrdbms-aad.database.windows.net/.default").token
    
        db_uri = f"postgresql://{dbuser}:{password}@{dbhost}/{dbname}?sslmode={sslmode}"
        return db_uri
    
  2. Adatbázis kapcsolati adatainak lekérése.

    1. Az Azure Portalon keresse meg és válassza ki a rugalmas Azure Database for PostgreSQL-kiszolgálópéldány nevét.
    2. A kiszolgáló Áttekintés lapján másolja ki a kiszolgáló teljes nevét. A kiszolgáló teljes neve mindig a my-server-name.postgres.database.azure.com< formátumú>.
    3. A bal oldali menü Biztonság területén válassza a Hitelesítés lehetőséget. Győződjön meg arról, hogy fiókja a Microsoft Entra-rendszergazdák listában szerepel. Ha nem, végezze el a Microsoft Entra integrációjának konfigurálását a kiszolgálón (csak jelszó nélküli).
  3. Környezeti változók beállítása a kapcsolati URI-elemekhez:

    set DBHOST=<server-name>
    set DBNAME=<database-name>
    set DBUSER=<username>
    set SSLMODE=require
    
  4. Jelentkezzen be az Azure-ba a munkaállomáson. Bejelentkezhet az Azure CLI, az Azure PowerShell vagy az Azure Developer CLI használatával.

    A hitelesítési kód a Microsoft Entra-azonosítóval való hitelesítésre és egy jogkivonat beszerzésére szolgál DefaultAzureCredential , amely engedélyezi a műveletek végrehajtását a kiszolgálópéldányon. DefaultAzureCredential támogatja a hitelesítési hitelesítő adatok típusainak láncát. A támogatott hitelesítő adatok közé tartoznak azok a hitelesítő adatok, amelyeket olyan fejlesztői eszközökbe jelentkezett be, mint az Azure CLI, az Azure PowerShell vagy az Azure Developer CLI.


A Python-példák futtatása

A cikkben szereplő minden kód példánál:

  1. Hozzon létre egy új fájlt egy szövegszerkesztőben.

  2. Adja hozzá a példakódot a fájlhoz.

  3. Mentse a fájlt a projektmappába egy .py kiterjesztéssel, például postgres-insert.py. Windows esetén a fájl mentésekor győződjön meg arról, hogy az UTF-8 kódolás ki van választva.

  4. A projektmappában például python a fájlnév és a fájlnév python postgres-insert.pyszerepel.

Tábla létrehozása és adatok beszúrása

Az alábbi példakód a függvény használatával csatlakozik a rugalmas Azure Database for PostgreSQL-kiszolgálóadatbázishoz, és egy SQL-utasítással psycopg.connect tölti be az INSERT adatokat. A cursor.execute függvény végrehajtja az SQL-lekérdezést az adatbázison.

import psycopg
from get_conn import get_connection_uri

conn_string = get_connection_uri()

conn = psycopg.connect(conn_string)
print("Connection established")
cursor = conn.cursor()

# Drop previous table of same name if one exists
cursor.execute("DROP TABLE IF EXISTS inventory;")
print("Finished dropping table (if existed)")

# Create a table
cursor.execute("CREATE TABLE inventory (id serial PRIMARY KEY, name VARCHAR(50), quantity INTEGER);")
print("Finished creating table")

# Insert some data into the table
cursor.execute("INSERT INTO inventory (name, quantity) VALUES (%s, %s);", ("banana", 150))
cursor.execute("INSERT INTO inventory (name, quantity) VALUES (%s, %s);", ("orange", 154))
cursor.execute("INSERT INTO inventory (name, quantity) VALUES (%s, %s);", ("apple", 100))
print("Inserted 3 rows of data")

# Clean up
conn.commit()
cursor.close()
conn.close()

Ha a kód sikeresen fut, a következő kimenetet hozza létre:

Connection established
Finished dropping table (if existed)
Finished creating table
Inserted 3 rows of data

Adatok beolvasása

Az alábbi példakód a rugalmas Azure Database for PostgreSQL-kiszolgálóadatbázishoz csatlakozik, és a cursor.execute parancsot használja az SQL-utasítással SELECT az adatok olvasásához. Ez a függvény egy lekérdezést fogad el, és visszaad egy eredményhalmazt, amely a cursor.fetchall() használatával iterál.

import psycopg
from get_conn import get_connection_uri

conn_string = get_connection_uri()

conn = psycopg.connect(conn_string)
print("Connection established")
cursor = conn.cursor()

# Fetch all rows from table
cursor.execute("SELECT * FROM inventory;")
rows = cursor.fetchall()

# Print all rows
for row in rows:
    print("Data row = (%s, %s, %s)" %(str(row[0]), str(row[1]), str(row[2])))

# Cleanup
conn.commit()
cursor.close()
conn.close()

Ha a kód sikeresen fut, a következő kimenetet hozza létre:

Connection established
Data row = (1, banana, 150)
Data row = (2, orange, 154)
Data row = (3, apple, 100)

Adatok frissítése

Az alábbi példakód a rugalmas Azure Database for PostgreSQL-kiszolgálóadatbázishoz csatlakozik, és a cursor.execute parancsot használja az SQL-utasítással UPDATE az adatok frissítéséhez.

import psycopg
from get_conn import get_connection_uri

conn_string = get_connection_uri()

conn = psycopg.connect(conn_string)
print("Connection established")
cursor = conn.cursor()

# Update a data row in the table
cursor.execute("UPDATE inventory SET quantity = %s WHERE name = %s;", (200, "banana"))
print("Updated 1 row of data")

# Cleanup
conn.commit()
cursor.close()
conn.close()

Adatok törlése

Az alábbi példakód a rugalmas Azure Database for PostgreSQL-kiszolgálóadatbázishoz csatlakozik, és a cursor.execute parancsot használja az SQL utasítással DELETE a korábban beszúrt leltárelem törléséhez.

import psycopg
from get_conn import get_connection_uri

conn_string = get_connection_uri()

conn = psycopg.connect(conn_string)
print("Connection established")
cursor = conn.cursor()

# Delete data row from table
cursor.execute("DELETE FROM inventory WHERE name = %s;", ("orange",))
print("Deleted 1 row of data")

# Cleanup
conn.commit()
cursor.close()
conn.close()