# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
import thread

class LinkedInAddContacts(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "https://www.linkedin.com/"
        self.verificationErrors = []
        self.accept_next_alert = True

    def test_linked_in_add_contacts(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("login-email").clear()
        driver.find_element_by_id("login-email").send_keys("XXXXXXX")
        driver.find_element_by_id("login-password").clear()
        driver.find_element_by_id("login-password").send_keys("XXXXXXXXXXXXX")
        driver.find_element_by_id("login-submit").click()

        print("going to main page")
        driver.find_element_by_xpath("//li/label").click() # main page
        time.sleep(10)

        print("trying to open my network")
        driver.find_element_by_css_selector("#mynetwork-nav-item > a.nav-item__link").click() # open My Network
        time.sleep(5)



        while True:

            try:
                driver.find_element_by_class_name("mn-person-card__person-btn-ext.button-secondary-medium").click()
                time.sleep(1)
            except:
                pass
                print("trying to scroll")
                driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
                time.sleep(1)

    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException as e: return False
        return True

    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException as e: return False
        return True

    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main()