Add ability to exclude languages from images

Address #9
jstrieb · Dec 17, 2020 · ad1f714 · ad1f714
1 parent 3d83a89
commit ad1f714
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/generate_images.py b/generate_images.py
@@ -99,10 +99,15 @@ async def main() -> None:
         # access_token = os.getenv("GITHUB_TOKEN")
         raise Exception("A personal access token is required to proceed!")
     user = os.getenv("GITHUB_ACTOR")
-    excluded = os.getenv("EXCLUDED")
-    excluded = {x.strip() for x in excluded.split(",")} if excluded else None
+    exclude_repos = os.getenv("EXCLUDED")
+    exclude_repos = ({x.strip() for x in exclude_repos.split(",")}
+                     if exclude_repos else None)
+    exclude_langs = os.getenv("EXCLUDED_LANGS")
+    exclude_langs = ({x.strip() for x in exclude_langs.split(",")}
+                     if exclude_langs else None)
     async with aiohttp.ClientSession() as session:
-        s = Stats(user, access_token, session, exclude_repos=excluded)
+        s = Stats(user, access_token, session, exclude_repos=exclude_repos,
+                  exclude_langs=exclude_langs)
         await asyncio.gather(generate_languages(s), generate_overview(s))
 
 

diff --git a/github_stats.py b/github_stats.py
@@ -231,9 +231,11 @@ class Stats(object):
     """
     def __init__(self, username: str, access_token: str,
                  session: aiohttp.ClientSession,
-                 exclude_repos: Optional[Set] = None):
+                 exclude_repos: Optional[Set] = None,
+                 exclude_langs: Optional[Set] = None):
         self.username = username
         self._exclude_repos = set() if exclude_repos is None else exclude_repos
+        self._exclude_langs = set() if exclude_langs is None else exclude_langs
         self.queries = Queries(username, access_token, session)
 
         self._name = None
@@ -316,6 +318,7 @@ async def get_stats(self) -> None:
                 for lang in repo.get("languages", {}).get("edges", []):
                     name = lang.get("node", {}).get("name", "Other")
                     languages = await self.languages
+                    if name in self._exclude_langs: continue
                     if name in languages:
                         languages[name]["size"] += lang.get("size", 0)
                         languages[name]["occurrences"] += 1