プログラムでワイモバイルにお支払い（PayPay回数稼ぎ）

import time

from helium import *

url = 'https://www.ymobile.jp/'


def login(phone_num, password):
    ymobile_button = find_all(S('.header-submenu-myymobile'))[0]
    click(ymobile_button)
    click('ログイン')
    driver = get_driver()
    time.sleep(2)
    driver.find_element_by_name('telnum').send_keys(phone_num)
    driver.find_element_by_name('password').send_keys(password)
    click('ログインする')


def pay_repeat_num(repeat_num, onetime_pay):
    for num in range(int(repeat_num)):
        time.sleep(2)
        click('料金案内')
        time.sleep(0.5)
        click('PayPay管理')
        click('申し込む')
        write(onetime_pay, into='円')
        click('次へ')
        click('注意事項に同意して申し込む')
        print(f'{num + 1}回目の支払い完了')
        click('My Y!mobileトップへ')
        time.sleep(5)


def main():
    phone_num = input('Yモバイルの電話番号を入力してください')
    password = input('パスワードを入力してください')
    start_chrome(url)
    login(phone_num, password)

    repeat_num = input('支払いの回数を入力してください')
    onetime_pay = input('1度に支払う金額を入力してください')
    pay_repeat_num(repeat_num, onetime_pay)

    kill_browser()


if __name__ == '__main__':
    main()