Как подключить и выполнить SQL-запросы к базе данных PostgreSQL из Python?

В этой статье основное внимание уделяется подключению к базе данных PostgreSQL из Python.

Installation:

• Установите PostgreSQL, если вы его не установили.
• Нам нужно установить библиотеку psycopg2 для подключения к базе данных PostgreSQL. Откройте командную строку и выполните приведенную ниже команду, чтобы установить psycopg2.

pip3 install psycopg2

Создание базы данных

Вы можете создать базу данных двумя способами:

1. Использование пользовательского интерфейса pgAdmin 4
2. Использование команды

1. Использование пользовательского интерфейса pgAdmin 4

Перейдите в pgAdmin и выполните следующие действия.

• Local_server[Правый клик] -> Создать -> База данных

Создать базу данных из всплывающего бокового меню

• Далее заполните форму База данных: WorkSpaceи сохраните.

2. Создать базу данных с помощью SQL-запроса

Перейдите в pgAdmin и выполните следующие действия:
Запустите приведенную ниже команду на вкладке «Запрос».

CREATE DATABASE WorkSpace;

Запустите команду создания базы данных

Подключение к базе данных

Нам нужно подключиться к базе данных PostgreSQL с помощью функции psycopg2.connect().

Где атрибуты функции connect():

host = hostname,
dbname = databaseName,
user = username,
password = [Your Password],
port = [port_id]

Если вы не знаете какой-либо из этих атрибутов функции connect(), вы можете выполнить следующие шаги:

Теперь вы знаете все свойства этой базы данных. Давай продолжим.

Теперь, чтобы подключиться к базе данных, нам нужно передать атрибуты в качестве аргументов функции connect().

Синтаксис:

conn = psycopg2.connect(
    host = 'localhost',
    dbname = 'For_Practice',
    user = 'postgres',
    password = '[Password]',
    port = 5432
)

Создайте курсор:

• Создайте объект курсора (то есть curr) и вызовите его метод execute() для выполнения запросов.
• Где метод execute() используется для запуска запроса, который передается в виде строки.

Синтаксис:

cur = conn.cursor()
cur.execute('[SQL queries]')

В конце нам нужно сохранить изменения с помощью метода commit() и, наконец, закрыть открытое соединение с помощью метода close().

Синтаксис:

conn.commit()
cur.close()

Python3

import psycopg2

 

conn = None

try:

    # connect to the PostgreSQL server

    print('Connecting to the PostgreSQL database...')

    conn = psycopg2.connect(

        host = 'localhost',

        dbname = 'For_Practice',

        user = 'postgres',

        password = '321654',

        port = 5432

    )

     

    # Creating a cursor with name cur.

    cur = conn.cursor()

    print('Connected to the PostgreSQL database')

     

    # Execute a query:

    # To display the PostgreSQL 

    # database server version

    cur.execute('SELECT version()')

    print(cur.fetchone())

     

    # Close the connection

    cur.close()

     

except(Exception, psycopg2.DatabaseError) as error:

    print(error)

finally:

    if conn is not None:

        conn.close()

        print('Database connection closed.')

Выход: