Twitter automization test (4 factors, 32 accounts)

test(
    input_test_case=f'{REAL_CHROMEDRIVER}_{AUTOMATION}_{NON_PROFILED}_{NON_HEADLESS}',
    input_username='',
    input_password=''
)

import os
import subprocess
import time

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.remote.webelement import WebElement
import undetected_chromedriver
from selenium.webdriver.common.by import By
import pickle

UNDETECTED_CHROMEDRIVER: str = '1-1'
REAL_CHROMEDRIVER: str = '1-2'
DEFAULT_CHROMEDRIVER: str = '1-3'

AUTOMATION: str = '2-1'
MANUAL: str = '2-2'

PROFILED: str = '3-1'
NON_PROFILED: str = '3-2'

HEADLESS: str = '4-1'
NON_HEADLESS: str = '4-2'

driver: webdriver.Chrome
options: Options
profile: str = ''
test_case: str
username: str
password: str
test_no: int
picture_log_no: int = 0
article_no: int = 0
like_no: int = 0


# 1-1
def set_undetected_chromedriver_options():
    global options
    options = undetected_chromedriver.ChromeOptions()


def open_undetected_chromedriver():
    global options, driver, profile
    if profile == '':
        driver = undetected_chromedriver.Chrome(options=options)
    else:
        driver = undetected_chromedriver.Chrome(options=options, user_data_dir=profile)


# 1-2
def set_real_chromedriver_options():
    global options
    options = webdriver.ChromeOptions()
    options.add_experimental_option("debuggerAddress", "127.0.0.1:9222")


def open_real_chromedriver():
    global options, driver, profile

    cmd = [r'/Applications/Google Chrome.app/Contents/MacOS/Google Chrome', '--remote-debugging-port=9222',
           '--no-first-run', '--no-default-browser-check']
    if profile != '':
        cmd.append(f'--user-data-dir={profile}')
        cmd.append(f'--profile-directory=default')
    subprocess.Popen(cmd)

    driver = webdriver.Chrome(options=options)


# 1-3
def set_default_chromedriver_options():
    global options
    options = webdriver.ChromeOptions()


def open_default_chromedriver():
    global options, driver
    driver = webdriver.Chrome(options=options)


# 2-1 (first time when no cookie)
def automated_login():
    def login():
        driver.find_element(By.XPATH, "//span[text()='로그인']").click()
        capture_picture_log()
        driver.find_element(By.XPATH, "//input[@autocomplete='username']").send_keys(username)
        capture_picture_log()
        driver.find_element(By.XPATH, "//span[text()='다음']").click()
        capture_picture_log()
        driver.find_element(By.XPATH, "//input[@name='password']").send_keys(password)
        capture_picture_log()
        driver.find_element(By.XPATH, "//span[text()='로그인하기']").click()
        capture_picture_log()

    automation_retry(function=login, pass_if_fail=True)


# 2-1
def automated_likes():
    def click_close_popup():
        close_buttons = driver_find_multiple_query_elements(["//div[@aria-label='닫기']", "//div[@aria-label='Close']"])
        if len(close_buttons) > 0:
            close_buttons[0].click()
        capture_picture_log()

    def click_next_article():
        global article_no
        articles = driver.find_elements(By.XPATH, "//article")
        article = articles[article_no]
        article_no = article_no + 1
        article.click()
        capture_picture_log()

    def click_people_who_like_button():
        people_who_like_button = driver_find_multiple_query_elements(
            ["//span[text()='마음에 들어요']", "//span[text()='Likes']"])
        # if no people_who_like_button
        if len(people_who_like_button) == 0:
            # go back
            driver.back()
            capture_picture_log()
            # retry
            return
        else:
            people_who_like_button[0].click()

    def follow_article_liked():
        global like_no
        follow_buttons = driver_find_multiple_query_elements(["//span[text()='팔로우']", "//span[text()='Follow']"])
        capture_picture_log()

        for i in range(len(follow_buttons)):
            follow_buttons[i].click()
            capture_picture_log()
            like_no = like_no + 1
            if like_no == 10:
                break

    def click_home_button():
        home_buttons = driver.find_elements(By.XPATH, "//a[@href='/home']")
        if len(home_buttons) > 0:
            home_buttons[0].click()

    print('click_next_article')
    automation_retry(function=click_close_popup)
    automation_retry(function=click_next_article)

    print('click_people_who_like_button')
    automation_retry(function=click_close_popup)
    automation_retry(function=click_people_who_like_button)

    print('follow_article_liked')
    automation_retry(function=follow_article_liked)

    print('click_home_button')
    automation_retry(function=click_close_popup)
    automation_retry(function=click_home_button)

    global like_no
    if like_no < 10:
        automated_likes()


