Use final part of url for name

HCookie · Oct 16, 2024 · 22c6cf6 · 22c6cf6
1 parent 484ecf2
commit 22c6cf6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/contributors/contributors.py b/contributors/contributors.py
@@ -184,7 +184,7 @@ def get_contributors(
                 contribution_count=user.contributions_count,
                 commit_url=commit_url,
                 sponsor_info="",
-                organisations=list(map(lambda x: x.url, user.organizations())),
+                organisations=list(map(lambda x: x.url.split('/')[-1], user.organizations())),
             )
             contributors.append(contributor)
     except Exception as e:

diff --git a/contributors/env.py b/contributors/env.py
@@ -73,7 +73,7 @@ def validate_date_format(env_var_name: str) -> str:
 def get_env_vars(
     test: bool = False,
 ) -> tuple[
-    str | None, list[str], int | None, int | None, bytes, str, str, str, str, bool, bool
+    str | None, list[str], int | None, int | None, bytes, str, str, str, str, bool, bool, list
 ]:
     """
     Get the environment variables for use in the action.
@@ -93,6 +93,7 @@ def get_env_vars(
         str: the end date to get contributor information to.
         str: whether to get sponsor information on the contributor
         str: whether to link username to Github profile in markdown output
+        list: organisations to show
 
     """