o hp@s<ddlZddlmZmZmZddlZddlZddlm Z ddl m Z dZ e Z ddZdd Zdd d Zd eeefdeeeefdeedeeefdef ddZdeeeefdeeeefdeedeeefdef ddZ ddeeeefdeeeefdeedeeefdededefddZdS)N)AnyDictList)deepcopy)load_teams_datacCs$t|d}|d|d|ddS)z4 Returns the team name for a given team ID. id short_namecode)r namer )TEAMS)team_idteamr0/var/www/html/fantasy/fantasy_api/squad/tr_v3.py get_team_names rcCs<t|g}|gd}|ddd|d<|jdS)N) r web_name element_typernow_cost total_pointsstatus opta_codescore DifficultyXPrcSst|S)N)rxrrr(sz!format_transfer..r)pd DataFrameapplyilocto_dict)playerdfrrrformat_transfers r&TcCsX|d|vs||ddtkrdS|d|krdS|d|kr"dS|ddkr*dSd S) z Validates a transfer selection: - Checks if the player Xp is not less than the Old. - Checks if the player cost is within the budget. r rrFrr availabilityrT)getMAX_PLAYERS_PER_TEAM)r$old_XPbudgetidscount is_availblerrrvalidate_trasfer_selection-s    r/ player_out all_playersr team_countr+c si}t|}|}d}d} | dkr%|d| t|dd}||fdd|D} t| dd d d } | D]B} | d |vrGq>t| d ||ddvr| d d } | dkr|t| |d<t| d|d<t| dd|d<|Sq>iS)z For a single player to transfer out, return the next batch of best transfer suggestions. Suggestions are sorted by XP, then price. Returns up to batch_size suggestions starting from 'start'. rcost_change_startr g?c<g|]}|ddkr|ddkr|dkr|qSrrrr.0pbankr0rr ^s<z7get_transfer_suggestions_for_player..cS|d|dfSNrrrrrrrr_z5get_transfer_suggestions_for_player..Tkeyreverser rradrgainrr;)setcopymathfloorsortedr/r&round) r0r1rr2r+ suggestionsuggested_players_idsteam_player_countplayer_selling_priceplayer_price_change candidates candidaterErr:r#get_transfer_suggestions_for_playerIs:   rSplayerscCsNg}|D]}t|||||}|r||q|jdddd|r%|dSgS)z get best ai suggestion for each player in one on one fashion if a transfer is possible then return the best suggestion based on total gain and remaining budget in case of 2 suggestions having the same gain cSs|dd|ddfS)NrErr;)r(rrrrrsz-get_ai_transfer_suggestions..Tr@r)rSappendsort)rTr1rr2r+Zsuggestion_resultr$rLrrrget_ai_transfer_suggestionsss  rW players_outstart batch_sizec sg}|D]fdd|D}}t|dddd}||qttj|} g} | D]} d} i} t|}t|}}d}t| D]R\}}|||d|vrTd } nAt|d |||d d vsgd } n.t || d <| |d| |ddd||d<||d8}||d d 7}qB| rt | t |kr| d| dt |ddt |dddd| Diq,| jddddtdt |dt | tdt | | ddS)z For multiple players to transfer out, return the next batch of best transfer combinations. Each combination is a mapping of player_out to best valid replacement. Results are sorted by total XP gain, then remaining budget. cr5r6rr7r+r0rrr<s ,z8get_transfer_suggestions_for_players..cSr=r>rrrrrrr?z6get_transfer_suggestions_for_players..Tr@rr FrrrCrrrr transfers rem_balance total_gainrDcombocSsg|]}|dqS)r rr7rrrr<scSr=)Nr^r]rrrrrrr?z Total Available Combos for z players : z Total Valid Suggestions : N)rJrUlist itertoolsproductrFr enumerater/r&addr(lenrKrVprint)rXr1rr2r+rYrZZcandidates_per_outrQZ all_combosresultsr_valid selectionZused_idsrNr;r^idxrRrr[r$get_transfer_suggestions_for_playerss`         rl)T)rr)rHtypingrrrpandasrrbrGrfantasy_data.servicesrr)r rr&r/strintrSrWfloatrlrrrrsd      *