Подтвердить что ты не робот

Sqlite и Python - вернуть словарь с использованием fetchone()?

Я использую sqlite3 в python 2.5. Я создал таблицу, которая выглядит так:

   create table votes (
      bill text,
      senator_id text,
      vote text)

Я обращаюсь к нему с чем-то вроде этого:

v_cur.execute("select * from votes")
row = v_cur.fetchone()
bill = row[0]
senator_id = row[1]
vote = row[2]

То, что я хотел бы сделать, это иметь fetchone (или какой-то другой метод), чтобы вернуть словарь, а не список, чтобы я мог ссылаться на поле по имени, а не по позиции. Например:

bill = row['bill'] 
senator_id = row['senator_id']
vote = row['vote']

Я знаю, что вы можете сделать это с MySQL, но кто-нибудь знает, как это сделать с помощью SQLite?

Спасибо!!!

4b9b3361

ОТВЕТЫ

Ответ 1

То, как я это делал в прошлом:

def dict_factory(cursor, row):
    d = {}
    for idx,col in enumerate(cursor.description):
        d[col[0]] = row[idx]
    return d

Затем вы установите его в своем соединении:

from pysqlite2 import dbapi2 as sqlite
conn = sqlite.connect(...)
conn.row_factory = dict_factory

Это работает под pysqlite-2.4.1 и python 2.5.4.

Ответ 2

В sqlite3 есть вариант для этого. Измените член row_factory объекта соединения на sqlite3.Row:

conn = sqlite3.connect('db', row_factory=sqlite3.Row)

или

conn.row_factory = sqlite3.Row

Это позволит вам получить доступ к элементам строки по имени - в словаре-стиле - или по индексу. Это намного эффективнее, чем создание вашей собственной работы.

Ответ 3

Недавно я пытался сделать что-то подобное при использовании sqlite3.Row(). Хотя sqlite3.Row() отлично подходит для создания словарного интерфейса или интерфейса типа кортежа, это не сработало, когда я транслировался в строке, используя ** kwargs. Таким образом, необходим быстрый способ преобразования его в словарь. Я понял, что объект Row() можно преобразовать в словарь просто с помощью itertools.

Python 2:

db.row_factory = sqlite3.Row
dbCursor = db.cursor()
dbCursor.execute("SELECT * FROM table")
row = dbCursor.fetchone()

rowDict = dict(itertools.izip(row.keys(), row))

Или в Python 3, проще:

dbCursor = db.cursor()
dbCursor.execute("SELECT * FROM table")
row = dbCursor.fetchone()
rowDict = dict(zip([c[0] for c in dbCursor.description], row))

Аналогично, вы можете использовать команду dbCursor.fetchall() и преобразовать весь набор строк в список словарей в цикле for.

Ответ 5

Я знаю, что вы не спрашиваете об этом, но почему бы просто не использовать sqlalchemy для создания orm для базы данных? то вы можете делать такие вещи, как


entry = model.Session.query(model.Votes).first()
print entry.bill, entry.senator_id, entry.vote

в качестве дополнительного бонуса ваш код будет легко переноситься в альтернативную базу данных, а соединения и еще что-то будут управляться бесплатно.

Ответ 6

Я использовал это:

def get_dict(sql):
    return dict(c.execute(sql,()).fetchall())

Затем вы можете сделать это:

c = conn.cursor()
d = get_dict("select user,city from vals where user like 'a%'");

Теперь d - словарь, где ключи user, а значения city. Это также работает для group by

Ответ 7

Я использую что-то вроде этого:

class SqliteRow(object):
    def __init__(self):
        self.fields = []

    def add_field(self, name, value):
        self.fields.append(name)
        setattr(self, name, value)

    def to_tuple(self):
        return tuple([getattr(self, x) for x in self.fields])

с этим:

def myobject_factory(cursor, row):
    myobject= MyObject()
    for idx, col in enumerate(cursor.description):
        name, value = (col[0], row[idx])

        myobject.add_field(name, value)
    return myobject

MyObject() - это класс, который наследуется от SqliteRow. Класс SqliteRow является базовым классом для каждого объекта, который я хочу вернуть запросом. Каждый столбец становится атрибутом и регистрируется в списке fields. Функция to_tuple используется для изменения всего объекта в форму, подходящую для запросов (просто пропустите весь объект и забудьте).

Чтобы получить разные типы классов этой функции. Вам потребуется создать фабричный объект, который будет генерировать объекты на основе списка полей (например: dict with {some_unique_value_made_of_fields: class})

Таким образом, я получаю простой ORM.

Ответ 8

Следующие работы для меня:

cursor = conn.cursor(buffered = True, dictionary = True)

Другой вариант:

cursor = conn.cursor(MySQLdb.cursors.DictCursor)

Остальная часть кода:

query = "SELECT * FROM table"
cursor.execute(query)
row = cursor.fetchone()

Источники: mysql.connector.cursor, MySQLdb.cursors.DictCursor