from data_center import DataCenter

def _collect_stock(data_center,code):
    start_date = data_center.get_start_date(code)
    if start_date is None :
        start_date = "1997-01-01"
    stock_info = data_center.query_stock_info(code,start_date)
    if stock_info is not None:
        data_center.save_stock_info(code,stock_info)

def collect_data(date="2022-07-18"):
    data_center = DataCenter()
    all_stock = data_center.query_all_stock(date)
    for i in range(len(all_stock)):
        print(f"collect stock : {all_stock[i]},{i}/{len(all_stock)}")
        _collect_stock(all_stock[i])

collect_data()