o
    h?i?                     @   sZ   d Z ddlmZ ddlmZmZ edgdZeddd Zed	d
e	fddZ
dS )z
Training Platform API

GET /api/v1/training-platforms            - list available platforms
GET /api/v1/training-platforms/{id}/status - platform current status
    )	APIRouter)get_training_platform	PLATFORMStraining_platforms)tags/c                   C   s   dd t D S )z$Return available training platforms.c                 S   s   g | ]}||d qS ))idname ).0platform_idr
   r
   O/lsinfo/ai/hellotax_ai/training_center/backend/app/api/v1/training_platforms.py
<listcomp>   s    z+list_training_platforms.<locals>.<listcomp>)r   r
   r
   r
   r   list_training_platforms   s   r   z/{platform_id}/statusr   c              
   C   sv   zt | }W n ty  } zddlm} |dt|dd}~ww z| }W n ty5   ddd}Y nw | |d	S )
z1Return the current status of a training platform.r   )HTTPExceptioni  )status_codedetailNFzCould not reach platform)	availableerror)r   status)r   
ValueErrorfastapir   strget_platform_status	Exception)r   platformexcr   r   r
   r
   r   r      s   
r   N)__doc__r   r   &app.services.training_platform.factoryr   r   routergetr   r   r   r
   r
   r
   r   <module>   s    
