sqltypes 0.1.0__tar.gz

sqltypes-0.1.0/PKG-INFO

@@ -0,0 +1,33 @@
+Metadata-Version: 2.1
+Name: sqltypes
+Version: 0.1.0
+Summary: Custom types for SQLModel/SQLalchemy
+Author-email: Marcel Claramunt <marcel@moveread.com>
+Project-URL: repo, https://github.com/marciclabas/python-storage.git
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic
+Requires-Dist: sqlalchemy
+Requires-Dist: lazy-loader
+
+# Sql Types
+
+> Custom types for SQLModel/SQLalchemy
+
+## Usage
+
+```python
+from typing import Sequence
+from pydantic import BaseModel
+from sqlmodel import Field, SQLModel
+from sqltypes import PydanticModel, SpaceDelimitedList
+
+class User(BaseModel):
+  name: str
+  age: int
+
+class MyDBItem(SQLModel, table=True):
+  id: int | None = Field(default=None, primary_key=True)
+  tags: Sequence[str] = Field(sa_type=SpaceDelimitedList)
+  user: User = Field(sa_type=PydanticModel(User))
+```

sqltypes-0.1.0/README.md

@@ -0,0 +1,21 @@
+# Sql Types
+
+> Custom types for SQLModel/SQLalchemy
+
+## Usage
+
+```python
+from typing import Sequence
+from pydantic import BaseModel
+from sqlmodel import Field, SQLModel
+from sqltypes import PydanticModel, SpaceDelimitedList
+
+class User(BaseModel):
+  name: str
+  age: int
+
+class MyDBItem(SQLModel, table=True):
+  id: int | None = Field(default=None, primary_key=True)
+  tags: Sequence[str] = Field(sa_type=SpaceDelimitedList)
+  user: User = Field(sa_type=PydanticModel(User))
+```

sqltypes-0.1.0/pyproject.toml

@@ -0,0 +1,25 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "sqltypes"
+version = "0.1.0"
+authors = [
+  {name="Marcel Claramunt", email="marcel@moveread.com"}
+]
+description = "Custom types for SQLModel/SQLalchemy"
+dependencies = [
+  "pydantic", "sqlalchemy", "lazy-loader"
+]
+requires-python = ">=3.10"
+readme = {file="README.md", content-type="text/markdown"}
+
+[project.urls]
+repo = "https://github.com/marciclabas/python-storage.git"
+
+# [project.optional-dependencies]
+# test = [
+#   "pytest < 5.0.0",
+#   "pytest-cov[all]"
+# ]

sqltypes-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

sqltypes-0.1.0/src/sqltypes/__init__.py

@@ -0,0 +1,6 @@
+"""
+### Sql Types
+> Custom types for SQLModel, SQLalchemy
+"""
+import lazy_loader as lazy
+__getattr__, __dir__, __all__ = lazy.attach_stub(__name__, __file__)

sqltypes-0.1.0/src/sqltypes/__init__.pyi

@@ -0,0 +1,9 @@
+from .meta import CustomTypeMeta
+from .string import CustomString, SpaceDelimitedList, CustomStringMeta
+from .pydantic import PydanticModel
+
+__all__ = [
+  'CustomTypeMeta',
+  'CustomStringMeta', 'CustomString', 'SpaceDelimitedList',
+  'PydanticModel',
+]

sqltypes-0.1.0/src/sqltypes/meta.py

@@ -0,0 +1,26 @@
+from typing import TypeVar, Callable
+from pydantic import BaseModel
+from sqlalchemy.types import TypeDecorator, TypeEngine
+
+T = TypeVar('T')
+D = TypeVar('D')
+E = TypeVar('E', bound=TypeEngine)
+M = TypeVar('M', bound=BaseModel)
+
+class CustomTypeMeta(type):
+  def __new__(cls, name: str, bases: tuple, dct: dict, Impl: type[E], dump: Callable[[T], D], parse: Callable[[D], T]):
+    
+    class CustomType(TypeDecorator):
+      impl = Impl
+      cache_ok = True
+
+      def process_bind_param(self, value: T | None, dialect) -> D | None:
+        if value is not None:
+          return dump(value)
+
+      def process_result_value(self, value: D | None, dialect) -> T | None:
+        if value is not None:
+          return parse(value)
+
+    # Return the new class type
+    return type(name, (CustomType, *bases), dct)

