Python通过psycopg2操作PostgreSQL

说明

本文假设已经安装和配置好了PostgreSQL,如果还没有安装和配置PostgreSQL,请移步至我的另一篇博文:Debian8下安装配置PostgreSQL

本例中用的操作系统为:

Debian GNU/Linux 8.2 (jessie)

Python版本为:

Python 2.7.9

psycopg2的安装

Debian下执行以下命令安装 psycopg2:

apt-get install python-psycopg2

安装好后,就可以写Python脚本来访问PostgreSQL了。

导入psycopg2

想要使用psycopg2,必须用import语句导入该包:

import psycopg2

connection类

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

创建connection对象

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

  • dbname – 数据库名字
  • user – 数据库角色名称
  • password – 数据库角色密码
  • host – 数据库地址
  • port – 端口

connection的主要方法

connection提供了常用的数据库操作:

  • commit():提交任何未提交的事务(transaction)到数据库。
  • rollback():回滚。
  • close():关闭数据库。如果关闭数据库时仍有未提交的事务,则执行回滚操作。

cursor类

创建cursor对象

psycopg2提供了一个cursor类,用来在数据库Session里执行PostgreSQL命令。cursor对象由connection.cursor()方法创建:

cur = conn.cursor()

cursor的主要方法

  • execute(query, vars=None):执行SQL语句。
  • fetchall():获取所有查询结果,返回值为tuple列表。

有了cursor对象,就可以操作数据库了。

建立数据库连接

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

创建表

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

插入数据

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

查询数据

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]))

更新数据

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

删除数据

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

完整示例

#coding=utf-8

import psycopg2

# 连接数据库
conn = psycopg2.connect(dbname="mydb", user="postgres",
        password="1234", host="127.0.0.1", port="5432")

# 创建cursor以访问数据库
cur = conn.cursor()

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

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

# 查询数据
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]))

# 更新数据
cur.execute("UPDATE Employee SET age=12 WHERE name='Gopher'")

# 删除数据
cur.execute("DELETE FROM Employee WHERE name='Gopher'")

# 提交事务
conn.commit()

# 关闭连接
conn.close()

你可能感兴趣的:(数据库,Python)