def post(self, request, *args, **kwargs):
employee = get_object_or_404(Employee, pk=self.kwargs['employee'])
equipment = get_object_or_404(Equipment, pk=self.kwargs['equipment'])
quantity = equipment.quantity_at_loaner(employee)
equipment_event = EquipmentEvent(event_type='Aflevering',
quantity=quantity,
employee=employee,
equipment=equipment)
equipment_event.save()
equipment.quantity_on_loan -= quantity
equipment.quantity_in_stock += quantity
equipment.save()
messages.success(self.request, 'Udstyr returneret')
return HttpResponseRedirect(self.get_success_url())
def loan(self, employee, building_site, quantity):
event = EquipmentEvent(event_type='Udlån', equipment=self,
quantity=quantity, employee=employee,
building_site=building_site)
event.save()