sqltypes-0.1.0/src/sqltypes/pydantic.py

@@ -0,0 +1,10 @@
+from typing import TypeVar, Generic
+from sqlalchemy.types import TypeDecorator, JSON
+from pydantic import BaseModel
+from .meta import CustomTypeMeta
+
+M = TypeVar('M', bound=BaseModel)
+
+class PydanticModel(type):
+  def __new__(cls, Model: type[M], name: str | None = None) -> type[TypeDecorator[M]]:
+    return CustomTypeMeta(f'DB{name or Model.__name__}', (), {}, JSON, dump=lambda M: M.model_dump(exclude_none=True), parse=Model.model_validate)

sqltypes-0.1.0/src/sqltypes/string.py

@@ -0,0 +1,15 @@
+from typing import TypeVar, Callable
+from sqlalchemy.types import TypeDecorator, String
+from .meta import CustomTypeMeta
+
+T = TypeVar('T')
+
+class CustomStringMeta(type):
+  def __new__(cls, name: str, bases, dct, dump: Callable[[T], str], parse: Callable[[str], T]) -> type[TypeDecorator[T]]:
+    return CustomTypeMeta(name, bases, dct, String, dump=dump, parse=parse)
+  
+def CustomString(name: str, dump: Callable[[T], str], parse: Callable[[str], T]) -> type[TypeDecorator[T]]:
+  return CustomTypeMeta(name, (), {}, String, dump=dump, parse=parse)
+
+class SpaceDelimitedList(metaclass=CustomStringMeta, dump=' '.join, parse=str.split):
+  ...

sqltypes-0.1.0/src/sqltypes.egg-info/PKG-INFO

@@ -0,0 +1,33 @@
+Metadata-Version: 2.1
+Name: sqltypes
+Version: 0.1.0
+Summary: Custom types for SQLModel/SQLalchemy
+Author-email: Marcel Claramunt <marcel@moveread.com>
+Project-URL: repo, https://github.com/marciclabas/python-storage.git
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic
+Requires-Dist: sqlalchemy
+Requires-Dist: lazy-loader
+
+# Sql Types
+
+> Custom types for SQLModel/SQLalchemy
+
+## Usage
+
+```python
+from typing import Sequence
+from pydantic import BaseModel
+from sqlmodel import Field, SQLModel
+from sqltypes import PydanticModel, SpaceDelimitedList
+
+class User(BaseModel):
+  name: str
+  age: int
+
+class MyDBItem(SQLModel, table=True):
+  id: int | None = Field(default=None, primary_key=True)
+  tags: Sequence[str] = Field(sa_type=SpaceDelimitedList)
+  user: User = Field(sa_type=PydanticModel(User))
+```

sqltypes-0.1.0/src/sqltypes.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+src/sqltypes/__init__.py
+src/sqltypes/__init__.pyi
+src/sqltypes/meta.py
+src/sqltypes/pydantic.py
+src/sqltypes/string.py
+src/sqltypes.egg-info/PKG-INFO
+src/sqltypes.egg-info/SOURCES.txt
+src/sqltypes.egg-info/dependency_links.txt
+src/sqltypes.egg-info/requires.txt
+src/sqltypes.egg-info/top_level.txt

sqltypes-0.1.0/src/sqltypes.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

sqltypes-0.1.0/src/sqltypes.egg-info/requires.txt

@@ -0,0 +1,3 @@
+pydantic
+sqlalchemy
+lazy-loader

sqltypes-0.1.0/src/sqltypes.egg-info/top_level.txt

@@ -0,0 +1 @@
+sqltypes