PostgreSQLにサクッとテストデータを登録するPythonテンプレ

Python手軽でいいね、ってことで

最近はJavascriptよりもPythonでスクリプトを書いている。

今日もパフォーマンステスト用に大量データを登録する必要があったので

SQLAlchemyを使ってスクリプトを書いてみた。

Python3で動作確認済み。

create_test_datas.py

# -*- coding: utf-8 -*-

import sqlalchemy

engine = sqlalchemy.create_engine("postgresql+psycopg2://＜DBユーザー名＞:＜DBパスワード＞@/＜DB名＞", client_encoding="utf8")

def create_table_defs(conn, meta):
    """テーブル定義"""
    tables = { 
        "items": sqlalchemy.Table("t_item", 
            meta, 
            autoload=True, 
            autoload_with=conn, 
            postgresql_ignore_search_path=True)
    }   

    return tables

def create_test_items(item_table):
    """テストデータ登録"""
    for i in range(1, 20000):
        item = {
            "item_cd": "S%08d" % i,
            "item_name1": "テスト商品%d" % i,
            "unit": "個"
        }
        item_table.insert().execute(**item)

with engine.connect() as conn:
    conn.execute("SET search_path TO public")
    meta = sqlalchemy.MetaData()
    meta.bind = engine
    tables = create_table_defs(conn, meta)
    create_test_items(tables["items"])