# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import sys
from contextlib import contextmanager
from pathlib import Path

MODULE_DIR = Path(__file__).parent.resolve()
GEN_DIR = MODULE_DIR / "gen"

@contextmanager
def protoc_in_sys_path():
  prev_path = sys.path
  sys.path = [str(GEN_DIR)] + prev_path
  try:
    yield None
  finally:
    sys.path = prev_path