B
|
Â[gíP ã @ s¸ d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
|
m Z m Z m Z mZmZ ddlmZmZ ddlmZmZmZ ddlZG dd dZejjd d
|
Z e ¡ Z ej d d d Z dd Z edddZ!e"dddZ#da$dd Z%dd Z&G dd dZ'd Z(d Z)d Z*d Z+i Z,d d! Z-d"d# Z.d:d%d&Z/e*d fd'd(Z0d)d* Z1d;d+d,Z2i Z3d<d-d.Z4d d e*fd/d0Z5d e*fd1d2Z6d=d3d4Z7d>d5d6Z8d7d8 Z9e:d9kr´e;e7 dS )?u
|
交æAPI
|
é N)Úasync_log_util)Úhx_logger_trade_debugÚhx_logger_trade_loopÚhx_logger_trade_callbackÚ logger_tradeÚ logger_system)Úhuaxin_trade_data_updateÚmiddle_api_protocol)Ú socket_utilÚ huaxin_utilÚtoolc @ s e Zd ZdZdd ZdS )Ú TradeCallbacku
|
交æåè°
|
c C s d S )N© )ÚselfÚ
|
order_infor r õ8 D:\éåä½å¸\low_suction_hx\trade\huaxin_trade_api.pyÚon_order s zTradeCallback.on_orderN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r r r r s r é )Ú max_workers)Úqueuec C sØ dd }t dt ¡ ¡ | d k rÔx®y| ¡ }|rÀt |¡}|d }|dkrv|d }t td| ¡ t
|
||¡ nJ|dkrÀz>|d }|d }t t| ¡ |d
|
krºt rºt | d ¡¡ W d X W q& Y q&X q&W d S ) Nc S s8 d| kr d S t td| d ¡ t| d | d d S )NÚ
|
request_idu responseï¼request_id-Údata)r Úinfor Ú set_response)Ú data_jsonr r r Ú__set_response# s z.__run_recv_queue_trade.<locals>.__set_responseu0 huaxin_trade_api __run_recv_pipe_trade 线ç¨ID:ÚtypeÚresponser u responseï¼request_id-Útrade_callbackr r ) r r r Ú get_thread_idÚgetÚjsonÚloadsr r Ú__response_thread_poolÚsubmitr" r )r r Úvalr Útype_r Úctyper r r Ú__run_recv_queue_trade! s2
|
r, c C sf t d¡ xVtdD ]J} t d¡ t ¡ }tdddddd |d
|
d t d¡ tddd |d d qW d S )Né< é é é Ú000333éd g ð?é Fg®Gáz®ï?)Ú
|
price_typeÚblockingÚ order_refÚ shadow_priceZ123123)ÚorderRefr5 )ÚtimeÚsleepÚranger Úcreate_order_refÚorderÚ cancel_order)Úir6 r r r Ú __test_orderL s
|
|
|
r@ )Útrade_callback_c s& a |atj fdddd ¡ dS )u
|
:param trade_callback_: 订ååè°
|
:param queue_strategy_r_trade_w_: æ¥æ¶äº¤æç»ææ°æ®éå
|
:return:
|
c s t S )N)r, r )Úqueue_strategy_r_trade_w_r r Ú<lambda>c ó zrun_trade.<locals>.<lambda>T)ÚtargetÚdaemonN)Úqueue_strategy_r_trade_wr" Ú threadingÚThreadÚstart)rB rA r )rB r Ú run_tradeX s rK )Údata_strc C s t | ¡ dS )uD
|
设置交æååºæ°æ®
|
:param data_str:
|
:return:
|
N)rG Ú
|
put_nowait)rL r r r Úadd_trade_callback_dataf s rN c C s dS )NTr r r r r Úis_pipe_channel_normalt s rO c C sT t dd¡} t| }|d dkr<|d d d | kr<dadS td7 atdkrPdad S )
|
Nr i@B Úcoder ÚsidTr0 r2 F)ÚrandomÚrandintÚ__test_trade_channelÚ trade_pipe_channel_error_count)rQ Úresultr r r Útest_trade_channely s rW c @ s e Zd ZdZdZdZdZdZdZdZ i Z
|
i Z i Z e dd Ze d
|
d Ze d d Ze dd Ze dd Ze dd Ze dd Ze dd ZdS )ÚClientSocketManagerÚtradeZ delegate_listZ deal_listZ position_listZmoneyZdealZl2_cmdc C s> || j kr"|| jkr:| j |¡S n|| jkr:| j |¡gS g S )N)ÚCLIENT_TYPE_TRADEÚsocket_client_dictr$ )ÚclsÚ_typer r r Ú list_client s
|
|
|
z ClientSocketManager.list_clientc C sr || j krR|| jkr g | j|< | j| ||f¡ t ¡ | j|< t d| ¡ n ||f| j|< t ¡ | j|< d S )Nz add_client:)rZ r[ ÚappendrH ÚLockÚsocket_client_lock_dictr r )r\ r] ÚridÚskr r r Ú
|
add_client s
|
|
|
z ClientSocketManager.add_clientc C s || j kr| j | ¡ S d S )N)ra Úlocked)r\ rb r r r Úis_client_locked« s
|
z$ClientSocketManager.is_client_lockedc s4 | j krÐ| jkrÎt j| fdddd}t dt| ¡ t dt j ¡ ¡ xÐ|D ]f}|d jkrdy2 j|d jdd r¦t d
|
|d ¡ |S W qd t j
|
k
|
rÈ t d ¡ Y qdX qdW n`| jkr0y: j| }|d jkr j|d jdd r|S W n t j
|
k
|
r. Y nX d S ) Nc s" | d j kr j | d ¡S dS )Nr )Úactive_client_dictr$ )Úx)r\ r r rC ¶ s z4ClientSocketManager.acquire_client.<locals>.<lambda>T)ÚkeyÚreverseu# acquire_client client_listæ°éï¼u/ acquire_client socket_client_lock_dictæ°éï¼r F)r5 zacquire_client success:zacquire_client TimeoutError) rZ r[ Úsortedr r Úlenra ÚkeysÚacquirerH Ú TimeoutErrorÚerror)r\ r] Z client_listÚdr )r\ r Úacquire_client± s0
|
|
|
|
z"ClientSocketManager.acquire_clientc C sX d}|| j kr.d}| j | ¡ r.| j | ¡ |rDt d| ¡ nt d| ¡ d S )NFTzrelease_client success:zrelease_client fail:)ra re Úreleaser r )r\ Z client_idZsucessr r r Úrelease_clientÏ s
|
z"ClientSocketManager.release_clientc C sä || j kr| j |¡ xÈ| jD ]¾}t| j| tkrx¦| j| D ]@}|d |kr@y|d ¡ W n Y nX | j| |¡ P q@W q t| j| tkr | j| d |kr y| j| d ¡ W n Y nX | j |¡ P q W d S )Nr r0 )ra Úpopr[ r ÚlistÚcloseÚremoveÚtuple)r\ rb Útrq r r r Ú
|
del_clientÜ s(
|
z ClientSocketManager.del_clientc C s t ¡ | j|< d S )N)r9 rg )r\ rb r r r Úheartù s zClientSocketManager.heartc C s8 x2| j ¡ D ]$}t ¡ | j | dkr | |¡ q W d S )Nr. )rg rm r9 r{ )r\ Úkr r r Údel_invalid_clientsý s z'ClientSocketManager.del_invalid_clientsN)r r r rZ ÚCLIENT_TYPE_DELEGATE_LISTÚCLIENT_TYPE_DEAL_LISTÚCLIENT_TYPE_POSITION_LISTÚCLIENT_TYPE_MONEYZCLIENT_TYPE_DEALZCLIENT_TYPE_CMD_L2r[ ra rg Ú classmethodr^ rd rf rr rt r{ r| r~ r r r r rX s$
|
rX r0 r3 g @Tc C s* d| dt t ¡ d dt dd¡ S )NÚr_Ú_i' r i )Úroundr9 rR rS )r r r r Ú__get_request_id s r c
|
C sP t ||¡}y t j|dd}t| | W n" tk
|
rJ } zW dd}~X Y nX dS )ub
|
请æ±ä»¿ç交æ
|
:param request_id:
|
:param type:
|
:param data:
|
:return:
|
i$' )ÚportN)r Zload_simulation_tradeÚrequestr Ú Exception)r r r ZfdatarV Úer r r Ú__request_delegate s r Fc
|
C s, |s t | }y|r"t tdd|¡ | ||t ¡ d}t |¡}t ¡ }tjt || |fdd
|
¡ t t ¡ | d }|dkrt td| d | ¡ |r¦t td
|
|¡ W n~ t k
|
râ } z t td |t |¡ |W d d }~X Y nF tk
|
r& } z&t td |t |¡ t |¡ |W d d }~X Y nX |S )uÿ
|
请æ±ï¼å°äº¤æï¼å
å«ä¸å/æ¤åï¼ä¸æ¥è¯¢ï¼å
å«æ¥æä»/è´¦æ·å¯ç¨éé¢/å§æå表/æ交å表ï¼éåå离
|
@param _type:
|
@param data:
|
@param request_id:
|
@param log_enable:
|
@param is_trade:
|
@return:
|
u/ 请æ±åéå¼å§ï¼client_id-{} request_id-{}r )r r r r9 T)rE ÚargsrF iè é
|
u åéèæ¶ï¼request_id-u èæ¶æ¶é´ï¼u" 请æ±åéæåï¼request_id-{}u+ 请æ±åéå¼å¸¸ï¼request_id-{} error-{}Nu, 请æ±åéå¼å¸¸ï¼ request_id-{} error-{})r r r r r9 r
|
Zencryp_client_params_signrH rI r rJ ÚintÚBrokenPipeErrorÚstrr ÚloggingÚ exception) r] r r Ú
|
log_enableÚis_tradeZ root_dataÚ
|
start_timeZuse_timer r r r Ú __request' s2
|
|
|
r c C s |r|t ¡ }zlxft d¡ | tkrBt | ¡}|r>t td| ¡ |S t ¡ | |kr|rdt td| ¡ td| qW W d X d S )Ng{®Gázt?u# 请æ±è¯»åæåï¼ request_id-{}u# 请æ±è¯»åè¶
æ¶ï¼ request_id-{}u 读åå
容è¶
æ¶: request_id=)r9 r: Ú__request_response_dictru r r r r )r r5 Útimeoutr r rV r r r Ú__read_responseN s
|
|
r c C s* | r&t td| d| ¡ |t| < n d S )Nu 请æ±ååº: request_id-u å
容-)r r r r )r r! r r r r e s
|
r c
|
C s¦ t t ¡ d }
|
|s$d| d|
|
}|s0t ¡ }|s>ttj}x<tdD ]0} ttjtjd| |||||| ||d |dd}qHW z|rt ||S d|iS W d t
|
¡ X d S )
|
uU
|
ä¸åå§æ
|
@param direction: 1-ä¹° 2-å
|
@param code:
|
@param volume:交æé
|
@param price:ä»·æ ¼ï¼å¦ææ¯åæ¶ä¸ä¼ ä»·æ ¼å°±æç
§5æ¡ä»·åï¼
|
@param price_type:
|
@param blocking:æ¯å¦é»å¡è¿ç¨
|
@param sinfo:
|
@param request_id:
|
@param order_ref:
|
@param shadow_price:
|
@return:
|
iè Zb_r
r0 ) r Ú
|
trade_typeÚ directionrP r6 Úvolumer4 Úpricer7 Úsinfor5 T)r r r6 N) r r9 r r< r rX rZ r; r r r Úadd_money_list) r rP r r r4 r5 r r r6 r7 Ú timestampr? r r r r= o s.
|
|
r= c
|
C sì t t ¡ d¡dkr,t t ¡ d¡dkr,d S |sZd| dtt ¡ d dt dd¡ }t ¡ }|spt t
|
j }|t krt t |< t | t ||f¡¡ x6tdD ]*} tt
|
j t
|
j d| |||||d |d
|
d }q¤W z
|
t||S t ¡ X d S ) Nz14:57:00r z15:00:01Zcb_r
iè i' r3 )r r r rP r8 ZorderActionRefÚ
|
orderSysIDr T)r r )r Útrade_time_subÚget_now_time_strr r9 rR rS r r< r rX rZ Ú__canceling_order_dictÚsetÚaddr% Údumpsr; r r r r )
|
r rP r¢ r8 r5 r r ZrecancelZorder_action_refr? r r r r> s,
|
*
|
|
|
r> c C s* t tjtj| rdndd}t|||dS )u
|
å§æå表
|
:param can_cancel:
|
:param blocking:
|
:param timeout:
|
:return: ä¾å¦ï¼{'code': 0, 'data': [{'securityID': '002400', 'orderLocalID': 'P900046788', 'direction': '0', 'orderSysID': '12002P900046788', 'insertTime': '13:11:40', 'insertDate': '20250109', 'acceptTime': '13:11:22', 'cancelTime': '', 'limitPrice': 8.23, 'accountID': '00044396', 'turnover': 0.0, 'orderRef': 131161, 'volume': 1000, 'volumeTraded': 0, 'orderStatus': '2', 'orderSubmitStatus': '1', 'statusMsg': ''}]}
|
r0 r )r Ú
|
can_cancel)r )r rX r r )r© r5 r r r r r Úget_delegate_listÆ s rª c C s t tjdtji}t|| |dS )u
|
è·åæ交å表
|
:param blocking:
|
:param timeout:
|
:return: {'code': 0, 'data': [{'tradeID': '17211275', 'securityID': '002184', 'orderLocalID': 'P900046786', 'direction': '0', 'orderSysID': '12002P900046786', 'price': 20.31, 'tradeTime': '13:11:40', 'volume': 1000, 'tradeDate': '20250109', 'tradingDay': '20250109', 'pbuID': '232600', 'accountID': '00044396', 'orderRef': 131146}]}
|
r )r )r rX r r )r5 r r r r r Ú get_deal_listÖ s r« c C s t tjdtji}t|| S )u
|
è·åæä»å表
|
:param blocking:
|
:return: {'code': 0, 'data': [{'investorID': '00044396', 'tradingDay': '20250109', 'securityName': 'æµ·å¾æ§å¶', 'securityID': '002184', 'historyPos': 0, 'historyPosFrozen': 0, 'todayBSPos': 1000, 'todayBSPosFrozen': 0, 'historyPosPrice': 0.0, 'totalPosCost': 20316.176271, 'prePosition': 0, 'availablePosition': 0, 'currentPosition': 1000, 'openPosCost': 20310.0, 'todayCommission': 6.176271, 'todayTotalBuyAmount': 20310.0, 'todayTotalSellAmount': 0.0}, {'investorID': '00044396', 'tradingDay': '20250109', 'securityName': 'ç广éå¢', 'securityID': '002400', 'historyPos': 0, 'historyPosFrozen': 0, 'todayBSPos': 0, 'todayBSPosFrozen': 0, 'historyPosPrice': 0.0, 'totalPosCost': 0.0, 'prePosition': 0, 'availablePosition': 0, 'currentPosition': 0, 'openPosCost': 0.0, 'todayCommission': 0.0, 'todayTotalBuyAmount': 8230.0, 'todayTotalSellAmount': 0.0}]}
|
r )r rX r r )r5 r r r r Úget_position_listã s r¬ c C s t tjdtji}t|| S )u)
|
è·åè´¦æ·èµéç¶åµ
|
:param blocking:
|
:return: 示ä¾ï¼ {'code': 0, 'data': [{'departmentID': '0001', 'investorID': '00032047', 'accountID': '00032047', 'currencyID': '1', 'usefulMoney': 39305420.68, 'frozenCash': 0.0, 'fetchLimit': 39305420.68, 'preDeposit': 39305420.68}]}
|
r )r rX r r )r5 r r r r Ú get_moneyï s r c C s& t ddd| iddd}t|dddS )NÚtestrQ )r r F)r T)r r )rQ r r r r rT û s rT Ú__main__)NTF)r3 FNNNN)NFNNF)T)T)<r r% r Úmultiprocessingr rR rH r9 Ú
|
log_moduler Úlog_module.logr r r r r rY r r Úutilsr
|
r r Úconcurrent.futuresÚ
|
concurrentr ÚfuturesÚThreadPoolExecutorr' ÚQueueZ__save_data_queuer, r@ rK r rN rU rO rW rX ZTRADE_DIRECTION_BUYZTRADE_DIRECTION_SELLÚTIMEOUTZTRADE_DELEGATEDr r r r r r r= r¥ r> rª r« r¬ r rT r Úprintr r r r Ú<module> s\ +
|
|
'
|
|
+
|
'
|
|
|
|
