def init(self):
df = ts.get_stock_basics()
if df is None: return False
df = df.reset_index(drop = False)
failed_list = list()
for _, code_id in df['code'].iteritems():
dbname = CStock.get_dbname(code_id)
if dbname not in self.mysql_dbs:
if not self.mysql_client.create_db(dbname): failed_list.append(code_id)
if len(failed_list) > 0:
logger.error("%s create failed" % failed_list)
return False
df['limitUpNum'] = 0
df['limitDownNum'] = 0
return self.redis.set(ct.STOCK_INFO, _pickle.dumps(df, 2))
def get_range_data(self, start_date, end_date, code):
sql = "select * from day where date between \"%s\" and \"%s\"" % (
start_date, end_date)
self.mysql_client.changedb(CStock.get_dbname(code))
return (code, self.mysql_client.get(sql))
