diff --git a/src/sbi/main.py b/src/sbi/main.py index 8883610..1599132 100644 --- a/src/sbi/main.py +++ b/src/sbi/main.py @@ -49,19 +49,19 @@ def createCSV(table_data): # ディレクトリ作成とファイル名取得する -def get_file_path(): +def get_file_path(index): today = datetime.date.today() # 出力:datetime.date(2020, 3, 22) yyyymm = "{0:%Y%m}".format(today) # 202003 yyyymmdd = "{0:%Y%m%d}".format(today) # 20200322 os.makedirs(SAVE_DIR + yyyymm, exist_ok=True) - filepath = SAVE_DIR + yyyymm + "/" + yyyymmdd + ".csv" + filepath = SAVE_DIR + yyyymm + "/" + yyyymmdd + "_" + str(index) + ".csv" return filepath # 作成した文字列データ(CSV)を指定場所に書き込み -def writeCSV(rawoutputCSV): - filepath = get_file_path() +def writeCSV(rawoutputCSV, index): + filepath = get_file_path(index) outputCSV = reshapeCSV(rawoutputCSV) with open(filepath, mode="w") as f: @@ -103,16 +103,17 @@ def reshapeCSV(rawoutputCSV): soup = BeautifulSoup(driver.page_source, "html.parser") - # ポートフォリオの1テーブル目を取得 - table_data = soup.find( + # ポートフォリオのテーブルを取得 + table_data = soup.find_all( "table", bgcolor="#9fbf99", cellpadding="4", cellspacing="1", width="100%" ) - fetch_data = createCSV(table_data) - lg.info("create CSV") - - writeCSV(fetch_data) - lg.info("write CSV") + # 取得したテーブルを上から順に、#1, #2 をつけて YYYYMMDD_#x.csv として保存 + for i in range(len(table_data)): + fetch_data = createCSV(table_data[i]) + lg.info("create CSV: #{}".format(i + 1)) + writeCSV(fetch_data, i + 1) + lg.info("write CSV") # ブラウザを閉じる driver.quit()