【Python】csvファイルの値をWebブラウザで入力する

PG,Python

処理概要

対象URL

処理内容

  1. csvファイルから値を読み込む
  2. 読み込んだ値をWebブラウザに入力する
  3. Webブラウザで計算処理を実行する
  4. 計算結果をメッセージボックスで表示する

csvファイル

  • ファイル名：input.csv
  • 実行ディレクトリに配置する
  • 以下のヘッダを保持する
    • 年齢
    • 性別
    • 身体活動レベル
    • 目標体重

csvファイルの内容

年齢,性別,身体活動レベル,目標体重
22,男,高い,65
23,女,ふつう,55
24,女,低い,57
25,男,高い,72

コード

# ----------------------------------------
# モジュールのインポート
# ----------------------------------------
import csv
import os
import tkinter.messagebox

import chromedriver_binary
from selenium import webdriver
from selenium.webdriver.chrome import service as fs
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager

# メッセージボックス用の設定
root = tkinter.Tk()
root.withdraw()

# ----------------------------------------
# csvファイルの読み込み
# ----------------------------------------
csv_file = 'input.csv'
if os.path.exists(csv_file):
    with open(csv_file, mode='r', encoding='utf8') as f:
        reader = csv.reader(f)
        line = [row for row in reader]

    input_row = len(line)
    print('処理対象件数： ' + str(input_row - 1))

    if input_row < 2:
        tkinter.messagebox.showwarning('件数チェックエラー', '処理対象データがないため処理を終了します。')
        exit()

else:
    tkinter.messagebox.showerror('ファイルチェックエラー', f'『{csv_file}』が存在しないため処理を終了します。')
    exit()

# ----------------------------------------
# ChromeDriverの設定
# ----------------------------------------
# ChromeDriverの最適化
CHROMEDRIVER = ChromeDriverManager().install()

# オプションの設定
chrome_service = fs.Service(executable_path=CHROMEDRIVER)
chrome_options = webdriver.ChromeOptions()
# ブラウザ表示が不要な場合はコメントインする
# chrome_options.add_argument('--headless')
chrome_options.add_experimental_option('excludeSwitches', ['enable-automation', 'enable-logging'])

# ----------------------------------------
# 処理開始
# ----------------------------------------
print('>>>処理開始')
driver = webdriver.Chrome(service=chrome_service, options=chrome_options)

driver.maximize_window()
driver.implicitly_wait(10)
driver.get('https://keisan.casio.jp/exec/system/1183427246/')

# ----------------------------------------
# 入力に使用する画面要素の設定
# ----------------------------------------
age_xpath = '//*[@id="var_age"]'
sx0_xpath = '//*[@id="inparea"]/tbody/tr[2]/td[2]/ul/ol/li[5]/label[1]'
sx1_xpath = '//*[@id="inparea"]/tbody/tr[2]/td[2]/ul/ol/li[5]/label[2]'
alv1_xpath = '//*[@id="inparea"]/tbody/tr[3]/td[2]/ul/ol/li[1]/label[1]'
alv2_xpath = '//*[@id="inparea"]/tbody/tr[3]/td[2]/ul/ol/li[1]/label[2]'
alv3_xpath = '//*[@id="inparea"]/tbody/tr[3]/td[2]/ul/ol/li[1]/label[3]'
kg_xpath = '//*[@id="var_kg"]'
ans0_xpath = '//*[@id="ans0"]'
execute_xpath = '//*[@id="executebtn"]'
clear_xpath = '//*[@id="clearbtn"]'

# ----------------------------------------
# 情報の入力と計算の実行
# ----------------------------------------
for i in range(1, input_row):
    age = line[i][0]
    sex = line[i][1]
    act_level = line[i][2]
    weight = line[i][3]

    driver.find_element(By.XPATH, age_xpath).send_keys(age)

    if sex == '男':
        driver.find_element(By.XPATH, sx0_xpath).click()
    elif sex == '女':
        driver.find_element(By.XPATH, sx1_xpath).click()

    if act_level == '低い':
        driver.find_element(By.XPATH, alv1_xpath).click()
    elif act_level == '高い':
        driver.find_element(By.XPATH, alv3_xpath).click()
    else:
        driver.find_element(By.XPATH, alv2_xpath).click()

    driver.find_element(By.XPATH, kg_xpath).send_keys(weight)

    driver.find_element(By.XPATH, execute_xpath).click()

    # 計算結果の表示
    energy = driver.find_element(By.XPATH, ans0_xpath).text
    message = str(i) + '／' + str(input_row - 1) + '件目' + '''
    \n１日に必要なエネルギー量は ''' + energy + ' Kcalです'
    tkinter.messagebox.showinfo('計算結果', message)
    driver.find_element(By.XPATH, clear_xpath).click()

# ----------------------------------------
# 処理終了
# ----------------------------------------
print('<<<処理終了')
tkinter.messagebox.showinfo('処理終了', '処理が終了しました')
driver.quit()

PG,Python

Posted by junichi