mirror of
https://github.com/Instagram/LibCST.git
synced 2025-12-23 10:35:53 +00:00
ca1f81f049
* Keep old exception messages (avoid breaking-changes for users relying on exception messages) * Move ``get_expected_str`` out of _exceptions.py, where it does not belong, to its own file in _parser/_parsing_check.py
25 lines
938 B
Python
25 lines
938 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
#
|
|
from typing import Type, TypeVar
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
def ensure_type(node: object, nodetype: Type[T]) -> T:
|
|
"""
|
|
Takes any python object, and a LibCST :class:`~libcst.CSTNode` subclass and
|
|
refines the type of the python object. This is most useful when you already
|
|
know that a particular object is a certain type but your type checker is not
|
|
convinced. Note that this does an instance check for you and raises an
|
|
exception if it is not the right type, so this should be used in situations
|
|
where you are sure of the type given previous checks.
|
|
"""
|
|
|
|
if not isinstance(node, nodetype):
|
|
raise ValueError(
|
|
f"Expected a {nodetype.__name__} but got a {node.__class__.__qualname__}!"
|
|
)
|
|
return node
|