blob: db3769f7168083402d11f48607ed0291abd7263b [file] [log] [blame]
import os
import datetime
from django.db import models
from core.models import PlCoreBase
from core.models import Sliver
from core.models import Slice
from core.models import ServiceResource
# Create your models here.
class Reservation(PlCoreBase):
startTime = models.DateTimeField()
slice = models.ForeignKey(Slice, related_name="reservations")
duration = models.IntegerField(default=1)
def __unicode__(self): return u'%s to %s' % (self.startTime, self.endTime)
@property
def endTime(self):
return self.startTime + datetime.timedelta(hours=self.duration)
def can_update(self, user):
return self.slice.can_update(user)
@staticmethod
def select_by_user(user):
if user.is_admin:
qs = Reservation.objects.all()
else:
slice_ids = [s.id for s in Slice.select_by_user(user)]
qs = Reservation.objects.filter(id__in=slice_ids)
return qs
class ReservedResource(PlCoreBase):
sliver = models.ForeignKey(Sliver, related_name="reservedResourrces")
resource = models.ForeignKey(ServiceResource, related_name="reservedResources")
quantity = models.IntegerField(default=1)
reservationSet = models.ForeignKey(Reservation, related_name="reservedResources")
class Meta(PlCoreBase.Meta):
verbose_name_plural = "Reserved Resources"
def __unicode__(self): return u'%d %s on %s' % (self.quantity, self.resource, self.sliver)
def can_update(self, user):
return self.sliver.slice.can_update(user)
@staticmethod
def select_by_user(user):
if user.is_admin:
qs = ReservedResource.objects.all()
else:
sliver_ids = [s.id for s in Sliver.select_by_user(user)]
qs = ReservedResource.objects.filter(id__in=sliver_ids)
return qs