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