forked from STI/Astroport.ONE
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from typing import Any, NamedTuple, TYPE_CHECKING
|
|
|
|
try:
|
|
from typing import TypedDict
|
|
except ImportError: # Python < 3.8
|
|
from typing_extensions import TypedDict
|
|
|
|
if TYPE_CHECKING:
|
|
from .source import Source # noqa: F401
|
|
|
|
__all__ = ["get_location", "SourceLocation", "FormattedSourceLocation"]
|
|
|
|
|
|
class FormattedSourceLocation(TypedDict):
|
|
"""Formatted source location"""
|
|
|
|
line: int
|
|
column: int
|
|
|
|
|
|
class SourceLocation(NamedTuple):
|
|
"""Represents a location in a Source."""
|
|
|
|
line: int
|
|
column: int
|
|
|
|
@property
|
|
def formatted(self) -> FormattedSourceLocation:
|
|
return dict(line=self.line, column=self.column)
|
|
|
|
def __eq__(self, other: Any) -> bool:
|
|
if isinstance(other, dict):
|
|
return self.formatted == other
|
|
return tuple(self) == other
|
|
|
|
def __ne__(self, other: Any) -> bool:
|
|
return not self == other
|
|
|
|
|
|
def get_location(source: "Source", position: int) -> SourceLocation:
|
|
"""Get the line and column for a character position in the source.
|
|
|
|
Takes a Source and a UTF-8 character offset, and returns the corresponding line and
|
|
column as a SourceLocation.
|
|
"""
|
|
return source.get_location(position)
|