| # Copyright 2021 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """File interaction utilities.""" |
| |
| import contextlib |
| from pathlib import Path |
| from typing import TextIO, TYPE_CHECKING, Union |
| |
| if TYPE_CHECKING: |
| import os |
| |
| |
| @contextlib.contextmanager |
| def Open(obj: Union[str, 'os.PathLike', TextIO], mode: str = 'r', **kwargs): |
| """Convenience ctx that accepts a file path or an already open file object.""" |
| if isinstance(obj, str): |
| with open(obj, mode=mode, **kwargs) as f: |
| yield f |
| elif isinstance(obj, Path): |
| yield obj.open(mode=mode, **kwargs) |
| else: |
| yield obj |
