o
    i	                     @   sZ   d dl Zd dlZd dlZd dlmZ G dd deZ	e
dkr+e	 Ze Zee dS dS )    N)Poolc                   @   sP   e Zd Zdd Zdd Zdd Zdd Zdd
dZdd Zdd Z	dddZ
dS )TsDatac                 C   s$   d| _ |  | _d | _t | _d S )Nz*http://file.tushare.org/tsdata/symbols.csv)url	_get_datastocksconspd	DataFramedataself r   I/opt/alphahud/venv/lib/python3.10/site-packages/tushare/stock/minsdata.py__init__
   s   
zTsData.__init__c                 C   s"   t j| jdtid}t|d jS )Nsymbol)dtype)r   read_csvr   objectlistvalues)r   dfr   r   r   r      s   zTsData._get_datac                 C   s6   t  }t  }t  }t  }t  }|||||gS N)tsget_apis)r   con1con2con3con4con5r   r   r   	_get_cons   s   zTsData._get_consc                 C   s   t | j | jd S )Nr   )randomshuffler   r   r   r   r   _get_server    s   
zTsData._get_server c                 C   sV   z$|   }tj||dd}|jdd}|g d }| jj|dd| _|W S    Y d S )N5min)connfreqT)inplacedatetimeopenclosehighlowvolumecodeignore_index)r"   r   barreset_indexr
   append)r   r   
start_dateend_dater%   r   r   r   r   _data&   s   zTsData._datac                 C   s8   zt j|dd}|g d }g d|_|W S    Y d S )N5)ktype)dater*   r+   r,   r-   r.   r/   r(   )r   
get_k_datacolumns)r   r   r   r   r   r   _get_k_data2   s   
zTsData._get_k_datac                 C   sN   t | | |}|d u r| jd u r|  | _| |}| jj|dd| _d S )NTr0   )printr=   r   r   r7   r
   r4   )r   symbolsr   r   r   r   
get_m_data=   s   



zTsData.get_m_dataNc                 C   sL   |d u r|   }td}|| j| | jd ur#| jD ]}t| q| jS )NiX  )r   
ThreadPoolmapr@   r   r   
close_apisr
   )r   r?   r   poolconr   r   r   get_dataH   s   

zTsData.get_data)r#   r#   r   )__name__
__module____qualname__r   r   r   r"   r7   r=   r@   rF   r   r   r   r   r   	   s    	
r   __main__)tusharer   pandasr   r    multiprocessing.dummyr   rA   r   r   rG   apirF   r   r>   r   r   r   r   <module>   s   J