526互联

Python数据库操作—— PostgreSQL

发布时间 2023-03-24 16:46:39作者: 徐俊112

使用pip3安装psycopg2

pip install psycopg2

　

增删改查（CRUD）
新建数据库
PostgreSQL不支持"SHOW DATABASES"语句，需要使用"SELECT * FROM PG_DATABASE;"进行查询，才能获取当前数据库列表，由于这里只需要做创建数据库前后的对比，所以只获取数据库名字段，即dtname。该程序创建的数据库为root用户所有。

创建数据库的时候，为避免出错，需要将数据库连接切换到自动提交模式。

import psycopg2
 
 
# 打印数据函数
def ShowData(str, data):
    arr = []
    for item in data:
        arr.append(item)
    print(str, arr)
 
 
# 打开数据库连接
db = psycopg2.connect(host='localhost',
                      port='5432',
                      user='root',
                      password='123456')
 
# 设置数据库连接打开自动提交模式
db.autocommit = True
 
# 创建游标对象
cursor = db.cursor()
 
# 查看现有数据库
sql = 'SELECT DATNAME FROM PG_DATABASE;'
cursor.execute(sql)
ShowData("现有数据库：", cursor.fetchall())
 
# 创建test数据库
sql = 'CREATE DATABASE "test";'
cursor.execute(sql)
 
# 查看新数据库创建完成后的数据库列表
sql = 'SELECT DATNAME FROM PG_DATABASE;'
cursor.execute(sql)
ShowData("创建完成后的数据库：", cursor.fetchall())
 
# 释放游标及数据库连接
cursor.close()
db.close()

　　

运行结果如下，可以看到数据库被成功创建：