ThreadingLib.py

import logging
import threading

import AutoSearch
import BrowserManager
import RewardsValidator


class ThreadingLib:
    """
    This class is used in order to launch Threads easily by passing all the parameters before starting the threads
    """
    def __init__(self, user: str, password: str, e: threading.Event, use_headless: bool = False,
                 incognito: bool = False, path: str = ""):
        """
        Constructor of the ThreadingLib class
        :param user: Username of the used Microsoft account
        :param password: Password of the used Microsoft account
        :param e: Event fired if 2FA is detected as enabled
        :param use_headless: Launches Chrome in the background if True (=invisible)
        :param incognito: Launches Chrome in private mode
        :param path: Path to the Google Chrome WebDriver executable
        """
        self.user = user
        self.password = password
        self.e = e
        self.code = []
        self.use_headless = use_headless
        self.incognito = incognito
        self.path = path

    def send_2fa_code(self, code: str):
        """
        Function to send the code to all the threads, the code is sent as a list because a list is stored by reference
        So basically a list is a pointer to a memory location, where the actual value is stored, all the threads will
        go read this exact same value in memory
        :param code: The code to send to all threads
        """
        self.code.append(code)

    def start_mobile(self):
        """
        Function used to perform the logic in order to start mobile searches
        """
        logging.info("=====LOGGING IN (MOBILE)=====")
        with BrowserManager.start_bing(
                self.user, self.password, self.e, self.code, incognito=self.incognito, use_headless=self.use_headless,
                mobile=True, path=self.path) as browser:
            logging.info("=====AUTO-SEARCH STARTED (MOBILE)=====")
            AutoSearch.start(browser, 25)
            logging.info("=====AUTO-SEARCH ENDED (MOBILE)=====")
            BrowserManager.close_browser(browser)

    def start_rewards(self, double_check: bool = False, write_to_desktop: bool = False):
        """
        Function used to perform the logic in order to start rewards validation
        :param write_to_desktop: Do we write a list of all the rewards that have not been validated
        :param double_check: Argument used to check for rewards twice
        """
        logging.info("=====LOGGING IN=====")
        with BrowserManager.start_bing(
                self.user, self.password, self.e, self.code, incognito=self.incognito, use_headless=self.use_headless,
                path=self.path) as browser:
            BrowserManager.goto_rewards(browser)
            logging.info("=====AUTO-REWARDS STARTED=====")
            RewardsValidator.start(browser, write_to_desktop=write_to_desktop)
            logging.info("=====AUTO-REWARDS ENDED=====")
            if double_check:
                BrowserManager.goto_rewards(browser)
                logging.info("=====AUTO-REWARDS STARTED (x2)=====")
                RewardsValidator.start(browser)
                logging.info("=====AUTO-REWARDS ENDED (x2)=====")
            BrowserManager.close_browser(browser)

    def start_desktop(self):
        """
        Function used to perform the logic in order to start desktop searches
        """
        logging.info("=====LOGGING IN=====")
        with BrowserManager.start_bing(
                self.user, self.password, self.e, self.code, incognito=self.incognito, use_headless=self.use_headless,
                path=self.path) as browser:
            logging.info("=====AUTO-SEARCH STARTED=====")
            AutoSearch.start(browser)
            logging.info("=====AUTO-SEARCH ENDED=====")
            BrowserManager.close_browser(browser)