Merge pull request #1 from fabricio-aguiar/refac

Refactoring the code to become more readable

Author: fao89

README.md

@@ -18,4 +18,4 @@ This bot launches the chromedriver, waits the user login into LinkedIn account.
 
 #### Installing ChromeDriver:
 * `sudo apt-get install chromium-chromedriver`
-* `export PATH=$PATH:/usr/lib/chromium-browser/`
+* `echo  'export PATH=$PATH:/usr/lib/chromium-browser/' >> ~/.bashrc`

easyapplybot.py

@@ -5,173 +5,227 @@
 from bs4 import BeautifulSoup
 import pyautogui
 from tkinter import filedialog, Tk
+import os
 
-APPLICATIONS = 30
 
+class EasyApplyBot:
 
-def getJobLinks(page):
-    links = []
-    for link in page.find_all('a'):
-        url = link.get('href')
+    MAX_APPLICATIONS = 30
 
-        if url:
-            if '/jobs/view' in url:
-                links.append(url)
-    return links
+    def __init__(self, language):
+        self.language = language
+        self.options = self.browser_options()
+        self.browser = webdriver.Chrome(chrome_options=self.options)
+        self.start_linkedin()
 
+    def browser_options(self):
+        options = Options()
+        options.add_argument("--start-maximized")
+        options.add_argument("--ignore-certificate-errors")
+        options.add_argument("user-agent=Chrome/51.0.2704.79 Safari/537.36 Edge/14.14393")
+        return options
 
-def getEasy(page):
-    achou = page.find("button", class_="jobs-s-apply__button js-apply-button")
-    return str(achou)
+    def start_linkedin(self):
+        self.browser.get("https://linkedin.com/uas/login")
 
+    def wait_for_login(self):
+        if language == "en":
+            title = "Sign In to LinkedIn"
+        elif language == "pt":
+            title = "Entrar no LinkedIn"
 
-def loadPage(browser):
-    tpage = 0
-    while tpage < 4000:
-        browser.execute_script("window.scrollTo(0,"+str(tpage)+" );")
-        tpage += 200
         time.sleep(1)
