patito 0.4.3__tar.gz → 0.5.0__tar.gz

{patito-0.4.3 → patito-0.5.0}/LICENSE

@@ -1,6 +1,7 @@
 MIT License
 
 Copyright (c) 2022 Oda Group Holding AS
+Copyright (c) 2023 Jakob Gerhard Martinussen and contributors
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

{patito-0.4.3 → patito-0.5.0}/PKG-INFO

@@ -1,33 +1,34 @@
 Metadata-Version: 2.1
 Name: patito
-Version: 0.4.3
+Version: 0.5.0
 Summary: A dataframe modelling library built on top of polars and pydantic.
 Home-page: https://github.com/kolonialno/patito
 License: MIT
 Keywords: validation,dataframe
 Author: Jakob Gerhard Martinussen
 Author-email: jakobgm@gmail.com
-Requires-Python: >=3.7,<4.0
+Requires-Python: >=3.8,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Provides-Extra: caching
 Provides-Extra: docs
 Provides-Extra: duckdb
 Provides-Extra: pandas
-Requires-Dist: Sphinx; extra == "docs"
-Requires-Dist: duckdb (>=0.5.0); (python_version >= "3.8" and python_version < "4.0") and (extra == "duckdb")
-Requires-Dist: importlib-metadata; python_version < "3.8"
-Requires-Dist: pandas; (python_version >= "3.8" and python_version < "4.0") and (extra == "pandas")
-Requires-Dist: polars (>=0.14.0)
-Requires-Dist: pyarrow (>=5.0.0); (python_version >= "3.8" and python_version < "4.0") and (extra == "duckdb")
+Requires-Dist: Sphinx (<7) ; extra == "docs"
+Requires-Dist: duckdb (>=0.6.0) ; (python_version >= "3.8" and python_version < "4.0") and (extra == "duckdb")
+Requires-Dist: pandas ; (python_version >= "3.8" and python_version < "4.0") and (extra == "pandas")
+Requires-Dist: polars (>=0.18.3)
+Requires-Dist: pyarrow (>=5.0.0) ; (python_version >= "3.8" and python_version < "4.0") and (extra == "caching" or extra == "duckdb")
 Requires-Dist: pydantic (>=1.7.0)
-Requires-Dist: sphinx-autobuild; extra == "docs"
-Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
-Requires-Dist: sphinx-rtd-theme; extra == "docs"
-Requires-Dist: sphinxcontrib-mermaid; extra == "docs"
+Requires-Dist: sphinx-autobuild ; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints ; extra == "docs"
+Requires-Dist: sphinx-rtd-theme ; extra == "docs"
+Requires-Dist: sphinx-toolbox ; extra == "docs"
+Requires-Dist: sphinxcontrib-mermaid ; extra == "docs"
 Requires-Dist: typing-extensions
 Project-URL: Documentation, https://patito.readthedocs.io
 Project-URL: Repository, https://github.com/kolonialno/patito
@@ -284,7 +285,7 @@ class Product(pt.Model):
     @property
     def url(self) -> str:
         return (
-            "https://oda.com/no/products/"
+            "https://example.com/no/products/"
             f"{self.product_id}-"
             f"{self.name.lower().replace(' ', '-')}"
         )
