activemodel 0.11.0__tar.gz → 0.12.0__tar.gz

{activemodel-0.11.0 → activemodel-0.12.0}/.envrc

@@ -10,5 +10,5 @@ export POSTGRES_DB=development
 export DATABASE_URL=postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${DATABASE_HOST}:5432/development
 
 export PYTHONBREAKPOINT=ipdb.set_trace
-
+export PYTHONPATH=.
 # export ACTIVEMODEL_LOG_SQL=true

{activemodel-0.11.0 → activemodel-0.12.0}/.github/workflows/build_and_publish.yml

@@ -4,6 +4,7 @@ on:
     branches:
       - main
       - master
+  pull_request:
 
 env:
   # avoid build failures due to flaky pypi
@@ -34,8 +35,8 @@ jobs:
     needs: [release-please]
     if: needs.release-please.outputs.release_created
     steps:
-      - uses: actions/checkout@v4
-      - uses: jdx/mise-action@v2
+      - uses: actions/checkout@v5
+      - uses: jdx/mise-action@v3
       - run: direnv allow . && direnv export gha >> "$GITHUB_ENV"
       - run: uv build
       - run: uv publish --token ${{ secrets.PYPI_API_TOKEN }}
@@ -49,8 +50,8 @@ jobs:
         python-version: ["3.13", "3.12"]
     runs-on: ${{ matrix.os }}
     steps:
-      - uses: actions/checkout@v4
-      - uses: jdx/mise-action@v2
+      - uses: actions/checkout@v5
+      - uses: jdx/mise-action@v3
       - run: mise use python@${{ matrix.python-version }}
       - run: docker compose up -d --wait
       - uses: iloveitaly/github-action-direnv-load-and-mask@master

{activemodel-0.11.0 → activemodel-0.12.0}/.github/workflows/repo-sync.yml

@@ -9,7 +9,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Fetching Local Repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
       - name: Repository Metadata Sync
         uses: iloveitaly/github-actions-metadata-sync@main
         with:

activemodel-0.12.0/.tool-versions

@@ -0,0 +1,3 @@
+python 3.13.6
+uv 0.8.10
+direnv 2.37.1

{activemodel-0.11.0 → activemodel-0.12.0}/CHANGELOG.md

@@ -1,5 +1,43 @@
 # Changelog
 
