class Updatedb(object): def __init__(self): self.db = Dboperation() self.code = None def _date_start_update(self): latest = self.db.get_latest_date() now = latest + datetime.timedelta(1) return str(now) def _get_ts_data(self): if self.code in market_index_list: return ts.get_h_data(code=self.code[1:], index=True, start=self._date_start_update()) else: return ts.get_h_data(code=self.code, autype='hfq', start=self._date_start_update()) def append_to_db(self): #market_latest_date = get_market_date(1) #if self.db.get_latest_date() == market_latest_date: # if self.code != sh: # return df = self._get_ts_data() if df is None: return db_data = df.T.to_dict() self.db.append_db(db_data) def run_update(self): for code in market_index_list + code_list.keys(): self.code = code self.db.initialize(code) if not self.db.exist: self.db.create_table() self.append_to_db() self.db.initialize(code)
def __init__(self): self.db = Dboperation() self.code = None