CircleCI: Handle null vcs.provider_name

con · Jun 12, 2024 · 97fcd40 · 97fcd40
1 parent 9ca71bc
commit 97fcd40
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/tinuous/circleci.py b/src/tinuous/circleci.py
@@ -96,7 +96,7 @@ def get_build_assets(
         if self.until is not None:
             log.info("Skipping pipelines newer than %s", self.until)
         for pipeline in self.get_pipelines():
-            if pipeline.vcs is None or pipeline.vcs.provider_name.lower() != "github":
+            if pipeline.vcs is None or not pipeline.vcs.is_github():
                 log.debug(
                     "Skipping pipeline %d as it is not associated with GitHub",
                     pipeline.number,
@@ -155,9 +155,11 @@ def get_build_assets(
                                             job=job.job_number,
                                             job_id=job.id,
                                             job_name=sanitize_pathname(job.name),
-                                            step=str(action.step)
-                                            if action.step is not None
-                                            else "UNK",
+                                            step=(
+                                                str(action.step)
+                                                if action.step is not None
+                                                else "UNK"
+                                            ),
                                             step_name=sanitize_pathname(step.name),
                                             index=action.index,
                                         )
@@ -360,7 +362,7 @@ class Trigger(BaseModel):
 
 
 class VCS(BaseModel):
-    provider_name: str
+    provider_name: Optional[str]
     # target_repository_url: str
     branch: Optional[str] = None
     # review_id: Optional[str] = None
@@ -370,6 +372,9 @@ class VCS(BaseModel):
     # commit: Optional[Commit] = None
     # origin_repository_url: str
 
+    def is_github(self) -> bool:
+        return self.provider_name is not None and self.provider_name.lower() == "github"
+
 
 class Pipeline(BaseModel):
     id: str