from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.keys import Keys
import sys
import time
import pyautogui
def savemsgsashtml(logins):
    url = "http://beon.ru/private/#!"
#    for i in range (0, len(logins) - 1):
#        url = url + logins[i]
#        if i != len(logins) - 1:
#            url = url + ","
#    driver.get(url)
#    while True:
#            if driver.find_element_by_id("tname_" + logins[0]).text ==:
#                time.sleep(3)
#            else:
#                break
    for i in range (0, len(logins) - 1):
        #driver.find_element_by_tag_name('body').send_keys(Keys.COMMAND + 't')
        #time.sleep(2)
        driver.get(url + logins[i])
        msgdiv = driver.find_element_by_id("messages_box_" + logins[i])
        height = 5
        last_height = 0
        while True:
            driver.execute_script("arguments[0].scrollTo(0, 0);", msgdiv)
            time.sleep(1)
            height = driver.execute_script("return arguments[0].scrollTop", msgdiv)
            print(height)
            if height <= 0:
                #source = driver.page_source
                #f=open("G://beon//users//" + logins[i] + ".html", "w+")
                #f.write(source.encode('utf-8').decode('utf-8'))
                #driver.find_element_by_id("tname_" + logins[i]).click()
                pyautogui.keyDown('ctrl')
                pyautogui.press('s')
                pyautogui.keyUp('ctrl')
                time.sleep(1)
                pyautogui.typewrite(logins[i])
                pyautogui.press('enter')
                time.sleep(2)
                pyautogui.keyDown('ctrl')
                pyautogui.press('t')
                pyautogui.keyUp('ctrl')
                time.sleep(1)
                driver.switch_to_window(driver.window_handles[-1])
                break
def parsehref(href):
    ss = ""
    for i in range(21, len(href) - 1):
        #print (href[i])
        ss = ss + href[i]
    return ss
def loginn(log, passw):
    driver.get("http://www.beon.ru/p/login.cgi")
    driver.find_element_by_xpath("//*[@id='forma1']/table/tbody/tr[2]/td[2]/input").send_keys(log)
    driver.find_element_by_xpath("//*[@id='forma1']/table/tbody/tr[3]/td[2]/input").send_keys(passw)
    driver.find_element_by_xpath("//*[@id='forma1']/table/tbody/tr[5]/td[2]/input").click()
def getlogins():
    logins = []
    row = 1
    page = 1
    while True:
        try:
            parent = driver.find_element_by_id("messagelist")
            elementList = parent.find_elements_by_class_name("rplain")
            current = parent.find_element_by_class_name("rcurrent")
            elementList.append(current)
        except NoSuchElementException:
            try:
                driver.get(driver.find_element_by_id("prevpage").get_attribute('href'))
                continue
            except NoSuchElementException:
                break
            #print (elementList)
        for i in range(0, len(elementList) - 1):
            user = parsehref(elementList[i].find_element_by_class_name("user").get_attribute('href'))
            #print(user)
            if not(any(user in s for s in logins)):
                logins.append(user)
                #print(logins)
        try:
            driver.get(driver.find_element_by_id("prevpage").get_attribute('href'))
        except NoSuchElementException:
            break
    return logins
login = input("login: ")
pw = input("password: ")
options = Options()
options.add_argument("--headless")
driver = webdriver.Firefox()
driver.implicitly_wait(3)
loginn(login, pw)
#time.sleep(5)
#print("lol")
driver.get("http://www.beon.ru/messages/sent/?c=1")##!!!!!!!!!!!!!!!
#x=1
#for i in range (1, x):
#                try:
#                    driver.find_element_by_id("row" + str(1))
#                except:
#               # ++i
#                   ++x
#                    print(str(x))
#                    sleep(2)
#print("lll")
logins = getlogins() #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
#logins = ['iibui', 'congratulator', 'Usuicat', 'Tekna884', 'ocogn', 'Outmysoul', 'mistick2014', 'diana123321d', 'iwannaknowwhatl', 'samantaspy', 'MADAROKl', 'mitilala']
savemsgsashtml(logins)
print(logins)

#driver.get("http://beon.ru/messages/sent/24.html")
#driver.close()