src.util package

Submodules

src.util.holiday module

util.holiday.dt_generator(start, end)

util.holiday.fetch(dt)

util.holiday.get_holiday(start_date: datetime.datetime, end_date: datetime.datetime) → set

src.util.transformer module

util.transformer.vwap_calc(df: pandas.core.frame.DataFrame, period: List[int]) → float

计算给定时段的成交量加权平均价

Args:

df: 格式: open high low close volume period: 时段

Returns: vwap

src.util.util module

util.utils.get_all_stock(root_path: str, date: Optional[datetime.datetime] = None) → list

获取数据路径下的所有股票代码

Args:

root_path: 数据路径 date: 日期. 可选,默认全部

Returns: 去重后的所有股票代码

Examples:

# Get all symbol code form the source path all_stock = get_all_stock(“/source/path”)

# Or limit date all_stock = get_all_stock(“/source/path”, datetime.today())

util.utils.get_trading_day(start: Union[datetime.datetime, str], end: Optional[Union[str, datetime.datetime]] = None) → list

获取指定时间范围内的所有交易日,排除节假日和周末

Args:

start: 开始时间

str 或 datetime 类型 需要注意的是 str 必须要是 YYYYMMDD 格式

end: 结束时间, 可选, 不包含

str 或 datetime 类型 需要注意的是 str 必须要是 YYYYMMDD 格式

Returns: 所有交易日

util.utils.is_sh_symbol_code(code: str) → bool

检查股票代码是不是属于上交所

Args:

code: 待检查的股票代码

Returns:

util.utils.is_symbol_code(code: str) → bool

检查 code 是不是合法有效的 A 股股票代码

Args:

code: 代码

Returns:

util.utils.is_sz_symbol_code(code: str) → bool

检查股票代码是不是属于深交所

Args:

code: 待检查的股票代码

Returns:

util.utils.traverse_path(path: str) → list

遍历路径下的所有文件, 包含子目录

Args:

path: 需要遍历的路径

Returns: 路径下的所有文件

Module contents