-    return BeautifulSoup(browser.page_source, "lxml")
-
-
-def avoidLock():
-    x, y = pyautogui.position()
-    pyautogui.moveTo(x+200, None, duration=1.0)
-    pyautogui.moveTo(x, None, duration=0.5)
-    pyautogui.keyDown('ctrl')
-    pyautogui.press('esc')
-    pyautogui.keyUp('ctrl')
-    time.sleep(0.5)
-    pyautogui.press('esc')
-
-
-def containEasy(browser, cargo, local, pp):
-    browser.get(
-        "https://www.linkedin.com/jobs/search/?f_LF=f_AL&keywords=" +
-        cargo + local + "&start="+str(pp))
-    avoidLock()
-    loadPage(browser)
-    return (browser, pp)
-
-
-def ViewBot(browser):
-    count = 0
-    count5 = 0
-    pp = 0
-    browser.set_window_size(0, 0)
-    aux = input('Enter the desired job title : ')
-    cargo = aux.replace(" ", "%20")
-    print("Location codes: \n[1] GLOBAL\n[2] Country\n[3] State\n[4] City")
-    aux = input('Enter the location code: ')
-    if aux == "1":
-        aux2 = "Worldwide"
-    if aux == "2":
-        aux2 = input('Enter the country name: ')
-    if aux == "3":
-        aux2 = input('Enter the state name: ')
-    if aux == "4":
-        aux2 = input('Enter the city name: ')
-    local = "&location=" + aux2.replace(" ", "%20") + "&sortBy=DD"
-    print("\nPlease select your curriculum\n")
-    time.sleep(1)
-    root = Tk()
-    resumeloctn = filedialog.askopenfilename(parent=root, initialdir="/",
-                                             title='Please select your curriculum')
-
-    root.destroy()
-
-    browser, pp = containEasy(browser, cargo, local, pp)
-
-    while count < APPLICATIONS:
-        # sleep to make sure everything loads, add random to make us look human.
-        time.sleep(random.uniform(3.5, 6.9))
-
-        page = BeautifulSoup(browser.page_source, 'lxml')
-
-        jobs = getJobLinks(page)
-        if jobs:
-                for job in jobs:
-                    root = 'http://www.linkedin.com'
-                    roots = 'https://www.linkedin.com'
-                    if root not in job or roots not in job:
-                        job = 'https://www.linkedin.com'+job
-                    browser.get(job)
-                    time.sleep(random.uniform(3.5, 4.5))
-                    print("\n" + str(count5 + pp + 1))
-                    pagina = BeautifulSoup(browser.page_source, "lxml")
-
-                    tej = getEasy(pagina)
-                    ember = pagina.find("div",
-                                        class_="jobs-s-apply--top-card jobs-s-apply--fadein inline-flex mr2 jobs-s-apply ember-view")
-                    count5 += 1
-
-                    if len(tej) > 4:
-                        tej = "* Easy Apply Button"
-                        embert = str(ember)
-                        list_of_words = embert.split()
-                        next_word = [word for word in list_of_words if "ember" in word and "id" in word]
-                        pember = next_word[0][:-1]
-                        xpath = '//*[@'+pember+']/button'
-                        time.sleep(1.5)
-                        triggerDropDown = browser.find_element_by_xpath(xpath)
-                        time.sleep(0.5)
-                        triggerDropDown.click()
-                        time.sleep(1.5)
-                        browser.find_element_by_xpath('//*[@id="file-browse-input"]').send_keys(resumeloctn)
-                        time.sleep(random.uniform(10, 14))
-                        try:
-                            browser.find_element_by_xpath("//*[contains(text(), 'Submit application')]").click() 
-                        except:
-                            browser.find_element_by_xpath("//*[contains(text(), 'Enviar candidatura')]").click() 
-                        time.sleep(random.uniform(2, 3.5))
-                        count += 1
-
-                    else:
-                        tej = "* NOT Easy Apply Button"
-
-                    try:
-                        print("-> "+browser.title+"\n"+tej+" - Checked! \n")
-                    except:
-                        pass
-
-                    if count5 == len(jobs):
-                        pp = pp + 25
-                        count5 = 0
-                        print("Going to next jobs page !")
-                        avoidLock()
-                        browser, pp = containEasy(browser, cargo, local, pp)
-        else:
-            print("I'm Lost Exiting")
-            break
-
-
-def Main():
-    print("\n\n       ===== Please Login to your LinkedIn account =====\n\n")
-    time.sleep(3)
-
-    options = Options()
-    options.add_argument("--start-maximized")
-    options.add_argument("--ignore-certificate-errors")
-    options.add_argument("user-agent=Chrome/51.0.2704.79 Safari/537.36 Edge/14.14393") 
-    browser = webdriver.Chrome(chrome_options=options)
-    browser.get("https://linkedin.com/uas/login")
-    time.sleep(8)
-
-    if "Sign In" in browser.title:
-        message = "Sign In to LinkedIn"
-    elif "Entrar" in browser.title:
-        message = "Entrar no LinkedIn"
-
-    while True:
-        if browser.title != message:
-            print("\nStarting LinkedIn bot\n")
-            break
-        else:
-            time.sleep(2)
-            print("\nPlease Login to your LinkedIn account\n")
-
-    ViewBot(browser)
-    browser.close()
+
+        while True:
+            if self.browser.title != title:
+                print("\nStarting LinkedIn bot\n")
+                break
+            else:
+                time.sleep(1)
+                print("\nPlease Login to your LinkedIn account\n")
+
+    def fill_data(self):
+        self.browser.set_window_size(0, 0)
+        self.browser.set_window_position(2000, 2000)
+        os.system("reset")
+
+        position = input('Enter the desired job title : ')
+        self.position = position.replace(" ", "%20")
+
+        print("Location codes: \n[1] GLOBAL\n[2] Country\n[3] State\n[4] City")
+        location_code = input('Enter the location code: ')
+        if location_code == "1":
+            location = "Worldwide"
+        elif location_code == "2":
+            location = input('Enter the country name: ')
+        elif location_code == "3":
+            location = input('Enter the state name: ')
+        elif location_code == "4":
+            location = input('Enter the city name: ')
+        self.location = "&location=" + location.replace(" ", "%20") + "&sortBy=DD"
+
+        print("\nPlease select your curriculum\n")
+        time.sleep(1)
+        root = Tk()
+        self.resumeloctn = filedialog.askopenfilename(parent=root, initialdir="/",
+                                                      title='Please select your curriculum')
+
+        root.destroy()
+
+    def start_apply(self):
+        self.wait_for_login()
+        self.fill_data()
+        self.applications_loop()
+
+    def applications_loop(self):
+        count_application = 0
+        count_job = 0
+        jobs_per_page = 0
+
+        os.system("reset")
+
+        self.browser.set_window_position(0, 0)
+        self.browser.maximize_window()
+        self.browser, _ = self.next_jobs_page(jobs_per_page)
+
+        while count_application < self.MAX_APPLICATIONS:
+            # sleep to make sure everything loads, add random to make us look human.
+            time.sleep(random.uniform(3.5, 6.9))
+
+            page = BeautifulSoup(self.browser.page_source, 'lxml')
+
+            jobs = self.get_job_links(page)
+
+            if not jobs:
+                print("Jobs not found")
+                break
+
+            for job in jobs:
+                count_job += 1
+                job_page = self.get_job_page(job)
+
+                if self.got_easy_apply(job_page):
+                    string_easy = "* has Easy Apply Button"
+                    xpath = self.easy_apply_xpath()
+                    self.click_button(xpath)
+                    self.send_resume()
+                    count_application += 1
+
+                else:
+                    string_easy = "* Doesn't have Easy Apply Button"
+
+                position_number = str(count_job + jobs_per_page)
+                print(f"\nPosition {position_number}:\n {self.browser.title} \n {string_easy} \n")
+
+                if count_job == len(jobs):
+                    jobs_per_page = jobs_per_page + 25
+                    count_job = 0
+                    print("Going to next jobs page !")
+                    self.avoid_lock()
+                    self.browser, jobs_per_page = self.next_jobs_page(jobs_per_page)
+
+        self.finish_apply()
+
+    def get_job_links(self, page):
+        links = []
+        for link in page.find_all('a'):
+            url = link.get('href')
+
+            if url:
+                if '/jobs/view' in url:
+                    links.append(url)
+        return set(links)
+
+    def get_job_page(self, job):
+        root = 'www.linkedin.com'
+        if root not in job:
+            job = 'https://www.linkedin.com'+job
+        self.browser.get(job)
+        self.job_page = self.load_page(sleep=0.5)
+        return self.job_page
+
+    def got_easy_apply(self, page):
+        button = page.find("button", class_="jobs-s-apply__button js-apply-button")
+        return len(str(button)) > 4
+
+    def get_easy_apply_button(self):
+        button_class = "jobs-s-apply--top-card jobs-s-apply--fadein inline-flex mr2 jobs-s-apply ember-view"
+        button = self.job_page.find("div", class_=button_class)
+        return button
+
+    def easy_apply_xpath(self):
+        button = self.get_easy_apply_button()
+        button_inner_html = str(button)
+        list_of_words = button_inner_html.split()
+        next_word = [word for word in list_of_words if "ember" in word and "id" in word]
+        ember = next_word[0][:-1]
+        xpath = '//*[@'+ember+']/button'
+        return xpath
+
+    def click_button(self, xpath):
+        triggerDropDown = self.browser.find_element_by_xpath(xpath)
+        time.sleep(0.5)
+        triggerDropDown.click()
+        time.sleep(1.5)
+
+    def send_resume(self):
+        self.browser.find_element_by_xpath('//*[@id="file-browse-input"]').send_keys(self.resumeloctn)
+        submit_button = None
+        time.sleep(1)
+        while not submit_button:
+            if language == "en":
+                submit_button = self.browser.find_element_by_xpath("//*[contains(text(), 'Submit application')]")
+            elif language == "pt":
+                submit_button = self.browser.find_element_by_xpath("//*[contains(text(), 'Enviar candidatura')]")
+        submit_button.click()
+        time.sleep(random.uniform(1.5, 2.5))
+
+    def load_page(self, sleep=1):
+        scroll_page = 0
+        while scroll_page < 4000:
+            self.browser.execute_script("window.scrollTo(0,"+str(scroll_page)+" );")
+            scroll_page += 200
+            time.sleep(sleep)
+
+        if sleep != 1:
+            self.browser.execute_script("window.scrollTo(0,0);")
+            time.sleep(sleep * 3)
+
+        page = BeautifulSoup(self.browser.page_source, "lxml")
+        return page
+
+    def avoid_lock(self):
+        x, _ = pyautogui.position()
+        pyautogui.moveTo(x+200, None, duration=1.0)
+        pyautogui.moveTo(x, None, duration=0.5)
+        pyautogui.keyDown('ctrl')
+        pyautogui.press('esc')
+        pyautogui.keyUp('ctrl')
+        time.sleep(0.5)
+        pyautogui.press('esc')
+
+    def next_jobs_page(self, jobs_per_page):
+        self.browser.get(
+            "https://www.linkedin.com/jobs/search/?f_LF=f_AL&keywords=" +
+            self.position + self.location + "&start="+str(jobs_per_page))
+        self.avoid_lock()
+        self.load_page()
+        return (self.browser, jobs_per_page)
+
+    def finish_apply(self):
+        self.browser.close()
+
 
 if __name__ == '__main__':
-    Main()
+
+    print("\nEasy Apply Bot\n")
+    languages = ["pt", "en"]
+    language = None
+
+    while language not in languages:
+        language = input("LinkedIn language [pt/en]: ")
+        if language not in languages:
+            print("Please, enter a valid language!\n")
+
+    bot = EasyApplyBot(language)
+    bot.start_apply()