from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
import time,os

rkuji_list_url = "https://rakucoin.appspot.com/rakuten/kuji/" 
rkuji_list_xpath = "//table/tbody/tr/td/a"

class class_rkuji:
    def __init__(self, profile_dir="/tmp/profile"):
        options = webdriver.ChromeOptions()
        options.add_argument("--window-size=1920,1080")
        options.add_argument("--user-data-dir=" + profile_dir )
        self.driver = webdriver.Remote(
            command_executor="http://selenium:4444/wd/hub",
            options=options
            )

    def get_rkuji_urls(self, url, xpath):
        self.driver.get(url)
        self.urls = self.driver.find_elements(By.XPATH, xpath)
        self.url_txts = [url.get_attribute("href") for url in self.urls ]

    def challenge_rkuji(self):
        for u in self.url_txts:
            if 'redirect' in u:
                print(f'ラッキーくじ：「{u}」はスキップ。(理由:redirect)')
                continue
            try:
                self.driver.get(u)
                kuji_title = self.driver.title
            except:
                print(f'ラッキーくじ：「{u}」はスキップ。(理由:アクセス失敗)')
                continue

            if(len(self.driver.find_elements(By.ID, "entry"))>0):
                print(f'ラッキーくじ：「{kuji_title}」くじを引きます。')
                time.sleep(5)
                try:
                    start_button = self.driver.find_element(By.ID, "entry")
                    start_button.click()
                    WebDriverWait(self.driver,60).until(EC.url_changes(u))
                    print(f'ラッキーくじ：「{kuji_title}」くじを引きました。')
                    self.driver.save_screenshot('./ss/r_kuji.png')
                    time.sleep(5)
                except:
                    (f'ラッキーくじ：「{kuji_title}」はスキップ。(理由:くじ引き失敗)')
            else:
                print(f'ラッキーくじ：「{kuji_title}」はスキップ。(理由:くじ引き不能)')

    def close(self):
        self.driver.quit()

def rkuji():
    driver = class_rkuji()
    driver.get_rkuji_urls(rkuji_list_url, rkuji_list_xpath)
    driver.challenge_rkuji()
    driver.close()