Skip to content

Add Error Handling for WebDriver Exceptions #16

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
SoClosee wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Add Error Handling for WebDriver Exceptions #16

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
95 changes: 58 additions & 37 deletions main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -196,41 +196,50 @@ def _handle_signal(self, signum, frame):
# Browser setup
# ------------------------------------------------------------------
def start_browser(self):
"""Initialize Chrome WebDriver."""
options = webdriver.ChromeOptions()
options.add_experimental_option("excludeSwitches", ["enable-logging"])
options.add_argument("--disable-notifications")
if self.headless:
options.add_argument("--headless=new")

if ChromeDriverManager is not None:
service = Service(ChromeDriverManager().install())
else:
service = Service()

self.driver = webdriver.Chrome(service=service, options=options)
self.driver.maximize_window()
logger.info("Browser started.")
"""Initialize Chrome WebDriver with error handling."""
try:
options = webdriver.ChromeOptions()
options.add_experimental_option("excludeSwitches", ["enable-logging"])
options.add_argument("--disable-notifications")
if self.headless:
options.add_argument("--headless=new")

if ChromeDriverManager is not None:
service = Service(ChromeDriverManager().install())
else:
service = Service()

self.driver = webdriver.Chrome(service=service, options=options)
self.driver.maximize_window()
logger.info("Browser started.")
except WebDriverException as e:
logger.error(f"Failed to start browser: {e}")
raise

def navigate_to_facebook(self):
"""Open Facebook so the user can log in."""
self.driver.get("https://www.facebook.com/")
logger.info("Navigated to Facebook. Please log in manually in the browser.")
"""Open Facebook with error handling."""
try:
self.driver.get("https://www.facebook.com/")
logger.info("Navigated to Facebook. Please log in manually in the browser.")
except WebDriverException as e:
logger.error(f"Failed to navigate to Facebook: {e}")
raise

def quit(self):
"""Close the browser cleanly."""
"""Close the browser cleanly with error handling."""
if self.driver:
try:
self.driver.quit()
except WebDriverException:
pass
logger.info("Browser closed.")
except WebDriverException as e:
logger.error(f"Failed to close browser: {e}")
finally:
logger.info("Browser closed.")

# ------------------------------------------------------------------
# Invite workflow
# ------------------------------------------------------------------
def _click_invite_button(self):
"""Find and click the main 'Invite' button on the group page."""
"""Find and click the main 'Invite' button on the group page with error handling."""
def finder(soup):
buttons = soup.find_all("div", attrs={"role": "button"})
for btn in buttons:
Expand All @@ -239,34 +248,46 @@ def finder(soup):
return btn
return None

el = find_element_with_retry(self.driver, finder, click=True)
if el is None:
raise RuntimeError("Could not find the 'Invite' button on the group page.")
logger.info("Clicked the 'Invite' button.")
try:
el = find_element_with_retry(self.driver, finder, click=True)
if el is None:
raise RuntimeError("Could not find the 'Invite' button on the group page.")
logger.info("Clicked the 'Invite' button.")
except WebDriverException as e:
logger.error(f"Failed to click invite button: {e}")
raise

def _click_invite_friends_menu(self):
"""Click the 'Invite Facebook friends' menu item."""
"""Click the 'Invite Facebook friends' menu item with error handling."""
def finder(soup):
items = soup.find_all("div", attrs={"role": "menuitem"})
for item in items:
if self.labels["invite_friends_menu"] in item.get_text():
return item
return None

el = find_element_with_retry(self.driver, finder, click=True)
if el is None:
raise RuntimeError("Could not find the 'Invite Facebook friends' menu item.")
logger.info("Clicked 'Invite Facebook friends' menu.")
try:
el = find_element_with_retry(self.driver, finder, click=True)
if el is None:
raise RuntimeError("Could not find the 'Invite Facebook friends' menu item.")
logger.info("Clicked 'Invite Facebook friends' menu.")
except WebDriverException as e:
logger.error(f"Failed to click invite friends menu: {e}")
raise

def _wait_for_dialog(self):
"""Wait for the invitation dialog to appear."""
"""Wait for the invitation dialog to appear with error handling."""
def finder(soup):
return soup.find("div", attrs={"aria-label": self.labels["invite_dialog"]})

el = find_element_with_retry(self.driver, finder, click=False)
if el is None:
raise RuntimeError("Invitation dialog did not appear.")
logger.info("Invitation dialog is open.")
try:
el = find_element_with_retry(self.driver, finder, click=False)
if el is None:
raise RuntimeError("Invitation dialog did not appear.")
logger.info("Invitation dialog is open.")
except WebDriverException as e:
logger.error(f"Failed to wait for invitation dialog: {e}")
raise

def _select_friends(self, target_count):
"""Select up to *target_count* friends from the dialog."""
Expand Down