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()