PostgreSQL | psycopg2语句汇总

psycopg2是Python语言的PostgreSQL数据库接口，是对Psycopg 1.1.x版本进行的几乎完全的改写。它的主要优势在于完全支持Python DB API 2.0，以及安全的多线程支持。它适用于随时创建、销毁大量游标的、和产生大量并发INSERT、UPDATE操作的多线程数据库应用。Psycopg包内含 ZPsycopgDA，一个Zope数据库接口。

                   python PostgreSQL连接访问数据库

1. psycopg2的安装

pip3 install psycopg2

2. 导入包

import psycopg2
import psycopg2.extras

3. 类介绍

3.1 connection类

connection类表示数据库连接对象。由psycopg2.connect()方法创建。

- 创建connection对象

psycopg2.connect()函数创建一个新的数据库会话（Session）并且返回一个连接对象。该函数的参数为：

　　· dbname – 数据库名字

　　· user – 数据库角色名称

　　· password – 数据库角色密码

　　· host – 数据库地址

　　· port – 端口

- 主要方法

　　· commit()：提交任何未提交的事务（transaction）到数据库。

　　· rollback()：回滚。

　　· close()：关闭数据库。如果关闭数据库时仍有未提交的事务，则执行回滚操作。

3.2 cursor类

用来在数据库Session里执行PostgreSQL命令。

- 创建对象

cursor对象由connection.cursor()方法创建：

cur = conn.cursor()

- 主要方法

　　· execute(query, vars=None)：执行SQL语句。

　　· fetchall()：获取所有查询结果，返回值为tuple列表。

4. 具体操作

大致流程是：

　　1）连接PostgreSQL并设定Cursor

　　2）执行SQL脚本

　　#有几点需要注意　　

      - sql脚本必须以';'结尾，不可以省略

　　- 不管sql中有几个参数，都需要用%s代替，只有%s, 不管值是字符还是数字, 一律%s.

　　- 第二个参数中，一定要传入元组，上述例子中(5)是不行的

　　- 执行完INSERT, UPDATE, DELETE这样的sql脚本后, 需要conn.commit()提交一下, 才会把数据提交到数据库当中.

　　3）得到查询结果

可以采用fetchone(), fetchall()和fetchmany() 三种方式。fetchone返回一个tuple或者None, fetchall返回一个list of tuple，如果没有结果则返回一个空的tuple。fetchmany返回list of tuple， list的长度由size参数决定，size的默认值是cursor.arraysize， 如果没有结果可以返回，那么返回一个空的list。

4.1 建立连接 

conn = psycopg2.connect(dbname="mydb", user="postgres",
        password="1234", host="127.0.0.1", port="5432")

4.2 创建表

cur.execute(
        'CREATE TABLE Employee ('
        'name    varchar(80),'
        'address varchar(80),'
        'age     int,'
        'date    date'
        ')'
    )

4.3 插入数据

cur.execute("INSERT INTO Employee "
        "VALUES('Gopher', 'China Beijing', 100, '2017-05-27')")

4.4 查询数据

cur.execute("SELECT * FROM Employee")
rows = cur.fetchall()
for row in rows:
    print('name=' + str(row[0]) + ' address=' + str(row[1]) + 
        ' age=' + str(row[2]) + ' date=' + str(row[3]))

4.5 更新数据

cur.execute("UPDATE Employee SET age=12 WHERE name='Gopher'")

4.6 删除数据

cur.execute("DELETE FROM Employee WHERE name='Gopher'")

参考：https://pynative.com/python-postgresql-tutorial/?utm_source=mybridge&utm_medium=blog&utm_campaign=read_more