Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions pyiceberg/view/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,12 @@ def uuid(self) -> UUID:
"""Return the view's UUID."""
return UUID(self.metadata.view_uuid)

def sql_for(self, dialect: str) -> SQLViewRepresentation:
"""Return the view representation for the sql dialect."""
return next(repr.root for repr in self.current_version().representations if repr.root.dialect == dialect)
def sql_for(self, dialect: str) -> SQLViewRepresentation | None:
"""Return the view representation for the sql dialect, or None if no representation could be resolved."""
return next(
(repr.root for repr in self.current_version().representations if repr.root.dialect.casefold() == dialect.casefold()),
None,
)

def __eq__(self, other: Any) -> bool:
"""Return the equality of two instances of the View class."""
Expand Down
10 changes: 8 additions & 2 deletions tests/test_view.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,6 +96,13 @@ def test_view_sql_for_dialect(view: View) -> None:
assert repr.sql == "SELECT * FROM prod.db.table"


def test_view_sql_for_dialect_ignore_case(view: View) -> None:
repr = view.sql_for("Spark")
assert isinstance(repr, SQLViewRepresentation)
assert repr.dialect == "spark"
assert repr.sql == "SELECT * FROM prod.db.table"


def test_view_schemas_multiple(example_view_metadata_v1_multiple_versions: dict[str, Any]) -> None:
view = View(("default", "test_view"), ViewMetadata.model_validate(example_view_metadata_v1_multiple_versions))
schemas = view.schemas()
Expand All @@ -117,5 +124,4 @@ def test_view_version_unknown_id(view: View) -> None:


def test_view_sql_for_unknown_dialect(view: View) -> None:
with pytest.raises(StopIteration):
view.sql_for("trino")
assert not view.sql_for("trino")