def test_date_functions(self):
allocation = Allocation(raster=60, resource=uuid())
allocation.start = datetime(2011, 1, 1, 12, 30)
allocation.end = datetime(2011, 1, 1, 14, 00)
self.assertEqual(allocation.start.hour, 12)
self.assertEqual(allocation.start.minute, 0)
self.assertEqual(allocation.end.hour, 13)
self.assertEqual(allocation.end.minute, 59)
start = datetime(2011, 1, 1, 11, 00)
end = datetime(2011, 1, 1, 12, 05)
self.assertTrue(allocation.overlaps(start, end))
self.assertFalse(allocation.contains(start, end))
start = datetime(2011, 1, 1, 13, 00)
end = datetime(2011, 1, 1, 15, 00)
self.assertTrue(allocation.overlaps(start, end))
self.assertFalse(allocation.contains(start, end))
def test_date_functions(self):
allocation = Allocation(raster=60, resource=uuid())
allocation.start = datetime(2011, 1, 1, 12, 30)
allocation.end = datetime(2011, 1, 1, 14, 00)
self.assertEqual(allocation.start.hour, 12)
self.assertEqual(allocation.start.minute, 0)
self.assertEqual(allocation.end.hour, 13)
self.assertEqual(allocation.end.minute, 59)
start = datetime(2011, 1, 1, 11, 00)
end = datetime(2011, 1, 1, 12, 05)
self.assertTrue(allocation.overlaps(start, end))
self.assertFalse(allocation.contains(start, end))
start = datetime(2011, 1, 1, 13, 00)
end = datetime(2011, 1, 1, 15, 00)
self.assertTrue(allocation.overlaps(start, end))
self.assertFalse(allocation.contains(start, end))
def move_allocation(
self, master_id, new_start=None, new_end=None,
group=None, new_quota=None, approve_manually=None,
reservation_quota_limit=0, whole_day=None):
assert master_id
assert any([new_start and new_end, group, new_quota])
# Find allocation
master = self.allocation_by_id(master_id)
mirrors = self.allocation_mirrors_by_master(master)
changing = [master] + mirrors
ids = [c.id for c in changing]
assert(group or master.group)
# Simulate the new allocation
new_start = new_start or master.start
new_end = new_end or master.end
if whole_day:
new_start, new_end = utils.align_range_to_day(new_start, new_end)
new = Allocation(start=new_start, end=new_end, raster=master.raster)
# Ensure that the new span does not overlap an existing one
existing_allocations = self.allocations_in_range(new.start, new.end)
for existing in existing_allocations:
if existing.id not in ids:
raise OverlappingAllocationError(new.start, new.end, existing)
for change in changing:
if change.partly_available:
# confirmed reservations
for reservation in change.reserved_slots:
if not new.contains(reservation.start, reservation.end):
raise AffectedReservationError(reservation)
# pending reservations
if change.is_master: # (mirrors return the same values)
for pending in change.pending_reservations.with_entities(
Reservation.start, Reservation.end):
if not new.contains(*pending):
raise AffectedPendingReservationError(pending)
else:
# confirmed reservations
if change.start != new.start or change.end != new.end:
if len(change.reserved_slots):
raise AffectedReservationError(
change.reserved_slots[0]
)
if change.is_master and \
change.pending_reservations.count():
raise AffectedPendingReservationError(
change.pending_reservations[0]
)
# the following attributes must be equal over all group members
# (this still allows to use move_allocation to remove an allocation
# from an existing group by specifiying the new group)
for allocation in self.allocations_by_group(group or master.group):
if approve_manually is not None:
allocation.approve_manually = approve_manually
if reservation_quota_limit is not None:
allocation.reservation_quota_limit = reservation_quota_limit
if new_quota is not None and allocation.is_master:
self.change_quota(allocation, new_quota)
for change in changing:
change.start = new.start
change.end = new.end
change.group = group or master.group
