sqlframe 3.33.1__tar.gz → 3.34.0__tar.gz

{sqlframe-3.33.1 → sqlframe-3.34.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 3.33.1
+Version: 3.34.0
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman

{sqlframe-3.33.1 → sqlframe-3.34.0}/setup.py

@@ -20,7 +20,7 @@ setup(
     python_requires=">=3.9",
     install_requires=[
         "prettytable<4",
-        "sqlglot>=24.0.0,<26.20",
+        "sqlglot>=24.0.0,<26.22",
         "typing_extensions",
     ],
     extras_require={
@@ -41,7 +41,7 @@ setup(
             "pytest>=8.2.0,<8.4",
             "pytest-forked",
             "pytest-postgresql>=6,<8",
-            "pytest-xdist>=3.6,<3.7",
+            "pytest-xdist>=3.6,<3.8",
             "pre-commit>=3.7,<5",
             "ruff>=0.4.4,<0.12",
             "types-psycopg2>=2.9,<3",

{sqlframe-3.33.1 → sqlframe-3.34.0}/sqlframe/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '3.33.1'
-__version_tuple__ = version_tuple = (3, 33, 1)
+__version__ = version = '3.34.0'
+__version_tuple__ = version_tuple = (3, 34, 0)

{sqlframe-3.33.1 → sqlframe-3.34.0}/sqlframe/base/column.py

@@ -128,6 +128,39 @@ class Column:
             "Tried to call a column which is unexpected. Did you mean to call a method on a DataFrame? If so, make sure the method is typed correctly and is supported. If not, please open an issue requesting support: https://github.com/eakmanrq/sqlframe/issues"
         )
 
+    def __getitem__(self, key: t.Any) -> Column:
+        """
+        An expression that gets an item at position ``ordinal`` out of a list,
+        or gets an item by key out of a dict.
+
+        .. versionadded:: 1.3.0
+
+        .. versionchanged:: 3.4.0
+            Supports Spark Connect.
+
+        Parameters
+        ----------
+        k
+            a literal value, or a slice object without step.
+
+        Returns
+        -------
+        :class:`Column`
+            Column representing the item got by key out of a dict, or substrings sliced by
+            the given slice object.
+
+        Examples
+        --------
+        >>> df = spark.createDataFrame([('abcedfg', {"key": "value"})], ["l", "d"])
+        >>> df.select(df.l[slice(1, 3)], df.d['key']).show()
+        +---------------+------+
+        |substr(l, 1, 3)|d[key]|
+        +---------------+------+
+        |            abc| value|
+        +---------------+------+
+        """
+        return self.getItem(key)
+
     def __getattr__(self, name: str) -> Column:
         """
         Enables accessing nested fields using dot notation for struct types.

{sqlframe-3.33.1 → sqlframe-3.34.0}/sqlframe.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 3.33.1
+Version: 3.34.0
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman

{sqlframe-3.33.1 → sqlframe-3.34.0}/sqlframe.egg-info/requires.txt

@@ -1,5 +1,5 @@
 prettytable<4
-sqlglot<26.20,>=24.0.0
+sqlglot<26.22,>=24.0.0
 typing_extensions
 
 [bigquery]
@@ -22,7 +22,7 @@ pyarrow<21,>=10
 pyspark<3.6,>=2
 pytest-forked
 pytest-postgresql<8,>=6
-pytest-xdist<3.7,>=3.6
+pytest-xdist<3.8,>=3.6
 pytest<8.4,>=8.2.0
 ruff<0.12,>=0.4.4
 types-psycopg2<3,>=2.9

{sqlframe-3.33.1 → sqlframe-3.34.0}/tests/integration/engines/test_engine_column.py

@@ -37,6 +37,7 @@ def test_column_get_item_array(get_session: t.Callable[[], _BaseSession], get_fu
     session = get_session()
     lit = get_func("lit", session)
     assert session.range(1).select(lit(["a", "b", "c"]).getItem(0).alias("value")).first()[0] == "a"
+    assert session.range(1).select(lit(["a", "b", "c"])[0].alias("value")).first()[0] == "a"
 
 
 def test_column_get_item_map(get_session: t.Callable[[], _BaseSession], get_func):
@@ -44,6 +45,7 @@ def test_column_get_item_map(get_session: t.Callable[[], _BaseSession], get_func
     lit = get_func("lit", session)
     if not isinstance(session, (PostgresSession, BigQuerySession)):
         assert session.range(1).select(lit({"key": "value"}).getItem("key")).first()[0] == "value"
+        assert session.range(1).select(lit({"key": "value"})["key"]).first()[0] == "value"
 
 
 def test_column_get_field_struct(get_session: t.Callable[[], _BaseSession]):

{sqlframe-3.33.1 → sqlframe-3.34.0}/tests/unit/standalone/test_column.py

@@ -230,5 +230,16 @@ def test_get_item():
     assert F.col("cola").getItem(1).sql() == "ELEMENT_AT(cola, (1 + 1)) AS element_at__cola__"
 
 
+def test_get_item_with_array_syntax():
+    assert F.col("cola")[1].sql() == "ELEMENT_AT(cola, (1 + 1)) AS element_at__cola__"
+
+
+def test_get_item_by_key_with_array_syntax():
+    assert (
+        F.lit({"key": "value"})["key"].sql()
+        == "ELEMENT_AT(MAP('key', 'value'), 'key') AS element_at__key__"
+    )
+
+
 def test_contains():
     assert F.col("cola").contains("foo").sql() == "CONTAINS(cola, 'foo')"