Function eval_source

tools/lib/file_sources.py:43–57

(files: dict[int, list[str] | str])

def eval_source(files: dict[int, list[str] | str]) -> dict[int, str]:
  # Returns valid file URLs given a list of possible file URLs for each segment (e.g. rlog.bz2, rlog.zst)
  valid_files: dict[int, str] = {}

  for seg_idx, urls in files.items():
    if isinstance(urls, str):
      urls = [urls]

    # Add first valid file URL
    for url in urls:
      if file_exists(url):
        valid_files[seg_idx] = url
        break

  return valid_files