feat: add follower count to calculation

devfle · Jan 2, 2023 · 743a67b · 743a67b
1 parent 00d7f9e
commit 743a67b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/graphql_query.py b/graphql_query.py
@@ -9,6 +9,9 @@
                             totalContributions
                         }}
                     }}
+                    followers {{
+                        totalCount
+                    }}
                 }}
             }}
         """
diff --git a/main.py b/main.py
@@ -40,8 +40,10 @@ def draw_progress_bar(current_progress: float | int) -> str:
                                         "percentage_level")(readme_instance.calc_current_level())
 
     CONTRIBUTION_EP = readme_instance.contribution_ep
+    FOLLOWER_EP = readme_instance.follower_ep
     # should be generated in later versions
-    ep_information = f"<pre>💪 1x contribute → { CONTRIBUTION_EP } experience points</pre>\n"
+    ep_information = (f"<pre>💪 1x contribute → { CONTRIBUTION_EP } experience points\n"
+                      f"🌟 1x follower → { FOLLOWER_EP } experience points</pre>\n")
 
     readme_path: str = getenv("INPUT_README_PATH")
     start_section: str = "<!--README_LEVEL_UP:START-->"

diff --git a/readme_level.py b/readme_level.py
@@ -24,7 +24,7 @@ class ReadmeLevel:
     project_ep: int = 5
     discussion_ep: int = 10
     star_ep: int = 40
-    follower_ep: int = 50
+    follower_ep: int = 25
 
     def __init__(self) -> None:
         pass
@@ -42,9 +42,13 @@ def fetch_user_data(self) -> dict[str, int] | None:
 
         if response.status_code == 200:
             response_data = response.json()
+
             user_data = (response_data["data"]["user"]
                          ["contributionsCollection"]["contributionCalendar"])
 
+            user_data["totalFollowers"] = (response_data["data"]["user"]
+                         ["followers"]["totalCount"])
+
             return user_data
 
         error("request to github api failed")
@@ -61,7 +65,7 @@ def _update_user_data(self) -> None:
         key_mapper = {
             "totalContributions": "contribution_count",
             "projects": "project_count",
-            "follower": "follower_count",
+            "totalFollowers": "follower_count",
             "discussions": "discussion_count"
         }