from selenium.webdriver.firefox.webdriver import WebDriver

__author__ = 'Boris Osyanin'


class Application:
    def __init__(self):
        self.wd = WebDriver()
        self.wd.implicitly_wait(60)
        print()

    def open_home_page(self):
        wd = self.wd
        wd.get("https://staging.remontnik.ru")

    def login_email(self, username, password):
        wd = self.wd
        self.open_home_page()
        wd.find_element_by_link_text("Войти").click()
        wd.find_element_by_name("credential").send_keys(username)
        wd.find_element_by_name("password").send_keys(password)
        wd.find_element_by_css_selector("div.auth-block__buttons > button").click()
        wd.find_element_by_link_text("Добавить заказ").is_displayed()

    def login_sms(self, phone):
        wd = self.wd
        self.open_home_page()
        wd.find_element_by_link_text("Войти").click()
        wd.find_element_by_partial_link_text("SMS").click()
        wd.find_element_by_name("phone").send_keys(phone)

    def logout(self):
        wd = self.wd
        wd.find_element_by_css_selector("li.user-menu.dropdown div.dropdown-toggle").click()
        wd.find_element_by_css_selector(".dropdown-menu > ul:nth-child(1) > li:nth-child(4) > a:nth-child(1)").click()
        wd.find_element_by_link_text("Добавить заказ")

    def destroy(self):
        self.wd.quit()