from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
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')
        for pms in message:
            print('---')
            print(pms.text)
            print('---')
        message = message[8]
        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)
    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