📜  PostgreSQL – 使用Python连接到数据库

📅  最后修改于: 2022-05-13 01:54:54.389000             🧑  作者: Mango

PostgreSQL – 使用Python连接到数据库

PostgreSQL API for Python允许用户使用psycopg2 模块与 PostgreSQL 数据库进行交互。在本文中,我们将研究使用Python连接到 PostgreSQL 数据库的过程。

先决条件:

首先,我们需要在命令提示符下使用以下命令安装 psycopg2 模块:

pip install psycopg2

创建数据库:

出于示例的目的,我们将需要一个示例数据库。要创建它,请按照以下步骤操作:

  • 首先打开一个 PostgreSQL 客户端工具,如 pgadmin4 或 psql。
  • 使用您的凭据再次登录数据库。
  • 最后使用下面的命令创建一个数据库(比如 School)
CREATE DATABASE school;

连接数据库:

要连接到上面创建的数据库(即学校),我们使用connect()函数。 connect()函数用于创建一个新的数据库会话并返回一个新的连接类实例。

为此,请使用以下语法:

Syntax: conn = psycopg2.connect("dbname=suppliers user=postgres password=postgres")

为了方便使用,您可以使用以下方法:

1. 首先使用如下所示的凭据创建一个database.ini文件:

[postgresql]
host=localhost
database=school
user=postgres
password=5555

现在,下面的config()函数读取database.ini文件并返回连接参数。在config.py 文件中添加了相同的 config()函数:

Python3
#!/usr/bin/python
from configparser import ConfigParser
  
  
def config(filename='database.ini', section='postgresql'):
    # create a parser
    parser = ConfigParser()
    # read config file
    parser.read(filename)
  
    # get section, default to postgresql
    db = {}
    if parser.has_section(section):
        params = parser.items(section)
        for param in params:
            db[param[0]] = param[1]
    else:
        raise Exception('Section {0} not found in the {1} file'.format(section, filename))
  
    return db


Python3
#!/usr/bin/python
import psycopg2
from config import config
  
def connect():
    """ Connect to the PostgreSQL database server """
    conn = None
    try:
        # read connection parameters
        params = config()
  
        # connect to the PostgreSQL server
        print('Connecting to the PostgreSQL database...')
        conn = psycopg2.connect(**params)
          
        # create a cursor
        cur = conn.cursor()
          
    # execute a statement
        print('PostgreSQL database version:')
        cur.execute('SELECT version()')
  
        # display the PostgreSQL database server version
        db_version = cur.fetchone()
        print(db_version)
         
    # close the communication with the PostgreSQL
        cur.close()
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()
            print('Database connection closed.')
  
  
if __name__ == '__main__':
    connect()


例子:

下面的 connect()函数连接到我们之前创建的学校数据库并返回 PostgreSQL 数据库版本。

Python3

#!/usr/bin/python
import psycopg2
from config import config
  
def connect():
    """ Connect to the PostgreSQL database server """
    conn = None
    try:
        # read connection parameters
        params = config()
  
        # connect to the PostgreSQL server
        print('Connecting to the PostgreSQL database...')
        conn = psycopg2.connect(**params)
          
        # create a cursor
        cur = conn.cursor()
          
    # execute a statement
        print('PostgreSQL database version:')
        cur.execute('SELECT version()')
  
        # display the PostgreSQL database server version
        db_version = cur.fetchone()
        print(db_version)
         
    # close the communication with the PostgreSQL
        cur.close()
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()
            print('Database connection closed.')
  
  
if __name__ == '__main__':
    connect()

输出: