from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import ElementNotVisibleException
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import sys
import time

def main():
    while True:
        checkMessages()


        time.sleep(180)
    return

def checkMessages():
    message = False
    try:
        message = chrome.find_element(By.CSS_SELECTOR, 'span.header.new_mail')
    except NoSuchElementException:
        print('Keine neuen Nachtichten')

    if not message == False:
        print('Neue Nachricht')
        message.click()
        message = chrome.find_elements(By.CLASS_NAME, 'vis')
        message = message[8]


        filter = message.find_element(By.ID, 'filter_none')
        filter.click()

        filter_table = message.find_element(By.ID, 'filter')
        neu_link = filter_table.find_element(By.PARTIAL_LINK_TEXT, "Neu")
        neu_link.click();

        vis_tds = chrome.find_elements(By.CSS_SELECTOR, '.vis td')
        for td in vis_tds:
            print(td.text)

        '''
        messages = message.find_elements_by_tag_name('a')
        for pm in messages:
            print('link: ')
            print(pm.text)
            pm.click()
            time.sleep(3)
            chrome.back
            time.sleep(3)
        '''

        message = chrome.find_elements(By.CLASS_NAME, 'vis')
        message = message[8]

        filter = message.find_element(By.ID, 'filter_new')
        filter.click()

        filter_table = message.find_element(By.ID, 'filter')
        neu_link = filter_table.find_element(By.PARTIAL_LINK_TEXT, "Kein")
        neu_link.click();

    else:
        print('Keine neuen Nachtichten')
    return

print("Name: ")
name = 'progammer'#sys.stdin.readline()

print("Password: ")
password = '123456q'#sys.stdin.readline()

print('Welt:')
world = '132'#sys.stdin.readline()

awe = True

chrome = webdriver.Chrome('D:\Python\selenium\webdriver\chrome\chromedriver.exe')
chrome.get('https://www.die-staemme.de')
chrome.implicitly_wait(15)
print('Oeffne die-staemme.de')

login_form = WebDriverWait(chrome, 15).until(EC.presence_of_element_located((By.ID, 'login_form')))
if login_form == False:
    awe = False

if(awe):
    login_user = login_form.find_element(By.ID, 'user')
    login_password = login_form.find_element(By.ID, 'password')
    login_button = login_form.find_element(By.CLASS_NAME, 'button_middle')
    login_user.send_keys(name)
    login_password.send_keys(password)
    login_button.click()

    server_form = WebDriverWait(chrome, 15).until(EC.presence_of_element_located((By.ID, 'server_select_list')))
    if server_form == False:
        awe = False

    if(awe):
        world_button = server_form.find_elements(By.CLASS_NAME, 'world_button_active')
        for worlds in world_button:
            text = str(worlds.text)
            welt = 'Welt '+str(world)
            if text == welt:
                worlds.click()
                print('Auf ' +welt+' eingeloggt')
                time.sleep(3)
                main()
                break