diff --git a/apps/ofm/services.py b/apps/ofm/services.py index 52ee19fd..1e3e41d5 100644 --- a/apps/ofm/services.py +++ b/apps/ofm/services.py @@ -1,7 +1,7 @@ from apps.wf.models import Ticket # TicketFlow, Transition, Workflow, CustomField, State, -from apps.ofm.models import LendingSeal, Vehicle, BorrowRecord, Publicity, MroomBooking, MroomSlot +from apps.ofm.models import LendingSeal, VehicleUse, BorrowRecord, Publicity, MroomBooking, MroomSlot from rest_framework.exceptions import ParseError @@ -59,7 +59,7 @@ def lending_save_ticket_data(ticket: Ticket, new_ticket_data: dict, **kwargs): def bind_vehicle(ticket: Ticket, transition, new_ticket_data: dict): - ins = Vehicle.objects.get(id=new_ticket_data['t_id']) + ins = VehicleUse.objects.get(id=new_ticket_data['t_id']) ticket_data = ticket.ticket_data ticket_data.update({ 't_model': 'Vehicle', @@ -77,8 +77,8 @@ def bind_vehicle(ticket: Ticket, transition, new_ticket_data: dict): def vehicle_save_ticket_data(ticket: Ticket, new_ticket_data: dict, **kwargs): try: - obj = Vehicle.objects.get(id=new_ticket_data['t_id']) - except Vehicle.DoesNotExist: + obj = VehicleUse.objects.get(id=new_ticket_data['t_id']) + except VehicleUse.DoesNotExist: raise ParseError("Publicity t_id 不存在") data_save = {k: v for k, v in new_ticket_data.items() if k not in ['t_model', 't_id']}