PythonでPostgreSQLを操作する方法

概要

PythonでPostgreSQLを操作するライブラリpsycopg2をインストールし、Pythonプログラムから実行するまでの手順を説明します。

用語

・Ubuntu：LinuxのOS。
・Python：機械学習プログラミングに人気のプログラミング言語。
・PostgresSQL：オープンソースのRDBMS。
・psycopg2：PythonからPostgreSQLを操作するためのライブラリ。

事前準備（前提とする環境）

・Ubuntu14.04
・Python2（バージョン2.7.2）
・PostgresSQL（バージョン9.3.24）

手順

psycopg2をインストール

psycopg2をインストールします。

$ sudo apt-get install python-dev libpq-dev
$ sudo apt-get install python-psycipg2
$ sudo pip install -U psycopg2

Pythonのサンプルプログラム（sample.py）

以下のとおりプログラムを書きます。今回は、pythonのPostgreSQLのバージョンを取得するだけのクエリです。

import sys
import psycopg2

# メイン処理
def main():
    # 事前処理（コネクションの生成）
    # パラメータでDB名、ユーザー名、パスワード、ホスト名（IPアドレス）を指定する
    conn = psycopg2.connect('dbname=hoge user=postgres password=postgres host=172.16.0.1')
    cur = conn.cursor()

    # クエリ実行
    cur.execute("select version()")

    # コミット（今回はselectなのでコミットは不要）
    # conn.commit()

    # 事後処理（コネクションクローズ）
    conn.close()

# プログラム開始処理
if __name__ == '__main__':
    main()

実行結果

$ python sample.py
                                                   version
--------------------------------------------------------------------------------------------------------------
 PostgreSQL 9.3.24 on x86_64-unknown-linux-gnu, compiled by gcc (Ubuntu 4.8.4-2ubuntu1~14.04.4) 4.8.4, 64-bit