はじめに
PythonでシンプルにDBへアクセスできる dataset を試しました。
前提条件
以下がインストール済みであることを前提とします。
- pipenv
- sqlite3
動作確認した環境のバージョンは以下のとおりです。
$ pipenv --version
pipenv, version 2018.11.26
$ sqlite3 --version
3.16.0 2016-11-04 19:09:39 0e5ffd9123d6d2d2b8f3701e8a73cc98a3a7ff5f
環境構築
$ mkdir sample && cd sample
$ pipenv --python 3.7
$ pipenv install dataset
$ cat <<EOF | sqlite3 ./sqlite3.db
CREATE TABLE depts(id integer primary key autoincrement, name text);
CREATE TABLE users(id integer primary key autoincrement, name text, email text, dept_id integer);
insert into depts (name) values ('A');
insert into depts (name) values ('B');
insert into depts (name) values ('C');
EOF
ソースコード
test.py
#!/usr/bin/env python
import dataset, traceback, json
db = dataset.connect('sqlite:///sqlite3.db')
users = db['users']
try:
db.begin()
users.delete()
users.insert(dict(name='test1', email='test1@example.com', dept_id=1))
users.insert(dict(name='test2', email='test2@example.com', dept_id=2))
users.insert(dict(name='test3', email='test3@example.com', dept_id=3))
users.insert(dict(name='test4', email='test3@example.com', dept_id=3))
users.insert(dict(name='test5', email='test3@example.com', dept_id=3))
db.commit()
except Exception as e:
traceback.print_exc()
db.rollback()
rows = db.query("""
select
u.id
, u.name
, u.email
, u.dept_id
, d.name as dept_name
from users u
left join depts d
on u.dept_id = d.id
where
u.dept_id = :dept_id
""", dept_id=3)
print("Result:")
for row in rows:
print(json.dumps(row))
実行
$ pipenv run python test.py
Result:
{"id": 3, "name": "test3", "email": "test3@example.com", "dept_id": 3, "dept_name": "C"}
{"id": 4, "name": "test4", "email": "test3@example.com", "dept_id": 3, "dept_name": "C"}
{"id": 5, "name": "test5", "email": "test3@example.com", "dept_id": 3, "dept_name": "C"}
とてもシンプルに書けていい感じです！！