今日の変更

Pythonを使ったアンケートとかができるシステム、K-enq.tdの制作が進んでいます。今日の変更は以下のとおり。

HTMLクラス

• cookieの出力に対応させた

Sessionクラス

• ハッシュ生成をMD5からhashlibに変更、ついでにSHA256を採用
• os.environ['ENV']という書き方を、os.environ.get('ENV', u'')と変更
• セッション削除のメソッド、delete()等、いくつか追加

多分これくらい。とりあえず、簡単なセッションの管理ができるようになりました。

os.environ['ENV']からos.environ.get('ENV', u'')と変更したのは、前者の書き方だと、コマンドラインでの実行時にエラーが出るから。けど、後者の書き方、こういうのはどのへんを見るとわかるんだろう。本を買ったらわかるのかな。

また、MD5からhashlibに変えたのは、MD5がリリース 2.5で撤廃されたことにともなう変更です。コマンドラインで実行したら、警告が出ていた。それを受けてhashlibにおきかえて、ついでにアルゴリズムをSHA256にした。単純な理由です。

せっかくなので、Sessionだけでも掲載しておきます。

# -*- coding: utf-8 -*-
'''
Class for provide session with sqlite3
'''

import os
import datetime
import sys
import random
import hashlib
import sqlite3

class Session:
    '''Class to provide session.'''

    def __init__(self, dbpath, sid=None, validity=u'3 hours'):
        self.sid = sid
        self.dbpath = dbpath
        self.dbtablename = u'sessiontable'
        self.validity = validity
        self.data = None
        
        connection = self._open_db()
        cursor = connection.cursor()
        cursor.execute('select * from sqlite_master \
        where type=\'table\' and name=?;', \
                       (self.dbtablename, ))
        tablecount = cursor.fetchall()
        if len(tablecount) == 0:
            cursor.execute('create table %s (id, data, created_time, \
            accessed_time, expire_time, remote_addr);' % self.dbtablename)

        cursor.execute('delete from %s where expire_time<datetime(\'now\');' \
                       % self.dbtablename)

        if isinstance(self.sid, basestring):
            cursor.execute('select id from %s where id=\'%s\';' \
                           % (self.dbtablename, self.sid))
            idcount = cursor.fetchall()
            if len(idcount) == 0:
                self._create_session_id()
                self._insert_session_record(cursor)
            else:
                self._update_session_record(cursor)
        else:
            self._create_session_id()
            self._insert_session_record(cursor)

        cursor.close()
        connection.commit()
        connection.close()



    def get_id(self):
        return self.sid

    def get_data(self):
        return self.data

    def get_created_time(self):
        connection = self._open_db()
        cursor = connection.cursor()
        cursor.execute('select created_time from %s where id=\'%s\';' \
                       % (self.dbtablename, self.sid))
        created_time = cursor.fetchone()
        cursor.close()
        connection.close()
        return created_time[0]

    def get_accessed_time(self):
        connection = self._open_db()
        cursor = connection.cursor()
        cursor.execute('select accessed_time from %s where id=\'%s\';' \
                       % (self.dbtablename, self.sid))
        accessed_time = cursor.fetchone()
        cursor.close()
        connection.close()
        return accessed_time[0]
        
    def get_expire_time(self):
        connection = self._open_db()
        cursor = connection.cursor()
        cursor.execute('select expire_time from %s where id=\'%s\';' \
                       % (self.dbtablename, self.sid))
        expire_time = cursor.fetchone()
        cursor.close()
        connection.close()
        return expire_time[0]



    def delete(self):
        connection = self._open_db()
        cursor = connection.cursor()
        cursor.execute('delete from %s where id=\'%s\';' \
                       % (self.dbtablename, self.sid))
        cursor.close()
        connection.commit()
        connection.close()


    # internal methods

    def _open_db(self):
        return sqlite3.connect(self.dbpath)

    def _create_session_id(self):
        connection = self._open_db()
        cursor = connection.cursor()
        while True:
            now = datetime.datetime.today()
            seed = str(os.getpid()) + \
                   str(now.isoformat()) + \
                   str(random.randint(0, sys.maxint - 1))
            message = hashlib.new('sha256')
            message.update(seed)
            sid = message.hexdigest()
            cursor.execute('select id from %s where id=\'%s\';' \
                           % (self.dbtablename, sid))
            idcount = cursor.fetchall()
            if len(idcount) == 0:
                self.sid = sid
                break
        cursor.close()
        connection.close()

    def _insert_session_record(self, cursor):
        cursor.execute('insert into %s (id, created_time, accessed_time, \
        expire_time, remote_addr) values(\'%s\', datetime(\'now\'), \
        datetime(\'now\'), datetime(\'now\', \'%s\'), \'%s\');' \
                       % (self.dbtablename, self.sid, self.validity, \
                          os.environ.get('REMOTE_HOST', u'')))

    def _update_session_record(self, cursor):
        cursor.execute('update %s set accessed_time=datetime(\'now\'), \
        expire_time=datetime(\'now\', \'%s\'), remote_addr=\'%s\' \
        where id=\'%s\';' \
                       % (self.dbtablename, self.validity, \
                          os.environ.get('REMOTE_HOST', u''), self.sid))