from selenium import webdriver
from selenium.webdriver.support.ui import Select
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from webdriver_manager.chrome import ChromeDriverManager
import os
import time
import signal

# 初期設定  ----------------------
USER_ID = "XXXXXXXXXXXXXXX"
PASSWORD = "YYYYYYYYYYYYYY"
DOWNLOAD_DIR = "\\_temp"         # ティックデータをダンウロードするフォルダを指定
DOWNLOAD_INTERVAL = 1            # 各ファイルのダウンロード待機期間。短すぎるとエラーになりやすい??
CURRENCY_PAIR = "XAUUSD"         # 通貨ペア or 商品名
YEAR_START= 2025                   # ダウンロード対象開始年 (-1: 存在する最初のもの)
YEAR_END = -1                    # ダウンロード対象終了年 (-1: 存在する最後のもの)

options = webdriver.ChromeOptions()
prefs = {"download.default_directory": os.path.abspath(DOWNLOAD_DIR)}
options.add_experimental_option("prefs", prefs)

# ドライバーを自動でインストール
driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()),options=options)  ### Selenium4の場合

# 最大の読み込み時間を設定 今回は最大30秒待機できるようにする
wait = WebDriverWait(driver=driver, timeout=30)

url = "https://www.oanda.jp/trade/web/fxTradeLogin.do"
try:
    # 手順1: 指定したURLを開く
    driver.get(url)

    driver.find_element(By.NAME, "username").send_keys(USER_ID)
    driver.find_element(By.NAME, "password").send_keys(PASSWORD)

    #　クッキー確認ダイアログが表示、ログイン要素を隠してしまうので execute_scripｔを利用して回避
    #driver.find_element(By.NAME, "action").click()
    element = driver.find_element(By.NAME, "action")
    driver.execute_script("arguments[0].click();", element)

    # テックデータのページを開く
    element = driver.find_element(By.XPATH, '//*[@id="main"]/div[2]/div/div/article/section[4]/div[2]/div[2]/div[2]/a').click()
    wait.until(EC.presence_of_all_elements_located)

    # 通貨ペア表示セクションに移動
    element = driver.find_element(By.XPATH, '//*[@id="pairs"]')
    driver.execute_script("arguments[0].scrollIntoView({block:\"center\"});", element)

    # 通貨ペアを選択
    element = driver.find_element(By.XPATH, '//*[@id="pairs_' + CURRENCY_PAIR + '"]')
    driver.execute_script("arguments[0].click();", element)

    # 年のドロップダウンを選択
    dropdown = driver.find_element(By.XPATH, '//*[@id="yaerSelector"]')

    for option in Select(dropdown).options:
        year = option.get_attribute('value')

        if (YEAR_START == -1 or (YEAR_START <= int(year))) \
            and (YEAR_END == -1 or (YEAR_END >= int(year))):

            # 年ドロップダウンのリストを選択し
            Select(dropdown).select_by_value(year)
            print("Processing year=" + year)

            dropdown2 = driver.find_element(By.XPATH, '//*[@id="monthSelector"]')
            for option2 in Select(dropdown2).options:
                month = option2.get_attribute('value')
                Select(dropdown2).select_by_value(month)
                print("  -Processing month=" + month)

                submit = driver.find_element(By.XPATH, '//*[@id="download_btn"]')
                driver.execute_script("arguments[0].click();", submit)

                time.sleep(DOWNLOAD_INTERVAL)

    #time.sleep(300)  #　ダウンロードが完了するのを5分待機
    print("Finished all downloads.")

    # ファイルダウンロードが完了するまで待機
    input()

# エラーが発生した時はエラーメッセージを吐き出す。
except Exception as e:
    print(e)
    print("An error occurred.")

# 最後にドライバーを終了する
finally:
    driver.quit()