from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.action_chains import ActionChains
import chromedriver_binary
import requests
import time
import sys
import os
import random
import json

import slackweb
from datetime import datetime as dt
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from webdriver_manager.chrome import ChromeDriverManager
# ランダムリスト
l = [3, 4, 5, 6, 7]
keys_json = open('config.json', 'r')
keys = json.load(keys_json)
# 任意に変更してください
DRIVER_PATH = keys["driver_path"]
userdata_dir = 'image_data'
os.makedirs(userdata_dir, exist_ok=True)
# WindowsPC用　exeを用意しないとうまくいかない可能性が高い
# driver_path = 'C:\driver\chromedriver.exe'
if __name__ == '__main__':

    through_flg = 0
    layout_num = 0
    footprint_num = 0

    # ブラウザーでみられるようになる
    # options.add_argument('--headless')
    # Chromeのドライバを得る
    print("★処理をスタートするよ！！")
    print("開始日時：" + dt.now().strftime("%Y-%m-%d %H:%M:%S"))

    # テーブルの作成
    # check_sqlite.createProfilesTable()

    # message = '現在のあしあと数: ' + str(footprint_num)
    # slack = slackweb.Slack(url="https://hooks.slack.com/services/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")
    # slack.notify(text=message)
    # sys.exit()

    options = webdriver.ChromeOptions()
    options.add_argument('--user-data-dir=' +
                         keys["path"] + '\\' + userdata_dir)
    # if DRIVER_PATH == '':
    #     browser = webdriver.Chrome(chrome_options=options)
    # else:
    #     browser = webdriver.Chrome(chrome_options=options,executable_path=DRIVER_PATH)
    # wait = WebDriverWait(browser, 10)
    browser = webdriver.Chrome(
        ChromeDriverManager().install(), options=options)

    print("ログイン")
    # pairsログイン
    browser.get("https://momon-ga.com/magazine/mo2127225/")

    sleep_time = 3
    print(str(sleep_time) + "秒待機")
    time.sleep(sleep_time)

    # if len(browser.find_elements_by_class_name('css-20y5ly')) == 1:
    #     element = browser.find_element_by_class_name('css-20y5ly')

    # layout_count = len(element.find_elements_by_tag_name('li'))
    # layout_count = len(element.find_elements_by_xpath('//li[@id="layout"]'))

    # print("aaaa")
    # sleep_time = 50
    # print(str(sleep_time) + "秒待機")
    # time.sleep(sleep_time)

    layout = browser.find_element_by_id('post-hentai')
    nodes = layout.find_elements_by_css_selector(
        'img')
    print(len(nodes))
    i = 1
    for node in nodes:
        # print(child_node.text)
        image_url = node.get_attribute("src")

        response = requests.get(image_url)
        image = response.content
        file_name = str(i) + ".webp"
        # print(os.path.join(userdata_dir, file_name))

        with open(os.path.join(userdata_dir, file_name), "wb") as aaa:
            aaa.write(image)

        i = i + 1

    exit()