Skip to content

Commit d2b275d

Browse files
committed
obtain repo/pr from task or link2pr OR raise ValueError
1 parent 28a4745 commit d2b275d

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

scripts/automated_ingestion/eessi_task.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,16 @@ def _find_state(self) -> TaskState:
147147
"""
148148
# obtain repo and pr from metadata
149149
log_message(LoggingScope.TASK_OPS, 'INFO', "finding state of task %s", self.description.task_object)
150-
repo = self.description.metadata['task']['repo']
151-
pr = self.description.metadata['task']['pr']
150+
task = self.description.metadata['task'] if 'task' in self.description.metadata else None
151+
link2pr = self.description.metadata['link2pr'] if 'link2pr' in self.description.metadata else None
152+
if task:
153+
repo = task['repo']
154+
pr = task['pr']
155+
elif link2pr:
156+
repo = link2pr['repo']
157+
pr = link2pr['pr']
158+
else:
159+
raise ValueError("no repo or pr found in metadata")
152160
log_message(LoggingScope.TASK_OPS, 'INFO', "repo: %s, pr: %s", repo, pr)
153161

154162
# iterate over all sequence numbers in repo/pr dir

0 commit comments

Comments
 (0)