o
    i                     @   sP   d Z ddlZdddZdd Zdd	 Zd
d Zedkr&eddZee dS dS )ul   
Created on 2021-12-13 16:29:25
---------
@summary: 股票代码 code 格式化
---------
@author: yangyx01
    N c                 C   s  t | } dtt j| } d}d}|rt | } d|v sd|v r0t| |k r*| |n| } d|  S |dkr[tdt | tjtj	B }|sE| S |
d	} t| |k rU| |n| } d
|  S |dksc|dkrtt| |k rn| |n| } d|  S |dkrt | drt| |k r| |n| } d|  S t | drt| |k r| |n| } d|  S | d  s|  S t | dd dv st | d dv rt| |k r| |n| } d|  S t | d dv st | dd dv rt| |k r| |n| } d|  S t| |k r| |n| } d|  S )uT   
    股票代码 code 格式化
    @param x:
    @param special:
    @return:
    r         u   行业u   概念z%s.TIu   港股z(\d+)   z%s.HKu   场外基金u   热基z%s.OFu	   可转债11z%s.SH12z%s.SZr      )920)84z%s.BJ)56)900)strjoinfilterisdigitlenzfillresearchSMgroup
startswithupper)xspecial	stock_lenhk_stock_lenr r!   Q/opt/alphahud/venv/lib/python3.10/site-packages/tushare/util/format_stock_code.pyformat_stock_code   sH   
$$r#   c                 C   s6   t |  d}| d  sdS t|dkrdS dS )N.T   F)r   r   splitr   r   )r   lr!   r!   r"   a_symbol_verify;   s   r)   c                 C   s:   |  d}t|dkrt|d r| S dt| r| S d)N,r   r   u<   请按照 "000001.SZ,000001.SZ,000001.SZ" 格式传入symbolu(   请按照 "000001.SZ" 格式传入symbol)r'   r   r)   )r   ssr!   r!   r"   symbol_verifyE   s   
r,   c                 C   s   t |  } | dr| drdS | dr| drdS | dr*| dr*dS | dr6| dr6dS | dr=dS | drI| d	rIdS | d
rU| d	rUdS | dra| d	radS | drm| drmdS dS )uT   
    判断代码是否是 股票 True  指数 False
    @param x:
    @return:
    39SZT300SHFr
   BJr   r	   9CSI)r   r   r   endswithr   r!   r!   r"   verify_stock_or_indexS   s(   
r7   __main__sz92052r6   )r   )	__doc__r   r#   r)   r,   r7   __name__sprintr!   r!   r!   r"   <module>   s   
/

