biod/study/analysis-tool/collection.py

#!/usr/bin/env python3
# Copyright 2022 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from __future__ import annotations

import glob
import pathlib
from typing import Union


class Collection:
    """Interface for the collection directory produced by the fpstudy tool."""

    def __init__(self, collection_dir: Union[pathlib.Path, str]):
        path = pathlib.Path(collection_dir)
        assert path.exists()
        self._collection_dir = path

    def discover_user_groups(self) -> list[tuple[int, str]]:
        user_groups_set: list[tuple[int, str]] = []
        for path in glob.glob(str(self._collection_dir) + "/*/*"):
            path_split = path.split("/")
            user_id, user_group = path_split[-2], path_split[-1]
            user_groups_set.append((int(user_id), str(user_group)))
        return user_groups_set