refactor(utils): leverage strtobool

Change-Id: Ibff24d023356bdc10a1a8347295d534c745c4d2d
Mergifyio · Dec 17, 2024 · 3d2cd3d · 3d2cd3d
1 parent 6d4f2bb
commit 3d2cd3d
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/pytest_mergify/utils.py b/pytest_mergify/utils.py
@@ -3,12 +3,17 @@
 
 CIProviderT = typing.Literal["github_actions", "circleci"]
 
+SUPPORTED_CIs = {
+    "GITHUB_ACTIONS": "github_actions",
+    "CIRCLECI": "circleci",
+}
+
 
 def get_ci_provider() -> CIProviderT | None:
-    if os.getenv("GITHUB_ACTIONS") == "true":
-        return "github_actions"
-    if os.getenv("CIRCLECI") == "true":
-        return "circleci"
+    for envvar, name in SUPPORTED_CIs.items():
+        if envvar in os.environ and strtobool(os.environ[envvar]):
+            return name
+
     return None