from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time
import requests
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options


class Bot:
    url = 'https://ru.aliexpress.com/category/202001121/digital-batteries.html?ltype=wholesale&spm=a2g0o.home.110.3.b3c02a45W9i43a'
    driver = webdriver.Chrome(executable_path=r'C:\Users\Pioner\PycharmProjects\untitled17\drivers\chromedriver.exe')
    urls = []  # Ссылки всех страниц
    url_sellers = []  # Ссылки на всех продавцов
    url_chat = []

    def __init__(self, url, driver):  # Логинимся
        self.url = url
        self.driver = driver

        driver.get(url)

        driver.switch_to_frame('alibaba-login-box')
        driver.find_element_by_id("fm-login-id").send_keys('LOGIN')
        driver.find_element_by_id("fm-login-password").send_keys('PASSWORD', Keys.RETURN)
        time.sleep(3)
        driver.get(url)
        time.sleep(5)

    def sell(self, url, driver):  # Генерируем ссылки всех страниц
        x = url[62:]
        z = url.replace(x, '')
        for i in range(1, 5):
            new_url = f'{z}/{i}{x}'
            Bot.urls.append(new_url)

    def seller_chat(self, url, driver):  # Собираем ссылки всех чатов с продавцами

        for i in url:
            driver.get(i)
            elements = driver.find_elements_by_xpath('//a[contains(@class,"store")]')
            for j in elements:
                z = j.get_attribute('href')
                if z != 'https://my.aliexpress.com/wishlist/wish_list_store_list.htm' and z not in Bot.url_sellers:

                    Bot.url_sellers.append(z)
                else:
                    continue

    def message(self, url, driver):  # Пишем смс
        for i in url:

            driver.get(i)
            add = driver.find_element_by_xpath('//div[@aria-haspopup="true"]')
            Hover = ActionChains(driver).move_to_element(add)
            Hover.perform()
            try:
                driver.find_element_by_xpath('//span[text()="Связаться сейчас"]').click()
            except:
                pass
            time.sleep(3)


            time.sleep(3)
            l=driver.execute_script(driver.find_element_by_xpath("//p[@class='me-menu-title me-current-title']"))
            print(l)



botAle = Bot(Bot.url, Bot.driver)
botAle.sell(Bot.url, Bot.driver)
botAle.seller_chat(Bot.urls, Bot.driver)
botAle.message(Bot.url_sellers, Bot.driver)