o
    i                     @   sT   d Z ddlZddlZddlmZ dZdd Zdd Z	dd
dZ
dddZdd ZdS )zU
Created on 2015/08/24
@author: Jimmy Liu
@group : waditu
@contact: jimmysoa@sina.cn
    N)consbkc                 C   s@   t j| gdgd}tjd}tj|tj}|j|dd d S )Ntoken)columns~Findex)	pd	DataFrameospath
expanduserjoinct	TOKEN_F_Pto_csv)r   df	user_homefp r   E/opt/alphahud/venv/lib/python3.10/site-packages/tushare/util/upass.py	set_token   s   r   c                  C   sR   t jd} t j| tj}t j|r"t|}t	|j
d d S ttj d S )Nr   r   r   )r   r   r   r   r   r   existsr	   read_csvstrlocprintTOKEN_ERR_MSG)r   r   r   r   r   r   	get_token   s   

r    c                 C   s   t j| ||ggg dtd}tjtr=t jttd}||j| k d 	 r-||j| k }|j
|dd}|jtdd d S |jtdd d S )	N)brokeruserpasswd)r   dtyper#   r!   T)ignore_indexFr   )r	   r
   objectr   r   r   BKr   r    anyappendr   )r    r!   r"   r   allr   r   r   
set_broker"   s   r+   c                 C   s8   t jtrtjttd}| dkr|S ||j| k S d S )Nr$   r   )r   r   r   r'   r	   r   r&   r    )r    r   r   r   r   
get_broker0   s   r,   c                   C   s   t t d S )N)r   remover'   r   r   r   r   remove_broker;   s   r.   )r   r   r   )r   )__doc__pandasr	   r   tushare.stockr   r   r'   r   r   r+   r,   r.   r   r   r   r   <module>   s   

