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()