class RoomMapper:
def __init__(self):
self.uow = UnitOfWork(self)
self.identitymap = RoomIdentityMap()
self.tdg = RoomTDG()
def insert(self, roomNumber):
room = Room(roomNumber)
self.identitymap.add(room)
self.uow.registerNew(room)
def commit(self):
self.uow.commit()
def applyInsert(self, objects):
for obj in objects:
self.tdg.insert(obj.roomNumber)
def applyUpdate(self):
pass
def applyDelete(self):
pass
def getRooms(self):
return self.tdg.getRooms()
class UserMapper:
def __init__(self):
self.uow = UnitOfWork(self)
self.identitymap = UserIdentityMap()
self.tdg = UserTDG()
def insert(self, username, password):
user = User(username, password)
self.identitymap.add(user)
self.uow.registerNew(user)
def isRegistered(self, username, password):
isRegistered = True
if self.tdg.isRegistered(username, password) == 0:
isRegistered = False
return isRegistered
def commit(self):
self.uow.commit()
def applyInsert(self, objects):
for obj in objects:
self.tdg.insert(obj.username, obj.password)
def applyUpdate(self):
pass
def applyDelete(self):
pass
def getListByCond(cls, *args, **kws):
cr = QueryCriteria(cls).filter(*args)
if kws.has_key('limit'):
cr.limit(int(kws['limit']))
if kws.has_key('offset'):
cr.offset(int(kws['offset']))
return UnitOfWork.inst().getListByCond(cr)
def createByBiz(cls, **kwargs):
'''
创建实体并自己注册到工作单元内
@param cls: 实体类型
'''
unitofwork = UnitOfWork.inst()
entity = cls(**kwargs)
unitofwork.register(entity)
return entity
def createByBiz(cls, **kwargs):
'''
创建实体并自己注册到工作单元内
@param cls: 实体类型
'''
unitofwork = UnitOfWork.inst()
entity = cls(**kwargs)
unitofwork.register(entity)
return entity
def getListByCond(cls, *args, **kws):
cr = QueryCriteria(cls).filter(*args)
if kws.has_key('limit'):
cr.limit(int(kws['limit']))
if kws.has_key('offset'):
cr.offset(int(kws['offset']))
return UnitOfWork.inst().getListByCond(cr)
def createByBiz(cls, **kwargs):
'''
创建实体并自己注册到工作单元内
@param cls: 实体类型
'''
if not kwargs.get('use_autoincrement_id'):
primaryKey = cls.primaryKey()
unitofwork = UnitOfWork.inst()
if not kwargs.has_key(primaryKey):
kwargs[primaryKey] = unitofwork.idgenerator().get()
entity = cls(**kwargs)
unitofwork.register(entity)
return entity
def createByBiz(cls, **kwargs):
'''
创建实体并自己注册到工作单元内
@param cls: 实体类型
'''
if not kwargs.get('use_autoincrement_id'):
primaryKey = cls.primaryKey()
unitofwork = UnitOfWork.inst()
if not kwargs.has_key(primaryKey):
kwargs[primaryKey] = unitofwork.idgenerator().get()
entity = cls(**kwargs)
unitofwork.register(entity)
return entity
def get(cls, **kwargs):
return UnitOfWork.inst().get(cls, tuple([kwargs.get(key) for key in cls._primary_key]))
def getListByCond(cls, condition='', *args):
return UnitOfWork.inst().getListByCond(cls, condition, args)
def getListByCond(cls, condition='', *args):
return UnitOfWork.inst().getListByCond(cls, condition, args)
def get(cls, **kwargs):
return UnitOfWork.inst().get(
cls, tuple([kwargs.get(key) for key in cls._primary_key]))
def getList(cls, entity_ids):
return UnitOfWork.inst().getList(cls, entity_ids)
class ReservationMapper:
def __init__(self):
self.uow = UnitOfWork(self)
self.identitymap = ReservationIdentityMap()
self.tdg = ReservationTDG()
def insert(self, username, roomNumber, status, timeslot, timestamp):
r = Reservation(username, roomNumber, status, timeslot, timestamp)
self.identitymap.add(r)
self.uow.registerNew(r)
def find(self, username, roomNumber, timeslot):
r = self.identitymap.find(self.getHash(username, roomNumber, timeslot))
if r is None:
r = self.loadReservation(self.tdg.find(username, roomNumber, timeslot))
if r is not None:
self.identitymap.add(r)
return r
def delete(self, username, roomNumber, timeslot):
r = self.identitymap.find(self.getHash(username, roomNumber, timeslot))
if r is None:
r = self.loadReservation(self.tdg.find(username, roomNumber, timeslot))
if r is not None:
self.uow.registerRemoved(r)
else:
self.identitymap.delete(r)
self.uow.registerRemoved(r)
def updatePendingReservation(self, roomNumber, timeslot):
r = self.identitymap.findNextPendingReservation(roomNumber, timeslot)
if r is None:
r = self.loadReservation(self.tdg.findNextPendingReservation(roomNumber, timeslot))
if r is not None:
self.identitymap.add(r)
self.uow.registerDirty(r)
else:
self.identitymap.setFilled(r)
self.uow.registerDirty(r)
def isTimeslotReserved(self, roomNumber, timeslot):
isReserved = True
r = self.identitymap.findReserved(roomNumber, timeslot)
if r is None:
if self.tdg.getFilledCount(roomNumber, timeslot) == 0:
isReserved = False
return isReserved
def hasReservation(self, username, roomNumber, timeslot):
hasReservation = True
r = self.identitymap.find(self.getHash(username, roomNumber, timeslot))
if r is None:
if self.tdg.getReservationCount(username, roomNumber, timeslot) == 0:
hasReservation = False
return hasReservation
def removeFromAllOtherWaitingLists(self, username, roomNumber, timeslot):
reservations = self.identitymap.findAllOtherPendingReservations(username, roomNumber, timeslot)
if reservations:
for r in reservations:
self.uow.registerRemoved(r)
self.identitymap.deleteAll(reservations)
self.tdg.deleteAllOtherPendingReservations(username, roomNumber, timeslot)
def getReservations(self, roomNumber, startTimeslot):
return self.tdg.getReservations(roomNumber, startTimeslot)
def getReservationForUsername(self, username, status):
return self.tdg.getReservationsForUsername(username, status)
def getNumOfReservations(self, username, timeslot):
return self.tdg.getNumOfReservations(username, timeslot)
# Called by UnitOfWork
def applyInsert(self, objects):
for obj in objects:
self.tdg.insert(obj.username, obj.roomNumber, obj.status, obj.timeslot, obj.timestamp)
def applyDelete(self, objects):
for obj in objects:
self.tdg.delete(obj.username, obj.roomNumber, obj.timeslot)
def applyUpdate(self, objects):
# Update the status of a reservation
for obj in objects:
self.tdg.setFilled(obj.username, obj.roomNumber, obj.timeslot)
def commit(self):
self.uow.commit()
def loadReservation(self, row):
reservation = None
if row:
reservation = Reservation(row[0], row[1], row[2], row[3], row[4])
return reservation
def getHash(self, username, roomNumber, timeslot):
lst = [username, roomNumber, timeslot]
return md5(''.join(str(s) for s in lst)).hexdigest()
def __init__(self):
self.uow = UnitOfWork(self)
self.identitymap = ReservationIdentityMap()
self.tdg = ReservationTDG()
def getList(cls, entity_ids):
return UnitOfWork.inst().getList(cls, entity_ids)
def getListByCond(cls, condition, *args, **kwargs):
return UnitOfWork.inst().getListByCond2(cls, condition, args, **kwargs)
def __init__(self):
self.uow = UnitOfWork(self)
self.identitymap = RoomIdentityMap()
self.tdg = RoomTDG()
def getListByCond(cls, condition, *args, **kwargs):
return UnitOfWork.inst().getListByCond2(cls, condition, args, **kwargs)
def getUnitOfWork(self):
if not self._unitofwork:
self._unitofwork = UnitOfWork.inst()
return self._unitofwork
def getUnitOfWork(self):
if not self._unitofwork:
self._unitofwork = UnitOfWork.inst()
return self._unitofwork
def __init__(self):
self.uow = UnitOfWork(self)
self.identitymap = UserIdentityMap()
self.tdg = UserTDG()