diff --git a/setup.py b/setup.py index b35c97b..0cdf9af 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ setuptools.setup( name="vanguard-api", - version="0.2.1", + version="0.2.2", author="MaxxRK", author_email="maxxrk@pm.me", description="An unofficial API for Vanguard Invest", @@ -13,7 +13,7 @@ long_description_content_type="text/markdown", license="MIT", url="https://github.com/MaxxRK/vanguard-api", - download_url="https://github.com/MaxxRK/vanguard-api/archive/refs/tags/v0.2.0.tar.gz", + download_url="https://github.com/MaxxRK/vanguard-api/archive/refs/tags/v0.2.2.tar.gz", keywords=["VANGUARD", "API"], install_requires=["playwright", "playwright-stealth"], packages=["vanguard"], diff --git a/vanguard/session.py b/vanguard/session.py index 05bef4d..460a98f 100644 --- a/vanguard/session.py +++ b/vanguard/session.py @@ -197,9 +197,14 @@ def login(self, username, password, last_four): """ try: self.password = password - self.go_url(login_page()) + try: + self.go_url(login_page()) + except PlaywrightTimeoutError: + self.close_browser() + raise Exception("Failed to load login page.") login_state = self.find_login_state() if login_state == 0: + self.close_browser() raise Exception("Failed to find login state") elif login_state == 1: return False