介绍

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 testdbpostgres@brahma:~$ psqlpsql (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 testdbYou 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 NULLtestdb(# );CREATE TABLEtestdb=# \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 objectdef 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 tablesql_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进行了基础操作，希望对大家有所帮助。