web-dev-qa-db-fra.com

Comment obtenir une liste de noms de colonnes à partir d'un curseur psycopg2?

Je voudrais un moyen général de générer des étiquettes de colonne directement à partir des noms de colonne sélectionnés et de rappeler de voir que le module psycopg2 de python prend en charge cette fonctionnalité.

pythonpsycopg2
100
SetJmp

De "Programmation Python" de Mark Lutz:

curs.execute("Select * FROM people")
colnames = [desc[0] for desc in curs.description]
168
SetJmp

Une autre chose que vous pouvez faire est de créer un curseur avec lequel vous pourrez référencer vos colonnes par leurs noms (c’est un besoin qui m’a amené à cette page):

import psycopg2
from psycopg2.extras import RealDictCursor

ps_conn = psycopg2.connect(...)
ps_cursor = psql_conn.cursor(cursor_factory=RealDictCursor)

ps_cursor.execute('select 1 as col_a, 2 as col_b')
my_record = ps_cursor.fetchone()
print (my_record['col_a'],my_record['col_b'])

>> 1, 2
23
Dennis

Pour obtenir les noms de colonne dans une requête distincte, vous pouvez interroger la table information_schema.columns.

#!/usr/bin/env python3

import psycopg2

if __== '__main__':
  DSN = 'Host=YOUR_DATABASE_Host port=YOUR_DATABASE_PORT dbname=YOUR_DATABASE_NAME user=YOUR_DATABASE_USER'

  column_names = []

  with psycopg2.connect(DSN) as connection:
      with connection.cursor() as cursor:
          cursor.execute("select column_name from information_schema.columns where table_schema = 'YOUR_SCHEMA_NAME' and table_name='YOUR_TABLE_NAME'")
          column_names = [row[0] for row in cursor]

  print("Column names: {}\n".format(column_names))

Pour obtenir les noms de colonne dans la même requête que les lignes de données, vous pouvez utiliser le champ de description du curseur:

#!/usr/bin/env python3

import psycopg2

if __== '__main__':
  DSN = 'Host=YOUR_DATABASE_Host port=YOUR_DATABASE_PORT dbname=YOUR_DATABASE_NAME user=YOUR_DATABASE_USER'

  column_names = []
  data_rows = []

  with psycopg2.connect(DSN) as connection:
    with connection.cursor() as cursor:
      cursor.execute("select field1, field2, fieldn from table1")
      column_names = [desc[0] for desc in cursor.description]
      for row in cursor:
        data_rows.append(row)

  print("Column names: {}\n".format(column_names))
22
bitek

Si vous voulez avoir une requête nommée Tuple obj from db, vous pouvez utiliser l'extrait suivant:

from collections import namedtuple

def create_record(obj, fields):
    ''' given obj from db returns named Tuple with fields mapped to values '''
    Record = namedtuple("Record", fields)
    mappings = dict(Zip(fields, obj))
    return Record(**mappings)

cur.execute("Select * FROM people")
colnames = [desc[0] for desc in cur.description]
rows = cur.fetchall()
cur.close()
result = []
for row in rows:
    result.append(create_record(row, colnames))

Cela vous permet d’accéder aux valeurs d’enregistrement comme s’il s’agissait de propriétés de classe, c.-à-d.

record.id, record.other_table_column_name, etc.

ou même plus court

from psycopg2.extras import NamedTupleCursor
with cursor(cursor_factory=NamedTupleCursor) as cur:
   cur.execute("Select * ...")
   return cur.fetchall()
10
zzart

J'avais aussi l'habitude de faire face à un problème similaire. J'utilise un truc simple pour résoudre ce problème. Supposons que vous ayez des noms de colonnes dans une liste comme

col_name = ['a', 'b', 'c']

Ensuite, vous pouvez suivre

for row in cursor.fetchone():
    print Zip(col_name, row)
1
user2925213

Après avoir exécuté la requête SQL, écrivez en suivant le script python écrit en 2.7).

total_fields = len(cursor.description)    
fields_names = [i[0] for i in cursor.description   
    Print fields_names
1
RAFI AFRIDI

J'ai remarqué que vous devez utiliser cursor.fetchone() après la requête pour obtenir la liste des colonnes dans cursor.description (c'est-à-dire dans [desc[0] for desc in curs.description])

0
user2618351