scrape.py 文件源码

def download_csv(ticker_name, period_start, period_end, frequency='daily'):
    url = BASE_URL + "{}&a={}&b={}&c={}&d={}&e={}&f={}&g={}&ignore=.csv".format(ticker_name,
                            period_start.month - 1,
                            period_start.day,
                            period_start.year,
                            period_end.month - 1,
                            period_end.day,
                            period_end.year,
                            FREQUENCY[frequency])
    path = CSV_PATH_PREFIX
    if not os.path.exists(path):
        os.makedirs(path)
    filename = "{}{}.csv".format(path, ticker_name)
    try:
        urllib.request.urlretrieve(url, filename)
    except urllib.ContentTooShortError as e:
        f = open(filename, 'w')
        f.write(e.content)
        f.close()
    finally:
        return filename