value_stock.py 文件源码

def main(keys):
    t0 = time.time()

    ARGS = dict(zip(['command', 'date'], sys.argv))
    date = ARGS.get('date', f'{datetime.date.today():%Y%m%d}')

    executor = futures.ThreadPoolExecutor(max_workers=len(keys))

    to_do = []
    for key in keys:
        config = copy.deepcopy(krx.load_config(key))
        config['contents']['data']['schdate'] = date
        future = executor.submit(download, key, config)
        to_do.append(future)

    done = 0
    total_records = 0
    for future in futures.as_completed(to_do):
        done += 1
        key, records = future.result()
        total_records += records
        print(f'[{done:,d}/{len(keys):,d}] {key} records fetched {records:>8,} records', file=sys.stderr)

    elapsed = time.time() - t0
    print(f'{total_records:,} Records fetched in {elapsed:.2f}s', file=sys.stderr)