Python PostgreSQL - Selecionar dados

Você pode recuperar o conteúdo de uma tabela existente no PostgreSQL usando a instrução SELECT. Nesta instrução, você precisa especificar o nome da tabela e ela retorna seu conteúdo em formato tabular, que é conhecido como conjunto de resultados.

Sintaxe

A seguir está a sintaxe da instrução SELECT no PostgreSQL -

SELECT column1, column2, columnN FROM table_name;

Exemplo

Suponha que criamos uma tabela com o nome CRICKETERS usando a seguinte consulta -

postgres=# CREATE TABLE CRICKETERS ( 
   First_Name VARCHAR(255), Last_Name VARCHAR(255), Age int, 
   Place_Of_Birth VARCHAR(255), Country VARCHAR(255)
);

CREATE TABLE

postgres=#

E se tivermos inserido 5 registros nele usando instruções INSERT como -

postgres=# insert into CRICKETERS values('Shikhar', 'Dhawan', 33, 'Delhi', 'India');
INSERT 0 1
postgres=# insert into CRICKETERS values('Jonathan', 'Trott', 38, 'CapeTown', 'SouthAfrica');
INSERT 0 1
postgres=# insert into CRICKETERS values('Kumara', 'Sangakkara', 41, 'Matale', 'Srilanka');
INSERT 0 1
postgres=# insert into CRICKETERS values('Virat', 'Kohli', 30, 'Delhi', 'India');
INSERT 0 1
postgres=# insert into CRICKETERS values('Rohit', 'Sharma', 32, 'Nagpur', 'India');
INSERT 0 1

A consulta SELECT seguinte recupera os valores das colunas FIRST_NAME, LAST_NAME e, COUNTRY da tabela CRICKETERS.

postgres=# SELECT FIRST_NAME, LAST_NAME, COUNTRY FROM CRICKETERS;
 first_name | last_name  | country
------------+------------+-------------
Shikhar     | Dhawan     | India
Jonathan    | Trott      | SouthAfrica
Kumara      | Sangakkara | Srilanka
Virat       | Kohli      | India
Rohit       | Sharma     | India
(5 rows)

Se você deseja recuperar todas as colunas de cada registro, você precisa substituir os nomes das colunas por "⚹" como mostrado abaixo -

postgres=# SELECT * FROM CRICKETERS;
first_name  | last_name  | age | place_of_birth | country
------------+------------+-----+----------------+-------------
Shikhar     | Dhawan     | 33  | Delhi          | India
Jonathan    | Trott      | 38  | CapeTown       | SouthAfrica
Kumara      | Sangakkara | 41  | Matale         | Srilanka
Virat       | Kohli      | 30  | Delhi          | India
Rohit       | Sharma     | 32  | Nagpur         | India
(5 rows)

postgres=#

Recuperando dados usando Python

A operação READ em qualquer banco de dados significa buscar algumas informações úteis do banco de dados. Você pode buscar dados do PostgreSQL usando o método fetch () fornecido pelo psycopg2.

A classe Cursor fornece três métodos: fetchall (), fetchmany () e fetchone (), onde,

  • O método fetchall () recupera todas as linhas no conjunto de resultados de uma consulta e as retorna como uma lista de tuplas. (Se executarmos isso depois de recuperar algumas linhas, ele retornará as restantes).

  • O método fetchone () busca a próxima linha no resultado de uma consulta e a retorna como uma tupla.

Note - Um conjunto de resultados é um objeto que é retornado quando um objeto cursor é usado para consultar uma tabela.

Exemplo

O programa Python a seguir se conecta a um banco de dados chamado mydb do PostgreSQL e recupera todos os registros de uma tabela chamada EMPLOYEE.

import psycopg2

#establishing the connection
conn = psycopg2.connect(
   database="mydb", user='postgres', password='password', host='127.0.0.1', port= '5432'
)

#Setting auto commit false
conn.autocommit = True

#Creating a cursor object using the cursor() method
cursor = conn.cursor()

#Retrieving data
cursor.execute('''SELECT * from EMPLOYEE''')

#Fetching 1st row from the table
result = cursor.fetchone();
print(result)

#Fetching 1st row from the table
result = cursor.fetchall();
print(result)

#Commit your changes in the database
conn.commit()

#Closing the connection
conn.close()

Resultado

('Ramya', 'Rama priya', 27, 'F', 9000.0)
[
   ('Vinay', 'Battacharya', 20, 'M', 6000.0),
   ('Sharukh', 'Sheik', 25, 'M', 8300.0),
   ('Sarmista', 'Sharma', 26, 'F', 10000.0),
   ('Tripthi', 'Mishra', 24, 'F', 6000.0)
]