Come ottenere i nomi dei campi quando si esegue una query sql semplice in django

Question

In una delle mie viste di django, eseguo query sul database utilizzando sql semplice (non orm) e restituisco i risultati.

sql = "select * from foo_bar" 
cursor = connection.cursor() 
cursor.execute(sql) 
rows = cursor.fetchall() 

Ottengo i dati bene, ma non i nomi di colonna. Come posso ottenere i nomi dei campi del set di risultati che viene restituito?

Answer 1

Secondo PEP 249, è possibile provare a utilizzare cursor.description, ma questo non è del tutto affidabile.

Answer 2

Ho trovato una bella soluzione nel blog di Doug Hellmann:

http://doughellmann.com/2007/12/30/using-raw-sql-in-django.html

from itertools import * 
from django.db import connection 

def query_to_dicts(query_string, *query_args): 
    """Run a simple query and produce a generator 
    that returns the results as a bunch of dictionaries 
    with keys for the column values selected. 
    """ 
    cursor = connection.cursor() 
    cursor.execute(query_string, query_args) 
    col_names = [desc[0] for desc in cursor.description] 
    while True: 
     row = cursor.fetchone() 
     if row is None: 
      break 
     row_dict = dict(izip(col_names, row)) 
     yield row_dict 
    return 

Esempio utilizzo:

row_dicts = query_to_dicts("""select * from table""") 

Answer 3

Sul Django docs, c'è un metodo molto semplice fornito (che fa effettivamente usare cursor.description, come rispose Ignacio).

def dictfetchall(cursor): 
    "Returns all rows from a cursor as a dict" 
    desc = cursor.description 
    return [ 
     dict(zip([col[0] for col in desc], row)) 
     for row in cursor.fetchall() 
    ]