B
|
Jüýbak ã @ sL d dl Z d dlZd dlZd dlmZ d dlZd dlZd dlZd dlZd dl Z d dl
|
Z
|
d dl Z d dl m Z e d¡Zi Zi ZG dd deZG dd dZd4d
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Z dd Z d Z d d Z d d Z d!d" Z!d#d$ Z"d%d& Z#d'd( Z$d)d* Z%d+d, Z&d-d. Z'd/d0 Z(d1d2 Z)e*d3krHdS )5é N)Údatetime)Úlogger_l2_tradeé c s4 e Zd ZdZdZ fddZdd Zdd Z ZS ) ÚL2DataExceptionr é c s t | ¡ || _|| _d S )N)ÚsuperÚ__init__ÚcodeÚmsg)Úselfr r
|
)Ú __class__© ú%D:\workspace\trade\l2_data_manager.pyr s zL2DataException.__init__c C s | j S )N)r
|
)r r r r Ú__str__# s zL2DataException.__str__c C s | j S )N)r )r r r r Úget_code& s zL2DataException.get_code) Ú__name__Ú
|
__module__Ú __qualname__ÚCODE_PRICE_ERRORZCODE_NO_CLOSE_PRICEr r r Ú __classcell__r r )r r r s
|
r c @ sd e Zd Zedd Zedd Zedd Zedd Zedd
|
d Zed d Z edddZ
|
d S )ÚTradePointManagerc C s t ¡ S )N)Ú _redisManagerÚgetRedisr r r r Z __get_redis, s z TradePointManager.__get_redisc C s, t ¡ }| d | ¡¡ | d | ¡¡ d S )Nzbuy_compute_index-{}zbuy_compute_num-{})r Ú _TradePointManager__get_redisÚdeleteÚformat)r Úredisr r r Údelete_buy_point1 s z"TradePointManager.delete_buy_pointc C s, t ¡ }| d | ¡¡ | d | ¡¡ d S )Nzbuy_cancel_compute_index-{}zbuy_cancel_compute_num-{})r r r r )r r r r r Údelete_buy_cancel_point8 s z)TradePointManager.delete_buy_cancel_pointc C sH t ¡ }| d | ¡¡}| d | ¡¡}|d kr4dS t|t|fS d S )Nzbuy_compute_index-{}zbuy_compute_num-{})Nr )r r Úgetr Úint)r r ÚindexÚ total_numr r r Úget_buy_compute_start_data? s z,TradePointManager.get_buy_compute_start_dataNc C sd t ¡ }t ¡ }|d k r,| d | ¡||¡ d | ¡}| |¡d krT| |||¡ n | ||¡ d S )Nzbuy_compute_index-{}zbuy_compute_num-{})r r ÚtoolÚ
|
get_expireÚsetexr r Úincrby)r Únum_addr! r ÚexpireÚkeyr r r Úset_buy_compute_start_dataJ s
|
z,TradePointManager.set_buy_compute_start_datac C sH t ¡ }| d | ¡¡}| d | ¡¡}|d kr4dS t|t|fS d S )Nzbuy_cancel_compute_index-{}zbuy_cancel_compute_num-{})Nr )r r r r r )r r r! r" r r r Ú!get_buy_cancel_compute_start_dataW s z3TradePointManager.get_buy_cancel_compute_start_datac C sd t ¡ }t ¡ }|d k r,| d | ¡||¡ d | ¡}| |¡d krT| |||¡ n | ||¡ d S )Nzbuy_cancel_compute_index-{}zbuy_cancel_compute_num-{})r r r$ r% r& r r r' )r r( r! r r) r* r r r Ú!set_buy_cancel_compute_start_datab s
|
z3TradePointManager.set_buy_cancel_compute_start_data)N)N) r r r Ú staticmethodr r r r# r+ r, r- r r r r r + s r Fc C sT t ¡ }t | ¡d ks|r\| d | ¡¡}|d k r\| tkrJt |¡t| < nt | t |¡¡ t | ¡d ksp|rPg }| d | ¡¡}x®|D ]¦}|
|
dd¡}| d¡}|d } |d } |d }
|
|d } |d
|
} |d } |d }|d }| |
|
| | | ||d}| |¡}t |¡}|||d t |d d}| |¡ qW t|dd d}t | |¡ d S )Nzl2-data-latest-{}zl2-{}-*zl2-Ú ú-r r r é é é é é )Ú operateTypeÚtimeÚnumÚpriceÚ
|
limitPriceÚ
|
cancelTimeÚcancelTimeUnitÚrer! )r* Úvalr= r! c S s$ t | d¡d dd¡| d¡fS )Nr> r7 ú:r/ r! )r Ú __getitem__Úreplace)Úer r r Ú<lambda> ó z load_l2_data.<locals>.<lambda>)r* )r r Úlocal_latest_datasr r ÚjsonÚloadsÚ
|
setdefaultÚlocal_today_datasÚkeysrA Úsplitr ÚappendÚsorted)r Úforcer Ú_dataÚdatasrJ Úkr* Z
|
split_datar6 r7 r8 r9 r: r; r< ÚitemÚvalueÚ
|
json_valueÚ new_datasr r r Ú load_l2_datao s<
|
|
|
|
|
|
rV c C s4 t | ¡sd S t ¡ }z| d | ¡d¡dkrt ¡ }d}| d | ¡¡}|d kr\d}nt |}|}x|D ]}|d }d|d }| |¡} | d krÈ|| }|| |d d } |
|
||t | ¡¡ qnt | ¡}
|
|
|
d |d krn|d |
|
d< |
|
||t |
|
¡¡ qnW |
|
d | ¡||¡ W d | d | ¡¡ X |S )
|
Nz
|
l2-save-{}Ú1r zl2-maxindex-{}r zl2-r* r= )r! r= )Úgpcode_managerÚ is_in_gp_poolr r Úsetnxr r$ r% r r r& rF ÚdumpsrG r ) r rP Úredis_instancer) r! Ú start_indexÚ max_indexrO r* rS rT r r r Ú
|
saveL2Data s6
|
|
|
|
r_ c C sN t t ¡ }t ¡ d¡}t | ¡}|d }|d }|d }|d }|d }g }i } t |¡}
|
xà|D ]Ø} | d } t
|
| d } | d }| d }|
|
d k rº|
|
t t | ¡¡krºd
|
}d |¡| d < | d }| d }| d }d ||| || |||¡}|| kr|| | d d
|
|| | d< qd| || d
|
d¡ | |t|d
|
¡ qdW |||||fS )Nz%Y%m%dÚdataÚclientr Úchannelr7 r9 r8 r: r z{}r6 r; r< z{}-{}-{}-{}-{}-{}-{}-{}r= )r* r> r= )r Útr7 r ÚnowÚstrftimerF rG rX Úget_limit_up_priceÚfloatr$ Úto_priceÚdecimalÚDecimalr rL rH Úlen)Ústrrd ÚdayÚdictr` ra r rb rP Z
|
dataIndexsÚlimit_up_pricerR r7 r9 r8 r: r6 r; r< r* r r r Ú parseL2DataÀ s:
|
|
|
|
"rp c C sx t | ¡}|d krg }x\|D ]T}xN|D ]F}|d |d kr&|d |d kr&t|d |d }||d< ||d< q&W q W |S )Nr* r= )rE r Úmax)r Z_datasZ latest_datar` Z_ldataZmax_rer r r Ú correct_dataé s
|
|
|
rr c C sl t ¡ }| d | ¡t ¡ t |¡¡ | tkr8|t| < n t | |¡ t
|
| t | t |dkrht | | d S )Nzl2-data-latest-{}r ) r r r& r r$ r% rF r[ rE rH Ú__set_l2_data_latest_countrk r_ )r rP Ú add_datasr r r r Ú save_l2_dataø s
|
ru c C sÐ |d k rt |dk rg S d}t | ¡}|d k rFt |dkrF|d d }d}d}x2t|D ]&}|d7 }|d |krXt || }P qXW t |dkr¼|dk s¦|d t |krªg S ||d d
S n||d d
S d S )Nr r/ r éÿÿÿÿr* )rk rE r Úreversed)r rP Zlast_keyZ__latest_datasÚcountr] Únr r r Ú get_add_data s$
|
rz c C s | d¡}t|d d t|d d t|d }| d¡}t|d d t|d d t|d }t|| dk rdS d S d S )
|
Nr? r i r é< r r1 TF)rK r Úabs)Ztime1Ztime2Ztime1_sZ time1_secondZtime2_sZ time2_secondr r r Ú__is_same_time s
|
,
|
,r} c
|
C s¼ t ¡ d¡}tt ¡ d }zt|dkr²t | t |d d d ¡sXt
|
t
|
j dt | t | |}t| |}t|dkr¦t| |¡ |t|d d d }t||r¦t | ¡\}}|d krvt| t|d d }|rît | ¡ t| t|d
|
d
|
}|d k r¨t| } t d | t | | ¡¡¡ d}
|
x®t |t| D ]} | | d } t | d dkrt | d dkr|
|
t | d t | | d 7 }
|
nDt | d dkrDt | d dkrD|
|
t | d t | | d 8 }
|
qDW t | |
|
|¡ t !| ¡} | d k r¨|
|
| d dkr¨t d | t | d ¡¡¡ yt "| ¡ t #| ¡ W n$ t$k
|
rp } zW d d }~X Y nX n2d}
|
x|D ]}|d } t | d dkrÎt | d dkrÎ|
|
t | d t |d 7 }
|
n@t | d dkrt | d dkr|
|
t | d t |d 8 }
|
qW t | |
|
¡ ||
|
}t !| ¡} | d k r¨|| d dkr¨t d | t |d ¡¡¡ yt "| ¡ t #| ¡ W n$ t$k
|
r¦ } zW d d }~X Y nX |d k r¦t %| ¡}|tj&ksÔ|tj'kr¦t (| ¡\}}|d kr¬t)| t|d
|
d
|
}|d k r¦t| } t d | t | | ¡¡¡ d}xht |t| D ]V} | | d } t | d dkrBt | d dkrB|t | d t | | d 7 }qBW t *| ||¡ núd}xV|D ]N}|d } t | d dkr¶t | d dkr¶|t | d t |d 7 }q¶W t *| |¡ || }t !| ¡} | d k r¦|| d dkr¦t d | t |d ¡¡¡ y"t +| ¡ t ,| ¡ t #| ¡ W n$ t$k
|
r¤ } zW d d }~X Y nX t-| || W d X d S )Nz%H:%M:%Siè r r> r9 u è¡ä»·ä¸å¹é
r r7 r4 r1 u æ¾å°ä¹°ç¹ï¼{} - {}r: r6 r8 r= éd i u) æ§è¡ä¹°å
¥ï¼{} - 计ç®ç»æç¹ï¼ {}rv u æ¾å°ä¹°æ¤ç¹ï¼{} - {}u æ§è¡æ¤éï¼{} - {}).r rd re Úroundrc r7 rk Ú data_processZis_same_code_with_pricerg r r rV rr rz rI Úextendr} r r# Ú__is_have_forbidden_featureÚ trade_managerZforbidden_tradeÚ__get_limit_up_buy_startr Úinfor rF r[ Úranger r+ rX rf Z start_buyr Ú ExceptionZget_trade_stateZTRADE_STATE_BUY_DELEGATEDZTRADE_STATE_BUY_PLACE_ORDERr, Ú __get_limit_up_buy_cancel_startr- Zstart_cancel_buyr ru )r rP Z now_time_strZ __start_timert Z latest_timeZc_indexZc_numÚ forbiddenZ total_datasZbuy_numsÚiÚ_valro rB r` Z
|
latest_numÚstateZ cancel_indexZ
|
cancel_numÚnumsZcancel_nums_addr r r Ú process_data. s¼
|
|
|
|
|
$"$&
|
|
|
|
$ $"
|
|
|
|
|
|
|
$&
|
$"
|
|
|
|
r c C s6 | d¡}t|d d t|d d t|d S )Nr? r i r r{ r )rK r )Ztime_strÚtsr r r Ú__get_time_second¼ s
|
r i
c C sX t | d dkrdS t | d dkr(dS t| d }t | d }|| d d k rTdS d
|
S ) Nr: r Fr6 r r9 r8 r~ i ¡ T)r rg )r> r9 r8 r r r Ú__is_limit_up_price_buyÅ s r c C sX t | d dkrdS t | d dkr(dS t| d }t | d }|| d dk rTdS d S )
|
Nr: r Fr6 r9 r8 r~ i ¡ T)r rg )r> r9 r8 r r r Ú __is_limit_up_price_buy_cancelÓ s r c C sâ t | }t|}||k r d S d}||kr,|}||kr<|| }d }d}d}xt|||d D ]z} || d }
|
t|
|
d tk rq^t|
|
r4| } d} d} x,t| d |D ]}t|| d r¨|} P q¨W | dkrüx,t| d |D ]}t|| d rÞ|} P qÞW | | dkr4| | dkr4t d | | || ¡¡ | S t|
|
r|d krb|| d d }d}| }n0||
|
d krz|d7 }n|| d d }d}| }n|
|
d dkr®d }d}d}|dkr^|dkr^t d | ||| ¡¡ |S q^W d S )
|
Nr rv r r> r7 u$ æ¾å°ç©çè¿ç»æ¶¨åä¹° {},{},{}r6 r2 u' æ¾å°åä¸ç§è¿ç»æ¶¨åä¹° {},{},{}) rI rk r r Ú
|
second_930r r r
r )r Ú
|
data_countÚ__continue_countrP Ú__lenr] Ú__timeZ_limit_up_count_1sZ _limit_up_count_1s_start_indexr r Úindex_0Úindex_1Úindex_2ÚjÚ_timer r r r â sd
|
|
|
|
r c C s, t | }t|}||k r d S d}||kr,|}||kr<|| }xêt|||d D ]Ô}|| d }t|d tk rrqPt|rP|}d} d}
|
x,t|d |D ]} t|| d r| } P qW | dkrêx,t| d |D ]} t|| d rÌ| }
|
P qÌW | | dkrP|
|
| dkrPt d | |t
|
|| ¡¡¡ |S qPW d S )Nr r r> r7 rv u è¿ç»3个涨åä¹°æ¤ {},{},{}) rI rk r r r r r r
r rF r[ ) r r r rP r r] r r r r r r r r r r . s< r c
|
C sî t | }t|}||k r d S d}||kr,|}||kr<|| }d}d}x¤t||D ]}|| d } | d dkr¨|d dkr¨||d d d dkr¨| d dkr¢|}qædS qP|dkrP| d dkrâ|| dkrâ|}|d7 }||kræd S qPdS qPW dS )
|
Nr rv r> r7 z09:30:00r r6 FT)rI rk r )
|
r r r rP r r] Z cancel_startZ cancel_countr r r r r r W s2 0 r c C s$ t ¡ }d | ¡}| |d|¡ d S )Nzlatest-l2-count-{}r )r r r r& )r rx r r* r r r rs z s
|
rs c C sL | d kst | dk rdS t ¡ }d | ¡}| |¡}|d kr@dS t|S d S )Nr r zlatest-l2-count-{})rk r r r r r )r r r* Úresultr r r Úget_l2_data_latest_count s
|
|
r c C sH d} t ¡ }| | ¡}|dkr(| | ¡ | | d¡ | | t ¡ ¡ d S )Nzl2-fixed-codesr Ú000000)r r Úscardr Úsaddr) r$ r% )r* r rx r r r Úinit_l2_fixed_codes s
|
|
r¢ c C s d}t ¡ }| || ¡ d S )Nzl2-fixed-codes)r r Úsrem)r r* r r r r Úremove_from_l2_fixed_codes s r¤ c C s d}t ¡ }| || ¡ d S )Nzl2-fixed-codes)r r r¡ )r r* r r r r Úadd_to_l2_fixed_codes¢ s r¥ c C s d}t ¡ }| || ¡S )Nzl2-fixed-codes)r r Ú sismember)r r* r r r r Úis_in_l2_fixed_codes© s r§ Ú__main__)F)+ri rF r7 rc r r ZmysqlrX Ú
|
mongo_dataÚ redis_managerr$ r Úlogr Ú RedisManagerr rE rI r r r rV r_ rp rr ru rz r} r r r r r r r r rs r r¢ r¤ r¥ r§ r r r r r Ú<module> sP
|
D
|
'*) L)#
|
|
