tushare_easy.utils module

tushare_easy utils

tushare_easy.utils.down2save(code, ktype='d', start=None, end=None, path='.')

tushare_easy.utils.down2save_update(code, ktype='d', start=None, end=None, path='.')

get k_chart date and save them locally, and update them.

get k_chart data using tushare.get_k_data save data locally with tab separated files could append data line by line if getting newer

Parameters:

code : str

ktype : format ‘%Y-%m-%d-%H-%M’

start, end : str or None

datetime.datetime format ‘%Y-%m-%d-%H-%M’ e.g. ‘2017-12-12-23-59’

path : str

where to search files locally

tushare_easy.utils.extract_start_end(df)

format start and end time

Parameters:

df : pd.DataFrame

data with setting time as index

time format:

‘%Y-%m-%d %H:%M’ or ‘%Y-%m-%d’

Returns:

start, end : str

datetime string, format: ‘%Y-%m-%d %H:%M’

tushare_easy.utils.fmt_filename(name_list)

format filename

Parameters:

name_list : list of str

four elements list: [code, start, end, ktype]

Returns:

str

filename

tushare_easy.utils.fmt_filename_demo(name_list)

tushare_easy.utils.get_arrow(strftime)

get arrow.arrow.Arrow from str with local timezone info

Parameters:

strftime : str

format ‘%Y-%m-%d-%H-%M’

Returns:

arrow.arrow.Arrow

tushare_easy.utils.get_codes(index=False)

tushare_easy.utils.get_data(code, ktype='d', start=None, end=None)

tushare_easy.utils.get_demo(df)

tushare_easy.utils.get_demo_new(df_last, df_newer)

tushare_easy.utils.get_end_date(ktype='d')

get up-to-date time of market

Parameters:ktype : {‘d’, ‘5’, ‘15’, ‘30’, ‘60’, ‘w’, ‘m’} Returns:end_datetime : arrow.arrow.Arrow

tushare_easy.utils.get_local(code, ktype='d', path='.')

get last file and demo

Parameters:

code : str

ktype : {‘d’, ‘5’, ‘15’, ‘30’, ‘60’, ‘w’, ‘m’}

path : str

where to search file

Returns:

filename_demo : filename of demo

tushare_easy.utils.get_threshhold_datetime(time_str, ktype='d')

get threshhold of recent time locally

Parameters:

time_str : str

format ‘%Y-%m-%d-%H-%M’

ktype : {‘d’, ‘5’, ‘15’, ‘30’, ‘60’, ‘w’, ‘m’}

Returns:

threshhold_datetime : arrow.arrow.Arrow

tushare_easy.utils.is_up_to_date(end_local_str, ktype='d')

check whether local data is up-to-date

Parameters:

end_local_str : str

format ‘%Y-%m-%d-%H-%M’

ktype : {‘d’, ‘5’, ‘15’, ‘30’, ‘60’, ‘w’, ‘m’}

Returns:

boolean

tushare_easy.utils.prep(df)

tushare_easy.utils.read_data(filepath, header=0)

read data from file

always with index, with header default

Parameters:

filepath : str

header : int or None

Returns:

pandas.DataFrame

tushare_easy.utils.save(df, filepath, mode='w', header=False)

tushare_easy.utils.set_date_index(df, labels=['date', 'month'])

set date as df’s index

Parameters:

df : pandas.DataFrame

data getting from tushare.get_k_data

labels : list-like

labels using to set as df index, sequenced preferred

Returns:

pandas.DataFrame

data with index set by a label in labels

tushare_easy.utils.split_filename(filename)