@@ -302,7 +303,7 @@ products = pl.DataFrame(
 milk_row = products.filter(pl.col("product_id" == 1))
 milk = Product.from_row(milk_row)
 print(milk.url)
-# https://oda.com/no/products/1-skimmed-milk
+# https://example.com/no/products/1-skimmed-milk
 ```
 
 If you "connect" the `Product` model with the `DataFrame` by the use of `patito.DataFrame.set_model()`, or alternatively by using `Product.DataFrame` directly, you can use the `.get()` method in order to filter the data frame down to a single row _and_ cast it to the respective model class:
@@ -317,6 +318,6 @@ products = Product.DataFrame(
 )
 milk = products.get(pl.col("product_id") == 1)
 print(milk.url)
-# https://oda.com/no/products/1-skimmed-milk
+# https://example.com/no/products/1-skimmed-milk
 ```
 

{patito-0.4.3 → patito-0.5.0}/README.md

@@ -249,7 +249,7 @@ class Product(pt.Model):
     @property
     def url(self) -> str:
         return (
-            "https://oda.com/no/products/"
+            "https://example.com/no/products/"
             f"{self.product_id}-"
             f"{self.name.lower().replace(' ', '-')}"
         )
@@ -267,7 +267,7 @@ products = pl.DataFrame(
 milk_row = products.filter(pl.col("product_id" == 1))
 milk = Product.from_row(milk_row)
 print(milk.url)
-# https://oda.com/no/products/1-skimmed-milk
+# https://example.com/no/products/1-skimmed-milk
 ```
 
 If you "connect" the `Product` model with the `DataFrame` by the use of `patito.DataFrame.set_model()`, or alternatively by using `Product.DataFrame` directly, you can use the `.get()` method in order to filter the data frame down to a single row _and_ cast it to the respective model class:
@@ -282,5 +282,5 @@ products = Product.DataFrame(
 )
 milk = products.get(pl.col("product_id") == 1)
 print(milk.url)
-# https://oda.com/no/products/1-skimmed-milk
+# https://example.com/no/products/1-skimmed-milk
 ```

{patito-0.4.3 → patito-0.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patito"
-version = "0.4.3"
+version = "0.5.0"
 description = "A dataframe modelling library built on top of polars and pydantic."
 authors = ["Jakob Gerhard Martinussen <jakobgm@gmail.com>"]
 license = "MIT"
@@ -11,28 +11,37 @@ documentation = "https://patito.readthedocs.io"
 keywords = ["validation", "dataframe"]
 
 [tool.poetry.dependencies]
-python = "^3.7"
+python = "^3.8"
 pydantic = ">=1.7.0"
-polars = ">=0.14.0"
+polars = ">=0.18.3"
 # Required for typing.Literal in python3.7
 typing-extensions = "*"
 pandas = {version = "*", optional = true, python = "^3.8"}
-duckdb = {version = ">=0.5.0", optional = true, python = "^3.8"}
+duckdb = {version = ">=0.6.0", optional = true, python = "^3.8"}
 pyarrow = {version = ">=5.0.0", optional = true, python = "^3.8"}
-importlib-metadata = {version = "*", python = "<3.8"}
 # Optional docs dependencies
-Sphinx = {version = "*", optional = true}
+Sphinx = {version = "<7", optional = true}
 sphinx-rtd-theme = {version = "*", optional = true}
 sphinx-autobuild = {version = "*", optional = true}
 sphinx-autodoc-typehints = {version = "*", optional = true}
+sphinx-toolbox = {version = "*", optional = true}
 sphinxcontrib-mermaid = {version = "*", optional = true}
 
 [tool.poetry.extras]
+# The pyarrow.parquet module is required for writing parquet caches to disk
+caching = ["pyarrow"]
 duckdb = ["duckdb", "pyarrow"]
 pandas = ["pandas"]
-docs = ["Sphinx", "sphinx-rtd-theme", "sphinx-autobuild", "sphinx-autodoc-typehints", "sphinxcontrib-mermaid"]
+docs = [
+  "Sphinx",
+  "sphinx-autobuild",
+  "sphinx-autodoc-typehints",
+  "sphinx-rtd-theme",
+  "sphinx-toolbox",
+  "sphinxcontrib-mermaid",
+]
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 black = ">=22.3.0"
 coverage = {version = "*", extras = ["toml"]}
 flake8 = "3.9.2"
@@ -80,7 +89,7 @@ exclude_lines = [
   "if not _PANDAS_AVAILABLE:",
   "except ImportError:",
 ]
-fail_under = 100
+fail_under = 99.64
 show_missing = true
 
 [tool.isort]
@@ -107,5 +116,21 @@ exclude = [
 
 [[tool.mypy.overrides]]
 module = ["tests.*", "noxfile"]
-disallow_untyped_defs = false
+allow_untyped_defs = true
+check_untyped_defs = true
 disallow_incomplete_defs = false
+# TODO: Go through and remove those we want to check in tests
+disable_error_code = [
+  "var-annotated",
+  "override",
+  "attr-defined",
+  "call-arg",
+  "type-var",
+  "misc",
+  "arg-type",
+  "assignment",
+]
+
+[[tool.mypy.overrides]]
+module = ["tests.test_validators"]
+warn_unused_ignores = false

{patito-0.4.3 → patito-0.5.0}/src/patito/__init__.py

@@ -6,6 +6,7 @@ from patito.exceptions import ValidationError
 from patito.polars import DataFrame, LazyFrame
 from patito.pydantic import Field, Model
 
+_CACHING_AVAILABLE = False
 _DUCKDB_AVAILABLE = False
 field = col("_")
 __all__ = [
@@ -16,6 +17,7 @@ __all__ = [
     "Model",
     "Series",
     "ValidationError",
+    "_CACHING_AVAILABLE",
     "_DUCKDB_AVAILABLE",
     "col",
     "exceptions",
@@ -24,17 +26,21 @@ __all__ = [
 ]
 
 try:
-    from patito.duckdb import Database, Relation, RelationSource
+    from patito import duckdb
 
     _DUCKDB_AVAILABLE = True
-    __all__ += [
-        "Database",
-        "Relation",
-        "RelationSource",
-    ]
+    __all__ += ["duckdb"]
 except ImportError:  # pragma: no cover
     pass
 
+try:
+    from patito.database import Database
+
+    _CACHING_AVAILABLE = True
+    __all__ += ["Database"]
+except ImportError:
+    pass
+
 
 try:
     from importlib.metadata import PackageNotFoundError, version