+## [0.12.0](https://github.com/iloveitaly/activemodel/compare/v0.11.0...v0.12.0) (2025-09-03)
+
+
+### Features
+
+* add ActiveModelFactory helpers for session management and typeid generation ([a6b9915](https://github.com/iloveitaly/activemodel/commit/a6b9915cc0c74c3b0d1421bc3d0300c1a2f63426))
+* add base SQLModel and ActiveModel polyfactory factories ([12a5e1d](https://github.com/iloveitaly/activemodel/commit/12a5e1db9be741f12db174d5a25f6c2eee61a446))
+* add one_or_none method to BaseModel for safe queries ([7574777](https://github.com/iloveitaly/activemodel/commit/75747773f99a7a6cc82d2ca784ce607abd7ae511))
+* add pytest db_session fixture ([22d2ad8](https://github.com/iloveitaly/activemodel/commit/22d2ad8cbd77260c1c73524f3251ef7fd145caec))
+* add scalar method to QueryWrapper and return from delete ([3d68097](https://github.com/iloveitaly/activemodel/commit/3d680972b6a60b43de6d3826d4289b0f81847b74))
+* add SQLAlchemy protocol generation script ([6e4524e](https://github.com/iloveitaly/activemodel/commit/6e4524ed73644bea5b837ce2a9a7a1994b19df00))
+* add test_session context manager for test DB interactions ([d7a8112](https://github.com/iloveitaly/activemodel/commit/d7a8112f6b303a50df62ca55a0ca8a9c864e4686))
+* export test_session from activemodel.pytest ([9bf692a](https://github.com/iloveitaly/activemodel/commit/9bf692a72937d9a87a1ba2e574cff6dc4000bafd))
+* **pytest:** omit truncation tables, truncation db fixture ([f20f17e](https://github.com/iloveitaly/activemodel/commit/f20f17ea0854dfbde09e86f9446d57d33ab516e8))
+* support engine options ([9b00ecd](https://github.com/iloveitaly/activemodel/commit/9b00ecd4a7b1ac95a6a16b1e7727ef37afe81e00))
+* support passing session to global_session context manager ([567708a](https://github.com/iloveitaly/activemodel/commit/567708aeb6642e75b21932d599cd1431f62c4c65))
+
+
+### Bug Fixes
+
+* add return type to ActiveModelFactory save method ([96f4a0a](https://github.com/iloveitaly/activemodel/commit/96f4a0a70eb72b14dea8df16b9e92a4b79122285))
+* add ruff link + check on autogenerated types ([91ae5ea](https://github.com/iloveitaly/activemodel/commit/91ae5ea079533dc7e5600cf99c757ec5c6cd872f))
+* allow reentrant global_session when using same session reference ([d4d7949](https://github.com/iloveitaly/activemodel/commit/d4d79493afdf7ef6ae36f9f6d1c2949eb2b7c393))
+* clarify duplicate TypeID prefix error message ([a73c221](https://github.com/iloveitaly/activemodel/commit/a73c221fd51c1efa70812eb1a21a96f0af7faadc))
+* prevent factories from setting timestamp fields by default ([5d78745](https://github.com/iloveitaly/activemodel/commit/5d7874506047b7ae91f22566e1dc2c64d68814f4))
+
+
+### Documentation
+
+* added related issue ([3c47e60](https://github.com/iloveitaly/activemodel/commit/3c47e601b8f3eb8a85d9cb84a3e64df68ea10d90))
+* clarify session management and querying in usage section ([cdadaff](https://github.com/iloveitaly/activemodel/commit/cdadafffa1ac835f2f6419f40de5bf2f281f94d2))
+* connection pool tip ([5d92b99](https://github.com/iloveitaly/activemodel/commit/5d92b992d4f71af0bf8244826cd4d12281fe350b))
+* fix JSONB usage in example and clarify imports in README ([386e69d](https://github.com/iloveitaly/activemodel/commit/386e69d5ad99baddfba119f62b6d3c6a53809857))
+* improve base model tablename docstring ([de98cbc](https://github.com/iloveitaly/activemodel/commit/de98cbcff0b2ef903207f4c65fb1aa6e014cf40c))
+* **pytest:** clarify ActiveModelFactory.save behavior with relationships and truncation ([6ee2cbe](https://github.com/iloveitaly/activemodel/commit/6ee2cbef23601943271a36a90f2f453de4519c72))
+* **typeid:** clarify exception on invalid UUID string ([bf13a7e](https://github.com/iloveitaly/activemodel/commit/bf13a7e7e537d41a94ff926dcabf316b0d62625b))
+* update TODOs for testing, constraints, and unique key suggestions ([6010f8f](https://github.com/iloveitaly/activemodel/commit/6010f8fec1c3bdb22d68ac3b545ebeafd6bd04ef))
+
 ## [0.11.0](https://github.com/iloveitaly/activemodel/compare/v0.10.0...v0.11.0) (2025-04-05)
 
 

{activemodel-0.11.0 → activemodel-0.12.0}/Makefile

@@ -8,6 +8,10 @@ up:
 db_open:
 	open -a TablePlus $$DATABASE_URL
 
+lint:
+	pyright
+	ruff format
+
 clean:
 	rm -rf *.egg-info
 	rm -rf .venv

activemodel-0.11.0/README.md → activemodel-0.12.0/PKG-INFO

@@ -1,3 +1,18 @@
+Metadata-Version: 2.4
+Name: activemodel
+Version: 0.12.0
+Summary: Make SQLModel more like an a real ORM
+Project-URL: Repository, https://github.com/iloveitaly/activemodel
+Author-email: Michael Bianco <iloveitaly@gmail.com>
+License-File: LICENSE
+Keywords: activemodel,activerecord,orm,sqlalchemy,sqlmodel
+Requires-Python: >=3.10
+Requires-Dist: python-decouple-typed>=3.11.0
+Requires-Dist: sqlmodel>=0.0.22
+Requires-Dist: textcase>=0.4.0
+Requires-Dist: typeid-python>=0.3.1
+Description-Content-Type: text/markdown
+
 # ActiveModel: ORM Wrapper for SQLModel
 
 No, this isn't *really* [ActiveModel](https://guides.rubyonrails.org/active_model_basics.html). It's just a wrapper around SQLModel that provides a more ActiveRecord-like interface.
@@ -44,15 +59,17 @@ from sqlmodel import SQLModel
 
 SQLModel.metadata.create_all(get_engine())
 
-# now you can create a user!
+# now you can create a user! without managing sessions!
 User(a_field="a").save()
 ```
 
 Maybe you like JSON:
 
 ```python
-from activemodel import BaseModel
+from sqlalchemy.dialects.postgresql import JSONB
 from pydantic import BaseModel as PydanticBaseModel
+
+from activemodel import BaseModel
 from activemodel.mixins import PydanticJSONMixin, TypeIDMixin, TimestampsMixin
 
 class SubObject(PydanticBaseModel):
@@ -66,11 +83,28 @@ class User(
     TypeIDMixin("user"),
     table=True
 ):
-    list_field: list[SubObject] = Field(sa_type=JSONB())
+    list_field: list[SubObject] = Field(sa_type=JSONB)
+```
+
+You'll probably want to query the model. Look ma, no sessions!
+
+```python
+User.where(id="user_123").all()
+
+# or, even better, for this case
+User.one("user_123")
 ```
 
+Magically creating sessions for DB operations is one of the main problems this project tackles. Even better, you can set
+a single session object to be used for all DB operations. This is helpful for DB transactions, [specifically rolling back
+DB operations on each test.](#pytest)
+
 ## Usage
 
+### Pytest
+
+TODO detail out truncation and transactions
+
 ### Integrating Alembic
 
 `alembic init` will not work out of the box. You need to mutate a handful of files:
@@ -192,6 +226,7 @@ SQLModel & SQLAlchemy are tricky. Here are some useful internal tricks:
 * Set the value on a field, without marking it as dirty `attributes.set_committed_value(instance, field_name, val)`
 * Is a model dirty `instance_state(instance).modified`
 * `select(Table).outerjoin??` won't work in a ipython session, but `Table.__table__.outerjoin??` will. `__table__` is a reference to the underlying SQLAlchemy table record.
+* `get_engine().pool.stats()` is helpful for inspecting connection pools and limits\
 
 ### TypeID
 
@@ -255,6 +290,7 @@ https://github.com/DarylStark/my_data/blob/a17b8b3a8463b9953821b89fee895e272f94d
 
 * https://github.com/woofz/sqlmodel-basecrud
 * https://github.com/0xthiagomartins/sqlmodel-controller
+* https://github.com/litestar-org/advanced-alchemy?tab=readme-ov-file
 
 ## Inspiration
 
@@ -267,4 +303,4 @@ https://github.com/DarylStark/my_data/blob/a17b8b3a8463b9953821b89fee895e272f94d
 
 ## Upstream Changes
 
-- [ ] https://github.com/fastapi/sqlmodel/pull/1293
+- [ ] https://github.com/fastapi/sqlmodel/pull/1293

activemodel-0.11.0/PKG-INFO → activemodel-0.12.0/README.md

@@ -1,18 +1,3 @@
-Metadata-Version: 2.4
-Name: activemodel
-Version: 0.11.0
-Summary: Make SQLModel more like an a real ORM
-Project-URL: Repository, https://github.com/iloveitaly/activemodel
-Author-email: Michael Bianco <iloveitaly@gmail.com>
-License-File: LICENSE
-Keywords: activemodel,activerecord,orm,sqlalchemy,sqlmodel
-Requires-Python: >=3.10
-Requires-Dist: pydash>=8.0.4
-Requires-Dist: python-decouple-typed>=3.11.0
-Requires-Dist: sqlmodel>=0.0.22
-Requires-Dist: typeid-python>=0.3.1
-Description-Content-Type: text/markdown
-
 # ActiveModel: ORM Wrapper for SQLModel
 
 No, this isn't *really* [ActiveModel](https://guides.rubyonrails.org/active_model_basics.html). It's just a wrapper around SQLModel that provides a more ActiveRecord-like interface.
@@ -59,15 +44,17 @@ from sqlmodel import SQLModel
 
 SQLModel.metadata.create_all(get_engine())
 
-# now you can create a user!
+# now you can create a user! without managing sessions!
 User(a_field="a").save()
 ```
 
 Maybe you like JSON:
 
 ```python
-from activemodel import BaseModel
+from sqlalchemy.dialects.postgresql import JSONB
 from pydantic import BaseModel as PydanticBaseModel
+
+from activemodel import BaseModel
 from activemodel.mixins import PydanticJSONMixin, TypeIDMixin, TimestampsMixin
 
 class SubObject(PydanticBaseModel):
@@ -81,11 +68,28 @@ class User(
     TypeIDMixin("user"),
     table=True
 ):
-    list_field: list[SubObject] = Field(sa_type=JSONB())
+    list_field: list[SubObject] = Field(sa_type=JSONB)
+```
+
+You'll probably want to query the model. Look ma, no sessions!
+
+```python
+User.where(id="user_123").all()
+
+# or, even better, for this case
+User.one("user_123")
 ```
 
+Magically creating sessions for DB operations is one of the main problems this project tackles. Even better, you can set
+a single session object to be used for all DB operations. This is helpful for DB transactions, [specifically rolling back
+DB operations on each test.](#pytest)
+
 ## Usage
 
+### Pytest
+
+TODO detail out truncation and transactions
+
 ### Integrating Alembic
 
 `alembic init` will not work out of the box. You need to mutate a handful of files:
@@ -207,6 +211,7 @@ SQLModel & SQLAlchemy are tricky. Here are some useful internal tricks:
 * Set the value on a field, without marking it as dirty `attributes.set_committed_value(instance, field_name, val)`
 * Is a model dirty `instance_state(instance).modified`
 * `select(Table).outerjoin??` won't work in a ipython session, but `Table.__table__.outerjoin??` will. `__table__` is a reference to the underlying SQLAlchemy table record.
+* `get_engine().pool.stats()` is helpful for inspecting connection pools and limits\
 
 ### TypeID
 
@@ -270,6 +275,7 @@ https://github.com/DarylStark/my_data/blob/a17b8b3a8463b9953821b89fee895e272f94d
 
 * https://github.com/woofz/sqlmodel-basecrud
 * https://github.com/0xthiagomartins/sqlmodel-controller
+* https://github.com/litestar-org/advanced-alchemy?tab=readme-ov-file
 
 ## Inspiration
 
@@ -282,4 +288,4 @@ https://github.com/DarylStark/my_data/blob/a17b8b3a8463b9953821b89fee895e272f94d
 
 ## Upstream Changes
 
-- [ ] https://github.com/fastapi/sqlmodel/pull/1293
+- [ ] https://github.com/fastapi/sqlmodel/pull/1293

{activemodel-0.11.0 → activemodel-0.12.0}/TODO

@@ -10,9 +10,12 @@ Docs are bad:
 
 TODO
 
+- [ ] nested sessions, maybe only with connections
+- [ ] tests for polyfactory support, setting session across all factories
 - [ ] sessions in tests, they don't work right now
 - [ ] snake case for attributes https://github.com/sqlalchemy/sqlalchemy/issues/7149
-
+- [ ] compound unique constraints don't have great names
+- [ ] https://grok.com/share/bGVnYWN5_7bfd5140-2351-4e4f-a8ba-bdedb59ac1e1 figure out better json tracking
 
 find_or_create
 
@@ -50,3 +53,14 @@ clear engine
 #         _engine = None
 #         _connection = None
 ```
+
+---
+
+There should be a better way to add unique keys
+
+```
+    __table_args__ = (
+        # users should never see two schemas with the same name
+        UniqueConstraint("note_type", "doctor_id", name="doctor_and_note_type_unique"),
+    )
+```

{activemodel-0.11.0 → activemodel-0.12.0}/activemodel/base_model.py

@@ -2,10 +2,11 @@ import json
 import typing as t
 from uuid import UUID
 
-import pydash
 import sqlalchemy as sa
 import sqlmodel as sm
+import textcase
 from sqlalchemy import Connection, event
+from sqlalchemy.dialects.postgresql import insert as postgres_insert
 from sqlalchemy.orm import Mapper, declared_attr
 from sqlalchemy.orm.attributes import flag_modified as sa_flag_modified
 from sqlalchemy.orm.base import instance_state
@@ -19,7 +20,6 @@ from . import get_column_from_field_patch  # noqa: F401
 from .logger import logger
 from .query_wrapper import QueryWrapper
 from .session_manager import get_session
-from sqlalchemy.dialects.postgresql import insert as postgres_insert
 
 POSTGRES_INDEXES_NAMING_CONVENTION = {
     "ix": "%(column_0_label)s_idx",
@@ -152,19 +152,19 @@ class BaseModel(SQLModel):
     @declared_attr
     def __tablename__(cls) -> str:
         """
-        Automatically generates the table name for the model by converting the class name from camel case to snake case.
-        This is the recommended format for table names:
+        Automatically generates the table name for the model by converting the model's class name from camel case to snake case.
+        This is the recommended text case style for table names:
 
         https://wiki.postgresql.org/wiki/Don%27t_Do_This#Don.27t_use_upper_case_table_or_column_names
 
-        By default, the class is lower cased which makes it harder to read.
+        By default, the model's class name is lower cased which makes it harder to read.
 
-        Many snake_case libraries struggle with snake case for names like LLMCache, which is why we are using a more
-        complicated implementation from pydash.
+        Also, many text case conversion libraries struggle handling words like "LLMCache", this is why we are using
+        a more precise library which processes such acronyms: [`textcase`](https://pypi.org/project/textcase/).
 
         https://stackoverflow.com/questions/1175208/elegant-python-function-to-convert-camelcase-to-snake-case
         """
-        return pydash.strings.snake_case(cls.__name__)
+        return textcase.snake(cls.__name__)
 
     @classmethod
     def foreign_key(cls, **kwargs):
@@ -234,6 +234,8 @@ class BaseModel(SQLModel):
         return result
 
     def delete(self):
+        "Delete record completely from the database"
+
         with get_session() as session:
             if old_session := Session.object_session(self):
                 old_session.expunge(self)
@@ -404,6 +406,19 @@ class BaseModel(SQLModel):
         with get_session() as session:
             return session.exec(statement).first()
 
+    @classmethod
+    def one_or_none(cls, *args: t.Any, **kwargs: t.Any):
+        """
+        Gets a single record from the database. Pass an PK ID or a kwarg to filter by.
+        Returns None if no record is found. Throws an error if more than one record is found.
+        """
+
+        args, kwargs = cls.__process_filter_args__(*args, **kwargs)
+        statement = select(cls).filter(*args).filter_by(**kwargs)
+
+        with get_session() as session:
+            return session.exec(statement).one_or_none()
+
     @classmethod
     def one(cls, *args: t.Any, **kwargs: t.Any):
         """

activemodel-0.12.0/activemodel/cli/__init__.py

@@ -0,0 +1,147 @@
+"""
+This module provides utilities for generating Protocol type definitions for SQLAlchemy's
+SelectOfScalar methods, as well as formatting and fixing Python files using ruff.
+"""
+
+import inspect
+import logging
+import os
+import subprocess
+from pathlib import Path
+from typing import Any  # already imported in header of generated file
+
+import sqlmodel as sm
+from sqlmodel.sql.expression import SelectOfScalar
+
+from test.test_wrapper import QueryWrapper
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+QUERY_WRAPPER_CLASS_NAME = QueryWrapper.__name__
+
+
+def format_python_file(file_path: str | Path) -> bool:
+    """
+    Format a Python file using ruff.
+
+    Args:
+        file_path: Path to the Python file to format
+
+    Returns:
+        bool: True if formatting was successful, False otherwise
+    """
+    try:
+        subprocess.run(["ruff", "format", str(file_path)], check=True)
+        logger.info(f"Formatted file using ruff at {file_path}")
+        return True
+    except subprocess.CalledProcessError as e:
+        logger.error(f"Error running ruff to format the file: {e}")
+        return False
+
+
+def fix_python_file(file_path: str | Path) -> bool:
+    """
+    Fix linting issues in a Python file using ruff.
+
+    Args:
+        file_path: Path to the Python file to fix
+
+    Returns:
+        bool: True if fixing was successful, False otherwise
+    """
+    try:
+        subprocess.run(["ruff", "check", str(file_path), "--fix"], check=True)
+        logger.info(f"Fixed linting issues using ruff at {file_path}")
+        return True
+    except subprocess.CalledProcessError as e:
+        logger.error(f"Error running ruff to fix the file: {e}")
+        return False
+
+
+def generate_sqlalchemy_protocol():
+    """Generate Protocol type definitions for SQLAlchemy SelectOfScalar methods"""
+    logger.info("Starting SQLAlchemy protocol generation")
+
+    header = """
+# IMPORTANT: This file is auto-generated. Do not edit directly.
+
+from typing import Protocol, TypeVar, Any, Generic
+import sqlmodel as sm
+from sqlalchemy.sql.base import _NoArg
+
+T = TypeVar('T', bound=sm.SQLModel, covariant=True)
+
+class SQLAlchemyQueryMethods(Protocol, Generic[T]):
+    \"""Protocol defining SQLAlchemy query methods forwarded by QueryWrapper.__getattr__\"""
+"""
+    # Initialize output list for generated method signatures
+    output: list = []
+
+    try:
+        # Get all methods from SelectOfScalar
+        methods = inspect.getmembers(SelectOfScalar)
+        logger.debug(f"Discovered {len(methods)} methods from SelectOfScalar")
+
+        for name, method in methods:
+            # Skip private/dunder methods
+            if name.startswith("_"):
+                continue
+
+            if not inspect.isfunction(method) and not inspect.ismethod(method):
+                logger.debug(f"Skipping non-method: {name}")
+                continue
+
+            logger.debug(f"Processing method: {name}")
+            try:
+                signature = inspect.signature(method)
+                params = []
+
+                # Process parameters, skipping 'self'
+                for param_name, param in list(signature.parameters.items())[1:]:
+                    if param.kind == param.VAR_POSITIONAL:
+                        params.append(f"*{param_name}: Any")
+                    elif param.kind == param.VAR_KEYWORD:
+                        params.append(f"**{param_name}: Any")
+                    else:
+                        if param.default is inspect.Parameter.empty:
+                            params.append(f"{param_name}: Any")
+                        else:
+                            default_repr = repr(param.default)
+                            params.append(f"{param_name}: Any = {default_repr}")
+
+                params_str = ", ".join(params)
+                output.append(
+                    f'    def {name}(self, {params_str}) -> "{QUERY_WRAPPER_CLASS_NAME}[T]": ...'
+                )
+            except (ValueError, TypeError) as e:
+                logger.warning(f"Could not get signature for {name}: {e}")
+                # Some methods might not have proper signatures
+                output.append(
+                    f'    def {name}(self, *args: Any, **kwargs: Any) -> "{QUERY_WRAPPER_CLASS_NAME}[T]": ...'
+                )
+
+        # Write the output to a file
+        protocol_path = (
+            Path(__file__).parent.parent / "types" / "sqlalchemy_protocol.py"
+        )
+
+        # Ensure directory exists
+        os.makedirs(protocol_path.parent, exist_ok=True)
+
+        with open(protocol_path, "w") as f:
+            f.write(header + "\n".join(output))
+
+        logger.info(f"Generated SQLAlchemy protocol at {protocol_path}")
+
+        # Format and fix the generated file with ruff
+        format_python_file(protocol_path)
+        fix_python_file(protocol_path)
+    except Exception as e:
+        logger.error(f"Error generating SQLAlchemy protocol: {e}", exc_info=True)
+        raise
+
+
+if __name__ == "__main__":
+    generate_sqlalchemy_protocol()

{activemodel-0.11.0 → activemodel-0.12.0}/activemodel/mixins/timestamps.py

@@ -20,7 +20,10 @@ class TimestampsMixin:
     >>> class MyModel(TimestampsMixin, SQLModel):
     >>>    pass
 
-    Originally pulled from: https://github.com/tiangolo/sqlmodel/issues/252
+    Notes:
+    
+    - Originally pulled from: https://github.com/tiangolo/sqlmodel/issues/252
+    - Related issue: https://github.com/fastapi/sqlmodel/issues/539
     """
 
     created_at: datetime | None = Field(

{activemodel-0.11.0 → activemodel-0.12.0}/activemodel/mixins/typeid.py

@@ -17,7 +17,7 @@ def TypeIDMixin(prefix: str):
     # NOTE this will cause issues on code reloads
     assert prefix
     assert prefix not in _prefixes, (
-        f"prefix {prefix} already exists, pick a different one"
+        f"TypeID prefix '{prefix}' already exists, pick a different one"
     )
 
     class _TypeIDMixin:

activemodel-0.12.0/activemodel/pytest/__init__.py

@@ -0,0 +1,2 @@
+from .transaction import database_reset_transaction, test_session
+from .truncate import database_reset_truncate

activemodel-0.12.0/activemodel/pytest/factories.py

@@ -0,0 +1,102 @@
+"""
+Notes on polyfactory:
+
+1. is_supported_type validates that the class can be used to generate a factory
+https://github.com/litestar-org/polyfactory/issues/655#issuecomment-2727450854
+"""
+
+import typing as t
+
+from polyfactory.factories.pydantic_factory import ModelFactory
+from polyfactory.field_meta import FieldMeta
+from typeid import TypeID
+
+from activemodel.session_manager import global_session
+
+# TODO not currently used
+# def type_id_provider(cls, field_meta):
+#     # TODO this doesn't work well with __ args:
+#     # https://github.com/litestar-org/polyfactory/pull/666/files
+#     return str(TypeID("hi"))
+
+
+# BaseFactory.add_provider(TypeIDType, type_id_provider)
+
+
+class SQLModelFactory[T](ModelFactory[T]):
+    """
+    Base factory for SQLModel models:
+
+    1. Ability to ignore all relationship fks
+    2. Option to ignore all pks
+    """
+
+    __is_base_factory__ = True
+
+    @classmethod
+    def should_set_field_value(cls, field_meta: FieldMeta, **kwargs: t.Any) -> bool:
+        # TODO what is this checking for?
+        has_object_override = hasattr(cls, field_meta.name)
+
+        # TODO this should be more intelligent, it's goal is to detect all of the relationship field and avoid settings them
+        if not has_object_override and (
+            field_meta.name == "id" or field_meta.name.endswith("_id")
+        ):
+            return False
+
+        return super().should_set_field_value(field_meta, **kwargs)
+
+
+# TODO we need to think through how to handle relationships and autogenerate them
+class ActiveModelFactory[T](SQLModelFactory[T]):
+    __is_base_factory__ = True
+    __sqlalchemy_session__ = None
+
+    # TODO we shouldn't have to type this, but `save()` typing is not working
+    @classmethod
+    def save(cls, *args, **kwargs) -> T:
+        """
+        Where this gets tricky, is this can be called multiple times within the same callstack. This can happen when
+        a factory uses other factories to create relationships.
+
+        In a truncation strategy, the __sqlalchemy_session__ is set to None.
+        """
+        with global_session(cls.__sqlalchemy_session__):
+            return cls.build(*args, **kwargs).save()
+
+    @classmethod
+    def foreign_key_typeid(cls):
+        """
+        Return a random type id for the foreign key on this model.
+
+        This is helpful for generating TypeIDs for testing 404s, parsing, manually settings, etc
+        """
+        # TODO right now assumes the model is typeid, maybe we should assert against this?
+        primary_key_name = cls.__model__.primary_key_column().name
+        return TypeID(
+            cls.__model__.model_fields[primary_key_name].sa_column.type.prefix
+        )
+
+    @classmethod
+    def should_set_field_value(cls, field_meta: FieldMeta, **kwargs: t.Any) -> bool:
+        # do not default deleted at mixin to deleted!
+        # TODO should be smarter about detecting if the mixin is in place
+        if field_meta.name in ["deleted_at", "updated_at", "created_at"]:
+            return False
+
+        return super().should_set_field_value(field_meta, **kwargs)
+
+    # @classmethod
+    # def build(
+    #     cls,
+    #     factory_use_construct: bool | None = None,
+    #     sqlmodel_save: bool = False,
+    #     **kwargs: t.Any,
+    # ) -> T:
+    #     result = super().build(factory_use_construct=factory_use_construct, **kwargs)
+
+    #     # TODO allow magic dunder method here
+    #     if sqlmodel_save:
+    #         result.save()
+
+    #     return result