blob: 1915c7e6fc770bae1275b326d09a7eb56c35db4e [file] [log] [blame]
# 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
import os
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, **kwargs)
yield obj