connect_and_launch.py

import asyncio
import time
import os
import logging

from selenium import webdriver
from selenium.common.exceptions import ElementNotInteractableException, \
                                       NoSuchElementException
from dotenv import load_dotenv
from chromedriver_py import binary_path

load_dotenv()
USER = os.getenv('USERNAME_C')
PASSWORD = os.getenv('PASSWORD_C')
URL = "https://aternos.org/go/"

# chrome variables
adblock = False  # for those with network wide ad blockers
headless = True  # if you want a headless window

options = webdriver.ChromeOptions()
if headless:
    options.add_argument('--headless')
options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
                     "AppleWebKit/537.36 (KHTML, like Gecko) "
                     "Chrome/87.0.4280.88 Safari/537.36")

driver = webdriver.Chrome(options=options, executable_path=binary_path)


async def start_server():
    """ Starts the server by clicking on the start button.
        The try except part tries to find the confirmation button, and if it
        doesn't, it continues to loop until the confirm button is clicked."""
    element = driver.find_element_by_xpath("//*[@id=\"start\"]")
    element.click()
    await asyncio.sleep(3)
    # hides the notification question
    driver.execute_script('hideAlert();')
    # server state span
    while get_status() == "Waiting in queue":
        # while in queue, check for the confirm button and try click it
        await asyncio.sleep(3)
        try:
            element = driver.find_element_by_xpath('//*[@id="confirm"]')
            element.click()
        except ElementNotInteractableException:
            pass


def get_status():
    """ Returns the status of the server as a string."""
    return driver.find_element_by_xpath('//*[@id="nope"]/main/section/div['
                                        '3]/div[3]/div[1]/div/span['
                                        '2]/span').text


def get_number_of_players():

    """ Returns the number of players as a string.
        Works: When server is online--Returns 0 if offline"""
    try:
        return driver.find_element_by_xpath('//*[@id="nope"]/main/section'
                                            '/div[3]/div[5]/div[2]/div['
                                            '1]/div[1]/div[2]/div[2]').text
    except NoSuchElementException:
        # Can't be 0/20 because max isn't always the same,
        # could maybe pull max players from options page
        return '0'


def get_ip():
    """ Returns the severs IP address.
        Works: Always works"""
    return driver.find_element_by_xpath('//*[@id="nope"]/main/section/div['
                                        '3]/div[1]').text[:-8]


def get_software():
    """ Returns the server software.
        Works: Always works"""
    return driver.find_element_by_xpath('//*[@id="software"]').text


def get_version():
    """ Returns the server version.
        Works: Always works"""
    return driver.find_element_by_xpath('//*[@id="version"]').text


def get_tps():
    """ Returns the server TPS
        Works; When the server is online--Returns '0' if offline"""
    try:
        return driver.find_element_by_xpath('//*[@id="nope"]/main/section'
                                            '/div[3]/div[5]/div[2]/div['
                                            '1]/div[3]/div[2]/div[2]').text
    except NoSuchElementException:
        return '0'


def get_server_info():
    """ Returns a string of information about the server
        Returns: server_ip, server_status, number of players, software,
        version"""
    return get_ip(), get_status(), get_number_of_players(), \
           get_software(), get_version(), get_tps()


def connect_account():
    """ Connects to the accounts through a headless chrome tab so we don't
        have to do it every time we want to start or stop the server."""
    driver.get(URL)
    # login to aternos
    element = driver.find_element_by_xpath('//*[@id="user"]')
    element.send_keys(USER)
    element = driver.find_element_by_xpath('//*[@id="password"]')
    element.send_keys(PASSWORD)
    element = driver.find_element_by_xpath('//*[@id="login"]')
    element.click()
    time.sleep(2)

    # selects server from server list
    element = driver.find_element_by_css_selector('body > div > main > section'
                                                  '> div > div.servers.single '
                                                  '> div > div.server-body')
    element.click()

    # by passes the 3 second adblock
    if adblock:
        adblockBypass()

    logging.info('Aternos Tab Loaded')


def adblockBypass():
    time.sleep(1)
    element = driver.find_element_by_xpath('//*[@id="sXMbkZHTzeemhBrPtXgBD'
                                           'DwAboVOOFxHiMjcTsUwoIOJ"]/div/'
                                           'div/div[3]/div[2]/div[3]/div'
                                           '[1]')
    element.click()
    time.sleep(3)
    logging.debug("Adblock Wall Bypassed")


async def stop_server():
    """ Stops server from aternos panel."""
    element = driver.find_element_by_xpath("//*[@id=\"stop\"]")
    element.click()


def quitBrowser():
    """ Quits the browser driver cleanly."""
    driver.quit()


def refreshBrowser():
    driver.refresh()