Python PostgreSQL - Atualizar tabela

Você pode modificar o conteúdo dos registros existentes de uma tabela no PostgreSQL usando a instrução UPDATE. Para atualizar linhas específicas, você precisa usar a cláusula WHERE junto com ela.

Sintaxe

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

UPDATE table_name
SET column1 = value1, column2 = value2...., columnN = valueN
WHERE [condition];

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 declaração a seguir modifica a idade do jogador de críquete, cujo primeiro nome é Shikhar -

postgres=# UPDATE CRICKETERS SET AGE = 45 WHERE FIRST_NAME = 'Shikhar' ;
UPDATE 1
postgres=#

Se você recuperar o registro cujo FIRST_NAME é Shikhar, você observará que o valor da idade foi alterado para 45 -

postgres=# SELECT * FROM CRICKETERS WHERE FIRST_NAME = 'Shikhar';
first_name  | last_name | age | place_of_birth | country
------------+-----------+-----+----------------+---------
Shikhar     | Dhawan    | 45  | Delhi          | India
(1 row)
postgres=#

Se você não usou a cláusula WHERE, os valores de todos os registros serão atualizados. A instrução UPDATE seguinte aumenta a idade de todos os registros na tabela CRICKETERS em 1 -

postgres=# UPDATE CRICKETERS SET AGE = AGE+1;
UPDATE 5

Se você recuperar o conteúdo da tabela usando o comando SELECT, poderá ver os valores atualizados como -

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

Atualizar registros usando python

A classe cursor de psycopg2 fornece um método com o nome método execute (). Este método aceita a consulta como parâmetro e a executa.

Portanto, para inserir dados em uma tabela no PostgreSQL usando python -

  • Importar psycopg2 pacote.

  • Crie um objeto de conexão usando o connect() , passando o nome de usuário, senha, host (padrão opcional: localhost) e banco de dados (opcional) como parâmetros para ele.

  • Desligue o modo auto-commit definindo false como valor para o atributo autocommit.

  • o cursor() método do Connectionclasse da biblioteca psycopg2 retorna um objeto cursor. Crie um objeto cursor usando este método.

  • Em seguida, execute a instrução UPDATE passando-a como um parâmetro para o método execute ().

Exemplo

O código Python seguinte atualiza o conteúdo da tabela Employee e recupera os resultados -

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()

#Fetching all the rows before the update
print("Contents of the Employee table: ")
sql = '''SELECT * from EMPLOYEE'''
cursor.execute(sql)
print(cursor.fetchall())

#Updating the records
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = 'M'"
cursor.execute(sql)
print("Table updated...... ")

#Fetching all the rows after the update
print("Contents of the Employee table after the update operation: ")
sql = '''SELECT * from EMPLOYEE'''
cursor.execute(sql)
print(cursor.fetchall())

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

#Closing the connection
conn.close()

Resultado

Contents of the Employee table:
[('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)]
Table updated......
Contents of the Employee table after the update operation:
[('Ramya', 'Rama priya', 27, 'F', 9000.0), 
   ('Sarmista', 'Sharma', 26, 'F', 10000.0), 
   ('Tripthi', 'Mishra', 24, 'F', 6000.0), 
   ('Vinay', 'Battacharya', 21, 'M', 6000.0), 
   ('Sharukh', 'Sheik', 26, 'M', 8300.0)]