polars-api 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

polars_api/__init__.py

@@ -0,0 +1,3 @@
+from .api import Api
+
+__all__ = ["Api"]

polars_api/api.py

@@ -0,0 +1,40 @@
+from typing import Optional
+
+import httpx
+import polars as pl
+
+
+@pl.api.register_expr_namespace("api")
+class Api:
+    def __init__(self, url: pl.Expr) -> None:
+        self._url = url
+
+    @staticmethod
+    def _get(url: str) -> Optional[str]:
+        result = httpx.get(url)
+        if result.status_code == 200:
+            return result.text
+        else:
+            return None
+
+    @staticmethod
+    def _post(url: str, body: str) -> Optional[str]:
+        result = httpx.post(url, json=body)
+        if result.status_code == 200:
+            return result.text
+        else:
+            return None
+
+    def get(self, params: Optional[pl.Expr] = None) -> pl.Expr:
+        return self._url.map_elements(
+            lambda x: self._get(x),
+            return_dtype=pl.Utf8,
+        )
+
+    def post(self, body: Optional[pl.Expr] = None) -> pl.Expr:
+        if body is None:
+            body = pl.lit("")
+        return pl.struct([self._url.alias("url"), body.alias("body")]).map_elements(
+            lambda x: self._post(x["url"], x["body"]),
+            return_dtype=pl.Utf8,
+        )

{polars_api-0.1.1.dist-info → polars_api-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: polars-api
-Version: 0.1.1
+Version: 0.1.2
 Summary: Polars extension for dealing with REST APIs
 Author-email: Diego Garcia Lozano <diegoglozano96@gmail.com>
 Project-URL: Homepage, https://diegoglozano.github.io/polars-api/

polars_api-0.1.2.dist-info/RECORD

@@ -0,0 +1,8 @@
+polars_api/__init__.py,sha256=JoRS_iy8pZLEz_ADKooXOqOPlUQcd5wvdcSa_PMuGLs,40
+polars_api/api.py,sha256=BU90-ZJQ5easjkRdzmrKM65JzQVzRqzpQxC33BT9Zus,1093
+tests/test_api.py,sha256=1OfyzFM1fUecl8TnY9CUkWt_zpUNgoIYnN_Ma9ZN77w,32
+polars_api-0.1.2.dist-info/LICENSE,sha256=Ms_a-6jJtWdrIBCOaYS_hKFCODG2QXHcNgsQJnbujnA,1076
+polars_api-0.1.2.dist-info/METADATA,sha256=aVwOpJnazzO9xwz5j7CPHbWkAZJdhLn2Blc4bKBGXSM,4046
+polars_api-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+polars_api-0.1.2.dist-info/top_level.txt,sha256=DB_mtsLtzuhP54ExuJV4MonMxWxqYnB8gpSywjrGN3A,17
+polars_api-0.1.2.dist-info/RECORD,,

{polars_api-0.1.1.dist-info → polars_api-0.1.2.dist-info}/top_level.txt

@@ -1 +1,2 @@
 polars_api
+tests

tests/test_api.py

@@ -0,0 +1,2 @@
+def test_get():
+    assert True

polars_api-0.1.1.dist-info/RECORD

@@ -1,5 +0,0 @@
-polars_api-0.1.1.dist-info/LICENSE,sha256=Ms_a-6jJtWdrIBCOaYS_hKFCODG2QXHcNgsQJnbujnA,1076
-polars_api-0.1.1.dist-info/METADATA,sha256=kflG9Gkj7WD5rRZYZLx1JzCXHALbtQGLug9nHtvjMlQ,4046
-polars_api-0.1.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-polars_api-0.1.1.dist-info/top_level.txt,sha256=8jrlIZHjbDXkT8co6HdxDmJ8hbSrMCCyUTF5XpxAc_w,11
-polars_api-0.1.1.dist-info/RECORD,,