87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
import os
|
|
|
|
from Common.CEnum import DATA_FIELD, KL_TYPE
|
|
from Common.ChanException import CChanException, ErrCode
|
|
from Common.CTime import CTime
|
|
from Common.func_util import str2float
|
|
from KLine.KLine_Unit import CKLine_Unit
|
|
|
|
from .CommonStockAPI import CCommonStockApi
|
|
|
|
|
|
def create_item_dict(data, column_name):
|
|
for i in range(len(data)):
|
|
data[i] = parse_time_column(data[i]) if column_name[i] == DATA_FIELD.FIELD_TIME else str2float(data[i])
|
|
return dict(zip(column_name, data))
|
|
|
|
|
|
def parse_time_column(inp):
|
|
# 20210902113000000
|
|
# 2021-09-13
|
|
if len(inp) == 10:
|
|
year = int(inp[:4])
|
|
month = int(inp[5:7])
|
|
day = int(inp[8:10])
|
|
hour = minute = 0
|
|
elif len(inp) == 17:
|
|
year = int(inp[:4])
|
|
month = int(inp[4:6])
|
|
day = int(inp[6:8])
|
|
hour = int(inp[8:10])
|
|
minute = int(inp[10:12])
|
|
elif len(inp) == 19:
|
|
year = int(inp[:4])
|
|
month = int(inp[5:7])
|
|
day = int(inp[8:10])
|
|
hour = int(inp[11:13])
|
|
minute = int(inp[14:16])
|
|
else:
|
|
raise Exception(f"unknown time column from csv:{inp}")
|
|
return CTime(year, month, day, hour, minute)
|
|
|
|
|
|
class CSV_API(CCommonStockApi):
|
|
def __init__(self, code, k_type=KL_TYPE.K_DAY, begin_date=None, end_date=None, autype=None):
|
|
self.headers_exist = True # 第一行是否是标题，如果是数据，设置为False
|
|
self.columns = [
|
|
DATA_FIELD.FIELD_TIME,
|
|
DATA_FIELD.FIELD_OPEN,
|
|
DATA_FIELD.FIELD_HIGH,
|
|
DATA_FIELD.FIELD_LOW,
|
|
DATA_FIELD.FIELD_CLOSE,
|
|
# DATA_FIELD.FIELD_VOLUME,
|
|
# DATA_FIELD.FIELD_TURNOVER,
|
|
# DATA_FIELD.FIELD_TURNRATE,
|
|
] # 每一列字段
|
|
self.time_column_idx = self.columns.index(DATA_FIELD.FIELD_TIME)
|
|
super(CSV_API, self).__init__(code, k_type, begin_date, end_date, autype)
|
|
|
|
def get_kl_data(self):
|
|
cur_path = os.path.dirname(os.path.realpath(__file__))
|
|
file_path = f"{cur_path}/../{self.code}.csv"
|
|
if not os.path.exists(file_path):
|
|
raise CChanException(f"file not exist: {file_path}", ErrCode.SRC_DATA_NOT_FOUND)
|
|
|
|
for line_number, line in enumerate(open(file_path, 'r')):
|
|
if self.headers_exist and line_number == 0:
|
|
continue
|
|
data = line.strip("\n").split(",")
|
|
if len(data) != len(self.columns):
|
|
raise CChanException(f"file format error: {file_path}", ErrCode.SRC_DATA_FORMAT_ERROR)
|
|
if self.begin_date is not None and data[self.time_column_idx] < self.begin_date:
|
|
continue
|
|
if self.end_date is not None and data[self.time_column_idx] > self.end_date:
|
|
continue
|
|
yield CKLine_Unit(create_item_dict(data, self.columns))
|
|
|
|
def SetBasciInfo(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def do_init(cls):
|
|
pass
|
|
|
|
@classmethod
|
|
def do_close(cls):
|
|
pass
|