o
    wh                     @   s   d dl mZ d dlmZ ddlmZmZmZmZm	Z	 edgdd Z
edgd	d
 Zedgdd Zedgdd Zedgdd ZdS )    )Response)api_view   )select_random_squad
pick_squadsuggest_transfersload_squad_using_manager_idai_suggest_transferGETc                 C   s"   t  }|r	t|S tddiddS )Nerrorz;Could not generate a valid team with the given constraints.  status)r   r   )requestteam r   0/var/www/html/fantasy/fantasy_api/squad/views.pysuggest_squad   s   r   POSTc                 C   s   | j }t|}t|S )N)datar   r   )r   r   Z	formationr   r   r   select_squad
   s   r   c                 C   sL   | j }|dg }|dd}|dg }|dd}t||||}t|S )Nchangesbudgetr   r   free)r   getr   r   )r   r   Z
players_idr   r   free_transfers	transfersr   r   r   suggest_transfer   s   r   c                 C   s>   | j }|dd}|dg }|dd}t|||}t|S )Nr   r   r   r   )r   r   r	   r   )r   r   r   r   r   r   r   r   r   ai_transfer   s   r   c              
   C   sD   zt t|W S  ty! } zt ddiddW  Y d}~S d}~ww )z'
    Load a squad from a given ID.
    detailz
Not found.r   r   N)r   r   	Exception)r   ider   r   r   load_from_id!   s   r#   N)rest_framework.responser   rest_framework.decoratorsr   servicesr   r   r   r   r	   r   r   r   r   r#   r   r   r   r   <module>   s    



