Почему не работают SELECT запросы в Psycopg2?

Question

[AlmazKayum]

Библиотека psycopg2-2.7.4

import psycopg2
class Database:
    def __init__(self, database):
        self.conn = psycopg2.connect(database)
        self.cursor = self.conn.cursor()
    def put_balance(self, bal, chat_id):
        with self.conn:
            self.cursor.execute('UPDATE profile SET points = %s WHERE chat_id = %s', (bal, chat_id))
    def get_balance(self, chat_id):
        with self.conn:
            return self.cursor.execute('SELECT points FROM profile WHERE chat_id = %s;', (chat_id,))

В классе Database все методы SELECT выдают None, в то же время как все методы UPDATE и INSERT работают.
Когда применяешь метод fetchall() или fetchone(), выходит следующая ошибка:
return self.cursor.execute('SELECT points FROM profile WHERE chat_id = %s;', (chat_id,)).fetchall()
AttributeError: 'NoneType' object has no attribute 'fetchall'
В чем может быть причина?

Answer 1

[Андрей Буров]

initd.org/psycopg/docs/usage.html

# Query the database and obtain data as Python objects
>>> cur.execute("SELECT * FROM test;")
>>> cur.fetchone()
(1, 100, "abc'def")

# Make the changes to the database persistent
>>> conn.commit()

Comments

[AlmazKayum]

У меня cursor c SELECT выдает None. Когда применяешь метод fetchall() или fetchone(), выходит следующая ошибка:
return self.cursor.execute('SELECT points FROM profile WHERE chat_id = %s;', (chat_id,)).fetchall()
AttributeError: 'NoneType' object has no attribute 'fetchall'

[Андрей Буров]

AlmazKayum, все верно

initd.org/psycopg/docs/cursor.html

The method returns None. If a query was executed, the returned values can be retrieved using fetch*() methods.

вам нужно делать не self.cursor.execute('...').fetchall(), а

self.cursor.execute('...')
self.curosr.fetchall()

[AlmazKayum]

Андрей Буров, заработало! Благодарю вас!!)
Еще небольшой вопросик, в моем случае с конструкцией with и классом для БД нужно ли делать conn.commit() и как это делать?

[Андрей Буров]

AlmazKayum, нужно

with self.conn:
            self.cursor.execute('UPDATE profile SET points = %s WHERE chat_id = %s', (bal, chat_id))
            self.conn.commit()

[AlmazKayum]

Андрей Буров, благодарю