From a1ed28ac9540494102d8f9bc90a96e1bcf6f450e Mon Sep 17 00:00:00 2001
From: root <root@7milch.com>
Date: Thu, 1 Jun 2023 17:40:41 +0900
Subject: [PATCH] =?UTF-8?q?=E8=A6=96=E8=81=B4=E4=B8=AD=E3=81=AE=E3=82=A8?=
 =?UTF-8?q?=E3=83=A9=E3=83=BC=E5=87=A6=E7=90=86=E8=BF=BD=E5=8A=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/main.py | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/app/main.py b/app/main.py
index 8dd098e..1b2c2aa 100644
--- a/app/main.py
+++ b/app/main.py
@@ -50,7 +50,7 @@ def init_profiles():
     wait = WebDriverWait(driver, 10)
     element = wait.until(EC.presence_of_element_located((By.TAG_NAME, 'body')))
 
-    driver.save_screenshot('./ss/ss.png')
+    driver.save_screenshot('./ss/initial.png')
 
     # Welcomeメッセージを閉じる
     try: 
@@ -71,22 +71,19 @@ def init_profiles():
         user = driver.find_element(By.CSS_SELECTOR, user_textarea_css)
         user.send_keys(username)
 
-        driver.save_screenshot('./ss/ss_usrname.png')
         user.send_keys(Keys.ENTER)
         time.sleep(10)
 
         # パスワード入力
         passwd = driver.find_element(By.CSS_SELECTOR, passwd_textarea_css)
         passwd.send_keys(password)
-        driver.save_screenshot('./ss/ss_password.png')
         passwd.send_keys(Keys.ENTER)
         time.sleep(10)
-        driver.save_screenshot('./ss/ss_password2.png')
 
     except Exception as e:
         print(e)
 
-    driver.save_screenshot('./ss/ss2.png')
+    driver.save_screenshot('./ss/begin.png')
 
     # Coockie 取得
     cookies = driver.get_cookies()
@@ -103,20 +100,28 @@ def watching_rch():
     wait = WebDriverWait(driver, 10)
     element = wait.until(EC.presence_of_element_located((By.TAG_NAME, 'body')))
 
-    # 10秒待機
-    time.sleep(10)
-
     for i in range(0, rch_num):
+        # 30秒待機
+        time.sleep(30)
+
         print( str( i + 1) + "番目のチャンネルの視聴を始めます。視聴時間は" + str(rch_time) + "秒に設定されています。")
-        channel_btns = driver.find_elements(By.CSS_SELECTOR, channel_btn_css)
-        channel_btn = channel_btns[i]
-        channel_btn.click()
-        time.sleep(10)
-        driver.save_screenshot('./ss/channel_' + str(i) + '.png')
+        driver.save_screenshot('./ss/watching.png')
+        try:
+            channel_btns = driver.find_elements(By.CSS_SELECTOR, channel_btn_css)
+            channel_btn = channel_btns[i]
+            channel_btn.click()
+        except:
+            print("チャンネルが選択できませんでした。")
+            driver.save_screenshot('./ss/error.png')
+
         time.sleep(rch_time)
 
-    driver.quit()
+        try:
+            driver.refresh()
+        except:
+            print("ページの更新に失敗しました。")
 
+    driver.quit()
 
 def main():
     init_profiles()