def importStockListing(cls):
"""
插入所有上市股票公司
:return:
"""
from oneilquant.ONEIL.Oneil import OneilKDZD as oneil
oq = oneil()
n1 = 0
df = oq.listingDate(n1)
# todo 如果已经插入，则判断是否有更新 调用savedf方法可以实现判断有更新
try:
# 批量创建对象，减少SQL查询次数
querysetlist = []
for i in df.index:
a = df.loc[i]
d = int(a.timeToMarket)
if a[0] == '6':
market = 1
else:
market = 0
category = 10
querysetlist.append(
Listing(code=a.name,
name=a['name'],
timeToMarket=datetime.datetime(
d // 10000, d // 100 % 100, d % 100),
category=category,
market=market))
cls.objects.bulk_create(querysetlist)
except Exception as e:
print(e.args)
return pd.DataFrame()
return df
def test_importStockcode(self):
"""
测试导入股票代码
:return:
"""
from oneilquant.ONEIL.Oneil import OneilKDZD as oneil
oq = oneil()
n1 = 365
df = oq.listingDate(n1)
len1 = len(df)
self.assertTrue(len(df) > 2500, "上市一年以上股票数量，2018年最少大于2500只")
print(df[:10]['timeToMarket'], df['name'])
print(df.index[:10])
def setUp(self):
self.oq = oneil()