class JamrDao(object):
def __init__(self):
self.userDS = DataSource("database.pickle.user")
self.roomDS = DataSource("database.pickle.room")
self.uidToWsKeyDic = {}
self.uidToWs = {}
def associateKey(self, uuid, wskey):
self.uidToWsKeyDic[uuid] = wskey
def getKey(self,uuid):
return self.uidToWsKeyDic[uuid]
def insertWebsocket(self, peer_address, ws):
self.uidToWs[peer_address] = ws
def getWebsocket(self,peer_address):
return self.uidToWs[peer_address]
def insertUser(self, user):
self.userDS.put(user['uid'], user)
def getUser(self, uuid):
'''
Returns null on not found
'''
return self.userDS.get(uuid)
def insertRoom(self, room):
self.roomDS.put(room['uid'], room)
def getRoom(self, uuid):
'''
Returns null on not found
'''
return self.roomDS.get(uuid)
def getAllUsers(self):
return self.userDS.getAll()
def getAllWebSockets(self):
return self.uidToWs.values()
from DataSource import DataSource
ds = DataSource()
ds.put("1", "1")
print ds.get("1")
print ds.get("2")
ds.save()
from DataSource import DataSource
ds = DataSource()
ds.put("1","1")
print ds.get("1")
print ds.get("2")
ds.save()
