o
    Ah                  
   @   s   d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ e Zejde	deejfde	d	ejfd
dZedddeejfdedee dee d	ejfddZedeejfded	ejfddZdS )    )	APIRouterHTTPExceptionDepends)database)Optional)VehicleAssignment)workforcevehicle/)response_modelpayloaddbc              
   C   s8   zt | |W S  ty } ztdt|dd }~ww )Ni  )status_codedetail)service!create_vehicle_assignment_service
ValueErrorr   str)r   r   e r   D/var/www/html/moveengine/app/v1/routers/platform/workforcevehicle.pycreate_vehicle_assignment	   s   r   z/{account_id}N
account_idworkforce_id
vehicle_idc                 C   s   t | |||S N)r   get_vehicle_assignments_service)r   r   r   r   r   r   r   get_vehicle_assignments   s   r   z/unassign/{workforce_id}c                 C   s   t | |S r   )r   unassign_vehicle_service)r   r   r   r   r   unassign_vehicle   s   r   )fastapir   r   r   app.dbr   typingr   'app.v1.models.platform.workforcevehicler   app.v1.services.platformr   r   routerpostget_mongo_dbMongoDBr   getr   r   r   r   r   r   r   <module>   sB    	