Merge pull request #86 from wimglenn/issue85

User.get_stats() works even if no stars collected yet. closes #85

Author: wimglenn

aocd/models.py

@@ -110,6 +110,8 @@ def get_stats(self, years=None):
             response = requests.get(url, cookies=self.auth, headers=USER_AGENT)
             response.raise_for_status()
             soup = bs4.BeautifulSoup(response.text, "html.parser")
+            if soup.article is None and "You haven't collected any stars" in soup.main.text:
+                continue
             if soup.article.pre is None and "overall leaderboard" in soup.article.text:
                 msg = "the auth token ...{} is expired or not functioning"
                 raise DeadTokenError(msg.format(self.token[-4:]))

aocd/version.py

@@ -1 +1 @@
-__version__ = "1.2.0"
+__version__ = "1.2.1"

tests/test_models.py

@@ -228,13 +228,22 @@ def test_get_stats_when_token_expired(requests_mock):
     user = User("token12345678")
     requests_mock.get(
         url="https://adventofcode.com/2019/leaderboard/self",
-        text="<article><p>Below is the <em>Advent of Code 2021</em> overall leaderboard</p></article>"
+        text="<article><p>Below is the <em>Advent of Code 2019</em> overall leaderboard</p></article>"
     )
     expected_msg = "the auth token ...5678 is expired or not functioning"
     with pytest.raises(DeadTokenError(expected_msg)):
         user.get_stats(years=[2019])
 
 
+def test_get_stats_when_no_stars_yet(requests_mock):
+    user = User("token12345678")
+    requests_mock.get(
+        url="https://adventofcode.com/2019/leaderboard/self",
+        text="<main>You haven't collected any stars... yet.</main>"
+    )
+    assert user.get_stats(years=[2019]) == {}
+
+
 def test_get_stats_slow_user(requests_mock):
     puzzle = Puzzle(year=2019, day=25)
     requests_mock.get(