sqliter-py 0.16.0__py3-none-any.whl → 0.17.0__py3-none-any.whl

sqliter/exceptions.py

@@ -209,3 +209,19 @@ class InvalidRelationshipError(SqliterError):
         "Invalid relationship path '{}': field '{}' is not a valid "
         "foreign key relationship on model {}"
     )
+
+
+class ManyToManyError(SqliterError):
+    """Base exception for many-to-many relationship errors."""
+
+    message_template = "Many-to-many error: {}"
+
+
+class ManyToManyIntegrityError(ManyToManyError):
+    """Raised when a M2M operation fails due to missing context or pk.
+
+    This error occurs when attempting to use a M2M relationship without
+    a database context or on an unsaved instance (no primary key).
+    """
+
+    message_template = "Many-to-many integrity error: {}"

sqliter/helpers.py

@@ -9,11 +9,38 @@ to database schema translation.
 from __future__ import annotations
 
 import datetime
+import re
 from typing import Union
 
 from sqliter.constants import SQLITE_TYPE_MAPPING
 
 
+def validate_table_name(table_name: str) -> str:
+    """Validate that a table name contains only safe characters.
+
+    Table names must contain only alphanumeric characters and underscores,
+    and must start with a letter or underscore. This prevents SQL injection
+    through malicious table names.
+
+    Args:
+        table_name: The table name to validate.
+
+    Returns:
+        The validated table name.
+
+    Raises:
+        ValueError: If the table name contains invalid characters.
+    """
+    if not re.match(r"^[a-zA-Z_][a-zA-Z0-9_]*$", table_name):
+        msg = (
+            f"Invalid table name '{table_name}'. "
+            "Table names must start with a letter or underscore and "
+            "contain only letters, numbers, and underscores."
+        )
+        raise ValueError(msg)
+    return table_name
+
+
 def infer_sqlite_type(field_type: Union[type, None]) -> str:
     """Infer the SQLite column type based on the Python type.
 

sqliter/model/model.py

@@ -26,7 +26,11 @@ from typing import (
 from pydantic import BaseModel, ConfigDict, Field
 from typing_extensions import Self
 
-from sqliter.helpers import from_unix_timestamp, to_unix_timestamp
+from sqliter.helpers import (
+    from_unix_timestamp,
+    to_unix_timestamp,
+    validate_table_name,
+)
 
 
 class SerializableField(Protocol):
@@ -123,32 +127,6 @@ class BaseDBModel(BaseModel):
 
         return cast("Self", cls.model_construct(**converted_obj))
 
-    @staticmethod
-    def _validate_table_name(table_name: str) -> str:
-        """Validate that a table name contains only safe characters.
-
-        Table names must contain only alphanumeric characters and underscores,
-        and must start with a letter or underscore. This prevents SQL injection
-        through malicious table names.
-
-        Args:
-            table_name: The table name to validate.
-
-        Returns:
-            The validated table name.
-
-        Raises:
-            ValueError: If the table name contains invalid characters.
-        """
-        if not re.match(r"^[a-zA-Z_][a-zA-Z0-9_]*$", table_name):
-            msg = (
-                f"Invalid table name '{table_name}'. "
-                "Table names must start with a letter or underscore and "
-                "contain only letters, numbers, and underscores."
-            )
-            raise ValueError(msg)
-        return table_name
-
     @classmethod
     def get_table_name(cls) -> str:
         """Get the database table name for the model.
@@ -171,7 +149,7 @@ class BaseDBModel(BaseModel):
         table_name: str | None = getattr(cls.Meta, "table_name", None)
         if table_name is not None:
             # Validate custom table names
-            return cls._validate_table_name(table_name)
+            return validate_table_name(table_name)
 
         # Get class name and remove 'Model' suffix if present
         class_name = cls.__name__.removesuffix("Model")
@@ -194,7 +172,7 @@ class BaseDBModel(BaseModel):
             )
 
         # Validate auto-generated table names (should always pass)
-        return cls._validate_table_name(table_name)
+        return validate_table_name(table_name)
 
     @classmethod
     def get_primary_key(cls) -> str:

sqliter/orm/__init__.py

@@ -10,7 +10,8 @@ Users can choose between modes via import:
 """
 
 from sqliter.orm.foreign_key import ForeignKey
+from sqliter.orm.m2m import ManyToMany
 from sqliter.orm.model import BaseDBModel
 from sqliter.orm.registry import ModelRegistry
 
-__all__ = ["BaseDBModel", "ForeignKey", "ModelRegistry"]
+__all__ = ["BaseDBModel", "ForeignKey", "ManyToMany", "ModelRegistry"]