Feat/polars integration (#87)

* feat(polars): add basic feature

* refactor(test): merge polars and pandas

Makefile

@@ -32,7 +32,7 @@ dev-setup: install-test-requirements install-doc-requirements
 	pre-commit install
 
 dev-install:
-	maturin develop -E pandas
+	maturin develop -E pandas,polars
 
 prod-install:
 	./prod_install.sh

prod_install.sh

@@ -1,4 +1,4 @@
 #!/bin/bash -xe
 rm -rf target/wheels/
 maturin build --release
-pip install --force-reinstall "$(echo target/wheels/*.whl)[pandas]"
+pip install --force-reinstall "$(echo target/wheels/*.whl)[pandas, polars]"

pyproject.toml

@@ -15,6 +15,8 @@ dependencies = [
 
 [project.optional-dependencies]
 pandas = ["pandas>=1.4.4,<1.6"]
+polars = ["polars>=0.16.14,<0.17"]
+
 
 [project.urls]
 "Source Code" = "https://github.com/ToucanToco/fastexcel"

python/fastexcel/__init__.py

@@ -4,6 +4,7 @@
     import pandas as pd
 
 import pyarrow as pa
+import polars as pl
 
 from ._fastexcel import __version__, _ExcelReader, _ExcelSheet
 from ._fastexcel import read_excel as _read_excel
@@ -47,6 +48,15 @@ def to_pandas(self) -> "pd.DataFrame":
         # We know for sure that the sheet will yield exactly one RecordBatch
         return self.to_arrow().to_pandas()
 
+    def to_polars(self) -> pl.DataFrame | pl.Series:
+        """Converts the sheet to a Polars `DataFrame`.
+
+        Requires the `polars` extra to be installed.
+        """
+        # We know for sure that the sheet will yield exactly one RecordBatch
+        batch = self.to_arrow()
+        return pl.from_arrow(data=pa.Table.from_batches([batch]))
+
     def __repr__(self) -> str:
         return self._sheet.__repr__()