Obtenga nombres de columna de la tabla PostgreSQL usando Psycopg2

Este artículo es una ilustración de cómo extraer nombres de columnas de una tabla de PostgreSQL usando psycopg2 y Python.

Mesa usada para demostración:

Ejemplo 1:

Primero, conectamos la base de datos PostgreSQL usando el método psycopg2.connect()  , luego creamos un cursor usando el método cursor(), luego usamos el cursor() y luego extraemos el primer elemento usando el corte.

import psycopg2
  
conn = psycopg2.connect(
    database="geeks",
    user='postgres',
    password='pass',
    host='localhost',
    port= '5432'
)
  
conn.autocommit = True
cursor = conn.cursor()
  
sql = '''SELECT * FROM products'''
  
  
cursor.execute(sql)
column_names = [desc[0] for desc in cursor.description]
for i in column_names:
    print(i)
conn.commit()
conn.close()

Producción:

product_no
name
price

Ejemplo 2: en el segundo enfoque, ejecutamos el siguiente comando SQL en el método cursor.execute().

«seleccione COLUMN_NAME de information_schema.columns donde table_schema = ‘SCHEMA_NAME’ y table_name=’TABLE_NAME'»

import psycopg2
  
conn = psycopg2.connect(
    database="geeks",
    user='postgres',
    password='root',
    host='localhost',
    port='5432'
)
  
conn.autocommit = True
  
  
with conn:
    with conn.cursor() as cursor:
        cursor.execute(
            "select COLUMN_NAME from information_schema.columns\
            where table_name='products'")
        column_names = [row[0] for row in cursor]
  
print("Column names:\n")
  
for i in column_names:
    print(i)

Producción:

Column names:

product_no
name
price