📅  最后修改于: 2020-04-15 07:43:45             🧑  作者: Mango
PostgreSQL是一个开源的对象关系数据库管理系统。以其可靠性,鲁棒性和性能而闻名。PostgreSQL有各种API(应用程序可编程接口)库,可用于各种流行的编程语言,例如Python。它为数据库管理提供了许多功能,例如视图,触发器,索引(使用B树)等。
有几个Python模块可以让我们使用PostgreSQL连接和操作数据库:
Psycopg2是PostgreSQL最受欢迎的Python驱动程序之一。它被积极维护并为不同版本的Python提供支持。它还提供对线程的支持,并且可以在多线程应用程序中使用。由于这些原因,它是开发人员的流行选择。
在本文中,我们将通过在Python中构建一个简单的数据库管理系统来探索使用psycopg2使用PostgreSQl的功能。
安装:
sudo pip3 install psycopg2
注意:如果您使用的是Python2,请使用pip install代替pip3
一旦psycopg已经安装在系统中,我们可以连接到数据库,并在Python执行查询。
在我们可以使用Python访问数据库之前,我们需要在postgresql中创建数据库。要创建数据库,请遵循以下步骤:
sudo -u postgres psql
\password
然后将提示您输入密码。记住它,因为我们将使用它来连接到Python中的数据库。
CREATE DATABASE test;
conn = psycopg2.connect(
database="test",
user = "adith",
password = "password",
host = "localhost",
port = "5432")
建立连接后,我们可以使用Python操作数据库。
将光标对象用于执行SQL查询。我们可以使用连接对象(conn)创建一个游标对象
cur = conn.cursor()
使用此对象,我们可以更改连接到的数据库。
执行完所有查询后,我们需要断开连接。不断开连接不会导致任何错误,但是通常认为断开连接是一种好习惯。
conn.close()
execute()方法接受一个参数: 要执行的SQL查询。SQL查询采用包含SQL语句的字符串形式。
cur.execute(“ SELECT * FROM emp")
一旦执行了查询,就可以使用fetchall()方法获取查询的结果。此方法不带参数,并返回选择查询的结果。
res = cur.fetchall()
查询结果存储在res变量中。
在PostgreSQL中创建数据库后,就可以使用Python访问该数据库。我们首先使用以下模式在数据库中创建一个名为test的emp表:(id INTEGER PRIMARY KEY,名称VARCHAR(10),salary INT,dept INT)。创建表后,没有任何错误,我们将值插入表中。
插入值后,我们可以查询表以选择所有行,并使用fetchall()
函数将其显示给用户
# 导入库
import psycopg2
# 连接数据库的功能.
def connect():
# 使用connect函数连接到名为test的数据库
try:
conn = psycopg2.connect(database ="test",
user = "adith",
password = "password",
host = "localhost",
port = "5432")
# 创建光标对象
cur = conn.cursor()
except (Exception, psycopg2.DatabaseError) as error:
print ("Error while creating PostgreSQL table", error)
# 重载conn和cur对象以供以后使用
return conn, cur
# 创建emp表的函数.
def create_table():
# 连接到数据库.
conn, cur = connect()
try:
# 测试数据库包含一个名为emp的表 : (id INTEGER PRIMARY KEY,
# name VARCHAR(10), salary INT, dept INT)
# 创建emp表
cur.execute('CREATE TABLE emp (id INT PRIMARY KEY, name VARCHAR(10),
salary INT, dept INT)')
# commit函数将永久保存对数据库所做的更改,如果有任何不希望的更改,可以使用rollback()函数,并且只需撤销先前查询的更改即
except:
print('error')
conn.commit()
# 将数据插入emp表的函数
def insert_data(id = 1, name = '', salary = 1000, dept = 1):
conn, cur = connect()
try:
# 将值插入emp表
cur.execute('INSERT INTO emp VALUES(%s, %s, %s, %s)',
(id, name, salary, dept))
except Exception as e:
print('error', e)
# 进行交易.
conn.commit()
# 从表中获取数据的功能
def fetch_data():
conn, cur = connect()
# 从emp中选择所有行
try:
cur.execute('SELECT * FROM emp')
except:
print('error !')
# 将结果存储在数据中
data = cur.fetchall()
# 返回结果
return data
# 打印数据的功能
def print_data(data):
print('Query result: ')
print()
# 遍历表中的所有行
for row in data:
# 打印列
print('id: ', row[0])
print('name: ', row[1])
print('salary: ', row[2])
print('dept: ', row[3])
print('----------------------------------')
# 删除表格的功能
def delete_table():
conn, cur = connect()
# 删除表格
try:
cur.execute('删除表emp')
except Exception as e:
print('error', e)
conn.commit()
# 测试代码
if __name__ == '__main__':
# 创建表
create_table()
# 插入一些值
insert_data(1, 'adith', 1000, 2)
insert_data(2, 'tyrion', 100000, 2)
insert_data(3, 'jon', 100, 3)
insert_data(4, 'daenerys', 10000, 4)
# 获取所有行
data = fetch_data()
# 打印行
print_data(data)
# 完成程序后,删除表
delete_table()
输出: