def _postprocess(self, obj, fossil, iface):
if iface is IRoomMetadataWithReservationsFossil:
(startDT, endDT) = (self._fromDT or MIN_DATETIME,
self._toDT or MAX_DATETIME)
if self._fromDT or self._toDT:
toDate = self._toDT.date() if self._toDT else None
fromDate = self._fromDT.date() if self._fromDT else None
resvEx = ReservationBase()
resvEx.startDT = startDT
resvEx.endDT = endDT
resvEx.room = obj
resvEx.isRejected = False
resvEx.isCancelled = False
if fromDate != toDate:
resvEx.repeatability = RepeatabilityEnum.daily
resvs = set(c.withReservation for c in resvEx.getCollisions())
else:
resvs = obj.getReservations()
iresvs1, iresvs2 = itertools.tee(itertools.ifilter(self._resvFilter, resvs), 2)
fresvs = fossilize(iresvs1, IRoomReservationMetadataFossil, tz=self._tz, naiveTZ=self._serverTZ)
for fresv, resv in itertools.izip(iter(fresvs), iresvs2):
self._addOccurrences(fresv, resv, startDT, endDT)
fossil['reservations'] = fresvs
return fossil
def rebuildRoomReservationsIndex():
from MaKaC.common.db import DBMgr
from MaKaC.rb_location import CrossLocationDB
from MaKaC.rb_room import RoomBase
from MaKaC.plugins.RoomBooking.default.dalManager import DALManager
from BTrees.OOBTree import OOBTree
DBMgr.getInstance().startRequest()
CrossLocationDB.connect()
root = DALManager.root
resvEx = ReservationBase()
resvEx.isConfirmed = None
allResvs = CrossLocationQueries.getReservations( resvExample = resvEx )
print "There are " + str( len( allResvs ) ) + " resvs and pre-resvs to index..."
c = 0
root[_ROOM_RESERVATIONS_INDEX] = OOBTree()
print "Room => Reservations Index branch created"
for resv in allResvs:
roomReservationsIndexBTree = root[_ROOM_RESERVATIONS_INDEX]
resvs = roomReservationsIndexBTree.get( resv.room.id )
if resvs == None:
resvs = [] # New list of reservations for this room
roomReservationsIndexBTree.insert( resv.room.id, resvs )
resvs.append( resv )
roomReservationsIndexBTree[resv.room.id] = resvs
c += 1
if c % 100 == 0:
print c
CrossLocationDB.commit()
CrossLocationDB.disconnect()
DBMgr.getInstance().endRequest()
def changeCreator(oldUser, newUser):
dbi = DBMgr.getInstance()
dbi.startRequest()
Factory.getDALManager().connect()
# check if the users exist
if AvatarHolder().getById(oldUser) is None:
print "There is no user with id %s" % oldUser
return
if AvatarHolder().getById(newUser) is None:
print "There is no user with id %s" % newUser
return
resvEx = ReservationBase()
resvEx.createdBy = oldUser
allResv4OldUser = CrossLocationQueries.getReservations(resvExample=resvEx)
if allResv4OldUser == []:
print "No reservations for user %s" % oldUser
return
# resvs = ReservationBase.getReservations()
# allResv4OldUser = [x for x in allResv if x.createdBy == oldUser]
if type(allResv4OldUser) is not list:
allResv4OldUser = [allResv4OldUser]
# Modify reservations
for r in allResv4OldUser:
r.createdBy = newUser
#print r.createdBy, r.id
# Update index
userReservationsIndexBTree = Reservation.getUserReservationsIndexRoot()
newUserResvs = userReservationsIndexBTree.get(newUser)
if newUserResvs == None:
newUserResvs = [] # New list of reservations for this room
userReservationsIndexBTree.insert(newUser, newUserResvs)
newUserResvs.extend(allResv4OldUser)
userReservationsIndexBTree[newUser] = newUserResvs[:]
if userReservationsIndexBTree.has_key(oldUser):
userReservationsIndexBTree.pop(oldUser)
userReservationsIndexBTree._p_changed = 1
# close DB connection
Factory.getDALManager().commit()
Factory.getDALManager().disconnect()
dbi.endRequest()
print "%s reservations have moved from creator %s to creator %s" % (
len(allResv4OldUser), oldUser, newUser)
def _getAnswer(self):
p = ReservationBase()
p.startDT = self._startDT
p.endDT = self._endDT
p.repeatability = self._repeatability
rooms = CrossLocationQueries.getRooms(location=self._location,
resvExample=p,
available=True)
return [room.id for room in rooms]
def getAverageOccupation(**kwargs):
"""
FINAL (not intented to be overriden)
Returns float <0, 1> representing how often - on the avarage -
the rooms are booked during the working hours. (1 == all the time, 0 == never).
"""
name = kwargs.get('location',
Location.getDefaultLocation().friendlyName)
# Get active, publically reservable rooms
from MaKaC.rb_factory import Factory
roomEx = Factory.newRoom()
roomEx.isActive = True
roomEx.isReservable = True
rooms = CrossLocationQueries.getRooms(roomExample=roomEx,
location=name)
# Find collisions with last month period
from MaKaC.rb_reservation import ReservationBase, RepeatabilityEnum
resvEx = ReservationBase()
now = datetime.now()
resvEx.endDT = datetime(now.year, now.month, now.day, 17, 30)
resvEx.startDT = resvEx.endDT - timedelta(
30, 9 * 3600) # - 30 days and 9 hours
resvEx.repeatability = RepeatabilityEnum.daily
collisions = resvEx.getCollisions(rooms=rooms)
totalWorkingDays = 0
weekends = 0
for day in iterdays(resvEx.startDT, resvEx.endDT):
if day.weekday() in [5, 6]: # Skip Saturday and Sunday
weekends += 1
continue
# if c.startDT is CERN Holiday: continue
totalWorkingDays += 1
booked = timedelta(0)
for c in collisions:
if c.startDT.weekday() in [5, 6]: # Skip Saturday and Sunday
continue
# if c.startDT is CERN Holiday: continue
booked = booked + (c.endDT - c.startDT)
totalBookableTime = totalWorkingDays * 9 * len(rooms) # Hours
bookedTime = booked.days * 24 + 1.0 * booked.seconds / 3600 # Hours
if totalBookableTime > 0:
return bookedTime / totalBookableTime
else:
return 0 # Error (no rooms in db)
def listResv4User(user):
dbi = DBMgr.getInstance()
dbi.startRequest()
Factory.getDALManager().connect()
resvEx = ReservationBase()
resvEx.createdBy = user
allResv = CrossLocationQueries.getReservations( resvExample = resvEx)
print "User %s has %s resevations created by him/her"%(user, len(allResv))
Factory.getDALManager().disconnect()
dbi.endRequest()
def _process( self ):
from MaKaC.rb_room import RoomBase
from datetime import datetime,timedelta
from MaKaC.rb_reservation import ReservationBase
startdt = enddt = datetime.now()
today = startdt.date()
startdt.replace( hour = 0, minute = 0)
enddt.replace( hour = 23, minute = 59)
self._req.content_type = "text/xml"
XG = xmlGen.XMLGen()
XG.openTag("response")
rooms = RoomBase.getRooms()
nbRooms = len(rooms)
nbPublicRooms = nbPrivateRooms = nbSemiPrivateRooms = 0
for r in rooms:
if not r.isReservable:
nbPrivateRooms += 1
elif not r.resvsNeedConfirmation:
nbPublicRooms += 1
else:
nbSemiPrivateRooms += 1
self._createIndicator(XG, "total", "total number of managed rooms", nbRooms)
self._createIndicator(XG, "public", "number of public rooms", nbPublicRooms)
self._createIndicator(XG, "semiprivate", "number of semi-private rooms", nbSemiPrivateRooms)
self._createIndicator(XG, "private", "number of private rooms", nbPrivateRooms)
resvex = ReservationBase()
resvex.isConfirmed = True
resvex.isCancelled = False
nbResvs = len(ReservationBase.getReservations( resvExample = resvex, days = [ startdt.date() ] ))
resvex.usesAVC = True
nbAVResvs = len(ReservationBase.getReservations( resvExample = resvex, days = [ startdt.date() ] ))
resvex.needsAVCSupport = True
resvex.needsAssistance = False
nbAVResvsWithSupport = len(ReservationBase.getReservations( resvExample = resvex, days = [ startdt.date() ] ))
self._createIndicator(XG, "nbbookings", "total number of bookings for today", nbResvs)
self._createIndicator(XG, "nbvc", "number of remote collaboration bookings (video or phone conference)", nbAVResvs)
self._createIndicator(XG, "nbvcsupport", "number of remote collaboration bookings with planned IT support", nbAVResvsWithSupport)
XG.closeTag("response")
return XG.getXml()
def getMyAverageOccupation(self, period="pastmonth"):
"""
FINAL (not intented to be overriden)
Returns float <0, 1> representing how often - on the avarage -
the room is booked during the working hours. (1 == all the time, 0 == never).
"""
# Find collisions with last month period
from MaKaC.rb_reservation import ReservationBase, RepeatabilityEnum
resvEx = ReservationBase()
now = datetime.now()
if period == "pastmonth":
resvEx.endDT = datetime(now.year, now.month, now.day, 17, 30)
resvEx.startDT = resvEx.endDT - timedelta(
30, 9 * 3600) # - 30 days and 9 hours
elif period == "thisyear":
resvEx.endDT = datetime(now.year, now.month, now.day, 17, 30)
resvEx.startDT = datetime(now.year, 1, 1, 0, 0)
resvEx.repeatability = RepeatabilityEnum.daily
collisions = resvEx.getCollisions(rooms=[self])
totalWorkingDays = 0
weekends = 0
for day in iterdays(resvEx.startDT, resvEx.endDT):
if day.weekday() in [5, 6]: # Skip Saturday and Sunday
weekends += 1
continue
# if c.startDT is CERN Holiday: continue
totalWorkingDays += 1
booked = timedelta(0)
for c in collisions:
if c.startDT.weekday() in [5, 6]: # Skip Saturday and Sunday
continue
# if c.startDT is CERN Holiday: continue
booked = booked + (c.endDT - c.startDT)
totalBookableTime = totalWorkingDays * 9 # Hours
bookedTime = booked.days * 24 + 1.0 * booked.seconds / 3600 # Hours
if totalBookableTime > 0:
return bookedTime / totalBookableTime
else:
return 0
def reservation(self, locList):
Factory.getDALManager().connect()
resvEx = ReservationBase()
resvEx.startDT = self._fromDT
resvEx.endDT = self._toDT
locList = filter(lambda loc: Location.parse(loc) is not None, locList)
if self._fromDT or self._toDT:
daysParam = (day.date() for day in rrule.rrule(rrule.DAILY, dtstart=self._fromDT, until=self._toDT))
else:
# slow!
daysParam = None
for loc in sorted(locList):
resvs = CrossLocationQueries.getReservations(location=loc, resvExample=resvEx, days=daysParam)
for obj in self._process(resvs, filter=self._resvFilter):
yield obj
Factory.getDALManager().disconnect()