def driver_find_multiple_query_elements(querys: list[str], by: By = By.XPATH) -> list[WebElement]:
    elements = []
    for query in querys:
        elements.extend(driver.find_elements(by, query))
    return elements


# 2-1
def automation_retry(function: callable, pass_if_fail: bool = False, retry: int = 5):
    for i in range(retry):
        try:
            function()
            return
        except:
            time.sleep(1)
            pass
    if not pass_if_fail:
        function()


# 2-2 (first time when no cookie)
def manual_login():
    print(f'username: {username}')
    print(f'password: {password}')
    input("manually login.\n")
    print('good job!')


# 2-2
def manual_likes():
    input("manually like people.\n")
    print('good job!')


# 3-1
def profiled():
    global profile, test_case, options
    profile = f'./profiles/{test_case}'
    options.add_argument(f'--user-data-dir={profile}')
    options.add_argument(f'--profile-directory=default')


# 3-2
def non_profiled():
    global profile
    profile = ''


# 4-1
def headless():
    global options
    options.add_argument('--headless')


# 4-2
def non_headless():
    return


# common
def open_twitter():
    global driver
    driver.get('https://www.twitter.com')


# common
def save_cookies():
    global test_case, driver
    pickle.dump(driver.get_cookies(), open(f"./cookies/{test_case}_cookie.pkl", "wb"))


# common
def load_cookies():
    global profile, driver
    cookies = pickle.load(open(f"./cookies/{test_case}_cookie.pkl", "rb"))
    for cookie in cookies:
        driver.add_cookie(cookie)


# common
def cookie_exists():
    return os.path.isfile(f"./cookies/{test_case}_cookie.pkl")


# common
def reset_4g_proxy():
    input("reset your 4g proxy ip.\n")
    print('good job!')


# common
def common_setup():
    global picture_log_no, article_no, like_no
    picture_log_no = 0
    article_no = 0
    like_no = 0

    if not os.path.isdir(f"./cookies"):
        os.mkdir(f"./cookies")

    if not os.path.isdir(f"./profiles"):
        os.mkdir(f"./profiles")

    if not os.path.isdir(f"./log"):
        os.mkdir(f"./log")
    if not os.path.isdir(f"./log/{test_no}"):
        os.mkdir(f"./log/{test_no}")
    if not os.path.isdir(f"./log/{test_no}/{test_case}"):
        os.mkdir(f"./log/{test_no}/{test_case}")


# common
def capture_picture_log():
    global picture_log_no
    time.sleep(5)
    driver.save_screenshot(f'./log/{test_no}/{test_case}/{picture_log_no}.png')
    picture_log_no = picture_log_no + 1


def get_test_case_arguments(input_test_case: str):
    test_arguments = input_test_case.split('_')
    argument_1 = test_arguments[0]
    argument_2 = test_arguments[1]
    argument_3 = test_arguments[2]
    argument_4 = test_arguments[3]
    return argument_1, argument_2, argument_3, argument_4


def test(input_test_case: str, input_username: str, input_password: str):
    global test_case, username, password

    test_case = input_test_case
    username = input_username
    password = input_password
    driver_type, action_type, profile_type, headless_type = get_test_case_arguments(input_test_case)

    if headless_type == HEADLESS and action_type == MANUAL:
        raise 'HEADLESS cannot be MANUAL'

    # setup
    common_setup()

    # set chromedriver options
    if driver_type == UNDETECTED_CHROMEDRIVER:
        set_undetected_chromedriver_options()
    elif driver_type == REAL_CHROMEDRIVER:
        set_real_chromedriver_options()
    elif driver_type == DEFAULT_CHROMEDRIVER:
        set_default_chromedriver_options()

    # set profile
    if profile_type == PROFILED:
        profiled()
    elif profile_type == NON_PROFILED:
        non_profiled()

    # set headless
    if headless_type == HEADLESS:
        headless()
    elif headless_type == NON_HEADLESS:
        non_headless()

    # open chromedriver
    if driver_type == UNDETECTED_CHROMEDRIVER:
        open_undetected_chromedriver()
    elif driver_type == REAL_CHROMEDRIVER:
        open_real_chromedriver()
    elif driver_type == DEFAULT_CHROMEDRIVER:
        open_default_chromedriver()

    open_twitter()

    # load cookie
    if cookie_exists():
        load_cookies()
        open_twitter()

    # login, like
    if action_type == AUTOMATION:
        automated_login()
        automated_likes()
    if action_type == MANUAL:
        manual_login()
        manual_likes()

    # save cookie
    save_cookies()

    print('SUCCESS')


test_no = int(input("test No. is?\n"))
print(f'test No. {test_no}')

test(
    input_test_case=f'{REAL_CHROMEDRIVER}_{AUTOMATION}_{NON_PROFILED}_{NON_HEADLESS}',
    input_username='',
    input_password=''
)