X-Git-Url: http://dolda2000.com/gitweb/?p=wrw.git;a=blobdiff_plain;f=wrw%2Fsession.py;h=814182759d584e30e5aff52df6c7781815e74458;hp=1e615e3d22b52cb4ded26acb05fd3cd4b8252b86;hb=HEAD;hpb=9bc70dab473c7057a20d182ce8566ad0a70a7c5d diff --git a/wrw/session.py b/wrw/session.py index 1e615e3..8141827 100644 --- a/wrw/session.py +++ b/wrw/session.py @@ -1,23 +1,14 @@ import threading, time, pickle, random, os -import cookie, env +from . import cookie, env, proto __all__ = ["db", "get"] -def hexencode(str): - ret = "" - for byte in str: - ret += "%02X" % (ord(byte),) - return ret - def gennonce(length): - nonce = "" - for i in xrange(length): - nonce += chr(random.randint(0, 255)) - return nonce + return os.urandom(length) class session(object): def __init__(self, lock, expire=86400 * 7): - self.id = hexencode(gennonce(16)) + self.id = proto.enhex(gennonce(16)) self.dict = {} self.lock = lock self.ctime = self.atime = self.mtime = int(time.time()) @@ -86,7 +77,7 @@ class db(object): def clean(self): now = int(time.time()) with self.lock: - clist = self.live.keys() + clist = list(self.live.keys()) for sessid in clist: with self.lock: try: @@ -204,7 +195,7 @@ class db(object): data = self.backdb[sessid] try: return pickle.loads(data) - except Exception, e: + except: raise KeyError() def freeze(self, sess):