?j <ddlmZddlZddlmZmZGddZy)) annotationsN) defaultdictdequec*eZdZdZdddZddZd dZy) PaperTraderzHLong-only moving-average paper trader with a fixed virtual cash balance.cd|cxkrdkrn td|k\r tdt||_t||_||_|_||_tfd|_i|_ g|_ y)NrzFallocation must be in (0, 1] and fast_window must be below slow_windowctS)N)maxlen)r) slow_windowspaper_trader.pyz&PaperTrader.__init__..s EQ\D]) ValueErrorfloat initial_cashcash fast_windowr allocationrhistory positionstrades)selfrrr rs ` r __init__zPaperTrader.__init__ s{:""ef f'2[&@ef f!$K$K &&$0;<]0^ 68"$ rcp|j|}|jt|t||jkryt t ||j d|jz }t ||jz }|jj|}||kDr||j|jz}||z }|xj|zc_ |t|d|j|<|jjttj|dt||dy||krr|p|d}|xj||zz c_ |j|=|jjttj|dt||dyy)N)quantity entry_pricebuy)timepairsidepricerrsell)rappendrlenr sumlistrrgetrrrintr) rr r"pricesfastslowpositionspendrs r stepzPaperTrader.stepsd# eEl# v;)) )4<!1!1 1 234t7G7GG6{T--->>%%d+ $;8+II/Eu}H II I08u#VDNN4 KK  DIIK(8$PU`efk`l{C D E $;8/ +H IIE) )It$ KK  DIIK(8$PVafglam|D E Frc ntfd|jjD}|j|z}|j|j||||jz dz dz|jj |j ddd|j|j|jddS)Nc3ZK|]"\}}|dj||dz$yw)rrN)r().0r posr*s r z'PaperTrader.snapshot...s1zV_VZ\_3z?VZZc->P-QQzs(+r dizMoving-average crossover)namerr r)rr market_valueequity return_pctrrstrategy) r&ritemsrrcopyrrr r)rr*r7r8s ` r snapshotzPaperTrader.snapshot-szcgcqcqcwcwcyzz \) --II(!D$5$559S@,,.kk#$'!;DL\L\mqm}m}NRN]N]^  rN)i'g?)rrrr)r r)rr)r strr"rreturnz str | None)r*zdict[str, float]rAdict)__name__ __module__ __qualname____doc__rr/r=rr rrsR %.  rr) __future__rr collectionsrrrrGrr rJs" *2 2 r