ジムバトルの討伐時間の自動取得！ジムバトルDB作成プログラムを紹介！【ポケモンGO】【ジム防衛】

import numpy as np
import pandas as pd
import time
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.webdriver import ActionChains

#chromeでアクセス
options = webdriver.ChromeOptions()
################################################################################################################################################################
# chromedriver.exeの場所を指定
service = webdriver.chrome.service.Service(executable_path = 'C:\\Users\\xxxxxxxx\\Desktop\\PokemonGo_project\\20230607_scraping\\chromedriver.exe')
################################################################################################################################################################
driver = webdriver.Chrome(service=service, options=options)
driver.get('https://pokemongo-get.com/damagecal/')
time.sleep(10)

select_list_A = driver.find_element(By.XPATH, '//*[@id="app"]/div[1]/div[1]/div/div[3]/div[4]/select')
# Selectオブジェクトを作成
dropdownA = Select(select_list_A)
# 選択肢を選択
dropdownA.select_by_visible_text('50')

select_list_B = driver.find_element(By.XPATH, '//*[@id="app"]/div[1]/div[2]/div/div[3]/div[4]/select')
# Selectオブジェクトを作成
dropdownB = Select(select_list_B)
# 選択肢を選択
dropdownB.select_by_visible_text('50')

#シャドウポケモンON
#shadow_button = driver.find_element(By.XPATH, '//*[@id="app"]/div[1]/div[1]/div/table/tbody/tr/td[2]')
#shadow_button.click()

# 検索欄取得
search_box_A = driver.find_element(By.XPATH, '//*[@id="app"]/div[1]/div[1]/div/input')
search_box_B = driver.find_element(By.XPATH, '//*[@id="app"]/div[1]/div[2]/div/input')

# ポケモンを受け取ったら撃破時間を取得する関数

def get_defeat_time(attack_pokemon, defence_pokemon):

    keyword_A=attack_pokemon[0]
    keyword_B=defence_pokemon

    search_box_A.clear()
    search_box_B.clear()

    search_box_A.send_keys(keyword_A)
    search_box_B.send_keys(keyword_B)

    result_button = driver.find_element(By.XPATH, '//*[@id="app"]/button[2]')
    result_button.click()

    time.sleep(2)

    # result_tableの要素を取得
    result_table = driver.find_element(By.XPATH, '//*[@id="app"]/div[3]/div[6]/table/tbody')
    # result_tableから<tr>要素を取得
    tr_elements = result_table.find_elements(By.TAG_NAME, 'tr')
    
    

    # 取得した<tr>要素を処理する
    for tr in tr_elements[1:]:
        figures = tr.find_elements(By.XPATH, './/*[contains(@class,"text-center")]')
        rank=figures[0].text
        defeat_time = float(figures[1].text)
        moves = tr.find_elements(By.XPATH, './/*[contains(@class,"flex-list-auto")]')
        fast_move = moves[0].text
        special_move = moves[1].text
        if (attack_pokemon[1] in fast_move) and (attack_pokemon[2] in special_move):
            print(keyword_A, ' vs ', keyword_B)
            print(rank)
            print(fast_move,special_move)
            print(defeat_time)
            return defeat_time
            
attack_pokemons = [
                   ["カミツルギ","はっぱカッター","リーフブレード"],
                   ["レシラム","ほのおのキバ","クロスフレイム"],
                   ["カイオーガ","たきのぼり","こんげんのはどう"],
                   ["デンジュモク","でんきショック","ほうでん"],
                   ["マンムー","こなゆき","ゆきなだれ"],
                   ["ルカリオ","カウンター","はどうだん"],
                   ["ウツロイド","どくづき","ヘドロばくだん"],
                   ["グラードン","マッドショット","だんがいのつるぎ"],
                   ["ファイヤー","つばさでうつ","ゴッドバード"],
                   ["ミュウツー","サイコカッター","サイコブレイク"],
                   ["フェローチェ","むしくい","むしのさざめき"],
                   ["ラムパルド","うちおとす","いわなだれ"],
                   ["ゲンガー","したでなめる","シャドーボール"],
                   ["パルキア","ドラゴンテール","りゅうせいぐん"],
                   ["ダークライ","バークアウト","あくのはどう"],
                   ["メタグロス","バレットパンチ","コメットパンチ"],
                   ["サーナイト","あまえる","マジカルシャイン"],
                   ["ポリゴンZ","ロックオン","はかいこうせん"],
                   ]
    
defence_pokemons = ["ハピナス",
                    "ソーナンス",
                    "ラッキー",
                    "トゲキッス",
                    "メタグロス",
                    "カイリュー",
                    "ベトベトン",
                    "ベトベトン(アローラ)",
                    "ヤドキング(ガラル)",
                    "デスカーン",
                    "デスバーン",
                    "ミカルゲ",
                    "ヌメルゴン",
                    "ミロカロス",
                    "カビゴン",
                    "ラグラージ",
                    "パンプジン(とくだい)",
                    "サーフゴー",
                    ]

# 動作確認&デモ用
attack_pokemons=attack_pokemons[:3]
defence_pokemons=defence_pokemons[:2]

df_time = pd.DataFrame(index=np.arange(len(attack_pokemons)),columns=["attack_pokemon","fast_move","special_move"])

for i,attack_pokemon in enumerate(attack_pokemons):
    df_time.loc[i,"attack_pokemon"]=attack_pokemon[0]
    df_time.loc[i,"fast_move"]=attack_pokemon[1]
    df_time.loc[i,"special_move"]=attack_pokemon[2]
    for j,defence_pokemon in enumerate(defence_pokemons):
        defeat_time = get_defeat_time(attack_pokemon,defence_pokemon)
        df_time.loc[i,defence_pokemon]=defeat_time
        
df_time.to_excel("check_demo.xlsx")
        
#いったん停止
time.sleep(2)

driver.quit()