使用python进行PostgreSQL 数据库连接

使用python进行PostgreSQL 数据库连接

PostgreSQL 数据库是最常用的关系型数据库之一，最吸引人的一点是它作为开源数据库且具有可拓展性，能够提供丰富的应用。运用python可以很简单的建立PostgreSQL 数据库连接，其中最受欢迎的就是psycopg。

1. 安装psycopg2

Psycopy是针对python的Postgres 数据库的适配模块，安装psycopg2可以整合python和Postgres 。使用cmd输入命令进行安装：

pip install psycopg2

也可以在pycharm中查找psycopg2安装包：

2. 图形化连接数据库

在pycharm中选择Database，点击左上角的+添加数据库，选择postgresql:
创建数据库连接后点击apply，数据库会显示在右侧窗格中。

3. 代码连接数据库

3.1 不使用配置文件

下面使用 psycopy2.connect()方法连接到postgresql数据库。通过调用cursor类中的execute()方法对数据库进行操作。在execute()中用SQL语句创建表。使用commit()将数据发送到数据库服务器，最后使用close()关闭数据库。commit()能够对数据库进行改变，且不可逆。

connect() 方法的参数一般包括:

• database: 要连接的数据库名称
• user:连接数据库的用户名
• password: 连接数据库的密码
• host: 数据库端口的地址，一般为 “localhost”,或者主机的IP地址
• port: 门户 默认为5432.

import psycopg2

con = psycopg2.connect(database="postgres",
                       user="fbase",
                       password="123456",
                       host="192.168.198.152",
                       port="8432")
print(con)
print("Database opened successfully")
cur = con.cursor()
cur.execute('SELECT version()')
db_version = cur.fetchone()
print(db_version)
con.close()

运行结果如下：

3.2 使用配置文件

可以使用配置文件来存储所有连接参数。

database.ini文件的内容如下：

[postgresql]
host = 192.168.198.152
database = postgres
user = fbase
password = 123456
port = 8432

下面的config()函数会读取database.ini文件并返回连接参数。该config()函数放置在config.py文件中：

from configparser import ConfigParser


def config(filename='../../resource/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)