介绍

PostgreSQL是一个开放源代码的对象关系数据库管理系统。PostgreSQL符合ACID且具有事务性。它具有触发器，外键并支持功能和存储过程。

PostgreSQL被Uber，Apple，Netflix和Instagram等巨头使用。

在本文中，我们将看到如何从Python Script连接到PostgreSQL并执行查询操作。

安装依赖环境：

使用python 3创建一个虚拟环境并激活它。安装以下软件包。

psycopg2==2.7.3.2

安装：

使用以下命令安装PostgreSQL数据库和实用程序。

$sudo apt-get update 
$sudo apt-get install postgresql postgresql-contrib

默认情况下，PostgreSQL在新安装时会设置用户和数据库“ postgres”。我们需要切换到该用户来使用postgres数据库。

$sudo -su postgres

现在通过在终端上输入psql进入Postgres提示符。

我们正在使用版本10.3。

如果在连接数据库时遇到任何错误，请确保PostgreSQL正在运行。使用以下命令检查状态。

$systemctl status postgresql

您可以使用以下命令检查日志中的错误。

$tail -f /var/log/postgresql

创建数据库：

在创建新数据库之前，让我们列出所有数据库。使用\l 或 \list相同。

要创建数据库，请键入 \q 退出psql终端，然后使用命令createdb testdb。

postgres@brahma:~$ createdb testdb
postgres@brahma:~$ psql
psql (10.3 (Ubuntu 10.3-1.pgdg16.04+1))
Type "help" for help.

postgres=# \l
 List of databases
 Name | Owner | Encoding | Collate | Ctype | Access privileges 
---------------+----------+----------+---------+-------+-----------------------
 postgres | postgres | UTF8 | en_IN | en_IN | 
 rana_test | postgres | UTF8 | en_IN | en_IN | 
 template0 | postgres | UTF8 | en_IN | en_IN | =c/postgres +
 | | | | | postgres=CTc/postgres
 template1 | postgres | UTF8 | en_IN | en_IN | =c/postgres +
 | | | | | postgres=CTc/postgres
 testdb | postgres | UTF8 | en_IN | en_IN | 
(5 rows)

postgres=# \c testdb
You are now connected to database "testdb" as user "postgres".
testdb=# 

创建表：

PostgreSQL中的大多数查询sytanx与MySQL相同。

create table users (
 id serial PRIMARY KEY,
 username varchar (20) NOT NULL,
 age smallint NOT NULL,
 location varchar (50) NOT NULL
); 

将上述语法复制粘贴到终端中，将创建新表。您可以通过键入\ d列出表。

testdb=# create table users (
testdb(# username varchar (20) NOT NULL,
testdb(# age smallint NOT NULL,
testdb(# location varchar (50) NOT NULL
testdb(# );
CREATE TABLE
testdb=# \d
 List of relations
 Schema | Name | Type | Owner 
--------+-------+-------+----------
 public | users | table | postgres
(1 row)

testdb=# 

您可以通过访问官方网站了解有关从psql终端查询的更多信息。现在，让我们尝试使用Python代码连接PostgreSQL并执行操作。

从Python脚本连接：

我们在虚拟环境中安装了psycopg软件包。使用Python脚本中的以下代码连接到数据库。

import psycopg2


# this function will return the connection object
def connect():
 conn = None
 try:
 conn = psycopg2.connect(host="localhost", user="postgres", password="root", database="testdb")
 except Exception as e:
 print(repr(e))

 return conn 

将数据插入表中：

首先获取连接和游标，然后创建查询。执行查询后，使用连接提交并关闭游标和连接。

conn = connect()
cur = conn.cursor()

last_insert_id = None

# inserting data in users table
sql_query = "insert into users (username, age, location) values (%s, %s, %s) returning id;"

sql_data = (
 "Ajay",
 "25",
 "New York"
)


cur.execute(sql_query, sql_data)
last_insert_id = cur.fetchone()[0]
print("Last Insert ID " + str(last_insert_id))

conn.commit()
cur.close()
conn.close()

return last_insert_id 

我们将数据插入表中并返回主键ID（即序列号）。

从表中获取数据：

PostgreSQL的选择查询与MySQL相同。

conn = connect()
cur = conn.cursor()

sql_query = "select username, age, location from users where location = %s;"
sql_data = ("Delhi")
cur.execute(sql_query, sql_data)

results = cur.fetchall()
return results

更新行：

conn = connect()
cursor = conn.cursor()

sql_query = "update users set location = %s where username = %s;"
sql_data = ("Mumbai", "Ajay")

cursor.execute(sql_query, sql_data)

cursor.close()
conn.close()

return True

要退出终端，请使用\ q命令。

总结：

以上介绍了PostgreSQL的基本用法及环境搭建，并且使用简单的Python脚本对PostgreSQL进行了基础操作，希望对大家有所帮助。