HARD DAY'S NIGHT ブログ

今回は、複雑な SELECT 文に対応する select_custom 関数を追加して、簡単なテーブルでテストをしてシリーズ最後にします。

select_custom 関数は、引数 sql にパラメータ(%s)付きSQLステートメント、bind にパラメータの値リストを渡します。

class SimpleDB(object):
... （省略）...

    def select_custom(self, sql, bind):
        result = None

        if self.__cursor is not None:
            self.__cursor.execute(sql, bind)
            columns = self.__cursor.description
            result = []
            for row in self.__cursor.fetchall():
                item = {}
                for (index, value) in enumerate(row):
                    item[columns[index][0]] = value
                result.append(item)
        return result
    ## End def select_custom

... （省略）...
## End class

次に、SimpleDBクラスで、テーブルへデータ登録、更新、削除、そして選択してテストします。

テーブル名は t_test でテーブル定義は以下です。

CREATE TABLE t_test (
     id                int NOT NULL AUTO_INCREMENT
    ,name              varchar(260) NULL
    ,PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 AUTO_INCREMENT=1;

以下、テスト用コードです。

if __name__ == '__main__':
    dbconfig = {
        "host": "localhost",
        "database": "<your database>",
        "user": "<your username>",
        "password": "<your secure password>"
        }

    try:
        db = SimpleDB(dbconfig['host'], dbconfig['database'], dbconfig['user'], dbconfig['password'])
        db.open()

        # insert
        data = { 'name' : '山田　太郎' }
        rowid = db.insert("t_test", data)
        print "rowid: %d" % rowid


        # select
        conditions = { 'id' : rowid }
        res = db.select("t_test", conditions, "id", True)
        if res is not None:
            for r in res:
                print "select : %s, %s" % (r['id'], r['name'])

        # select_custom
        bind = [ rowid ]
        res = db.select_custom("select * from t_test where id = %s order by id", bind)
        if res is not None:
            for r in res:
                print "select_custom : %s, %s" % (r['id'], r['name'])

        # update
        conditions = { 'id' : rowid }
        data = { 'name' : '山田　花子' }
        rows_affected = db.update("t_test", data, conditions)
        print "%d 件更新" % rows_affected

        # select all
        conditions = None
        res = db.select("t_test", conditions, "id", True)
        if res is not None:
            for r in res:
                print "select : %s, %s" % (r['id'], r['name'])

        # delete
        conditions = { 'name' : '山田　花子' }
        rows_affected = db.delete("t_test", conditions)
        print "%d 件削除" % rows_affected

        # select all
        conditions = None
        res = db.select("t_test", conditions)
        if res is not None:
            for r in res:
                print "select : %s, %s" % (r['id'], r['name'])
        
        db.close()
    except Exception as e:
        exc_type, exc_obj, exc_tb = sys.exc_info()
        fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
        print "%s, %s, %s" % (exc_type, fname, exc_tb.tb_lineno)

実行結果：

# python simpledb.py 
rowid: 15
select : 15, 山田　太郎
select_custom : 15, 山田　太郎
1 件更新
select : 15, 山田　花子
1 件削除

参考までに：

SimpleDBクラスのソース: simpledb.py
（ファイル名を simpledb.py_.txt から “simpledb.py” に変更してください。）