Add IntegerIdMixin and export related classes

fastapi_users/__init__.py

@@ -7,6 +7,9 @@ from fastapi_users.fastapi_users import FastAPIUsers  # noqa: F401
 from fastapi_users.manager import (  # noqa: F401
     BaseUserManager,
     InvalidPasswordException,
+    InvalidID,
+    UUIDIDMixin,
+    IntegerIDMixin,
 )
 
 __all__ = [
@@ -14,4 +17,7 @@ __all__ = [
     "FastAPIUsers",
     "BaseUserManager",
     "InvalidPasswordException",
+    "InvalidID",
+    "UUIDIDMixin",
+    "IntegerIDMixin",
 ]

fastapi_users/manager.py

@@ -612,4 +612,14 @@ class UUIDIDMixin:
             raise InvalidID() from e
 
 
+class IntegerIDMixin:
+    def parse_id(self, value: Any) -> int:
+        if isinstance(value, float):
+            raise InvalidID()
+        try:
+            return int(value)
+        except ValueError as e:
+            raise InvalidID() from e
+
+
 UserManagerDependency = DependencyCallable[BaseUserManager[models.UP, models.ID]]

tests/test_manager.py

@@ -7,6 +7,8 @@ from pytest_mock import MockerFixture
 
 from fastapi_users.jwt import decode_jwt, generate_jwt
 from fastapi_users.manager import (
+    IntegerIDMixin,
+    InvalidID,
     InvalidPasswordException,
     InvalidResetPasswordToken,
     InvalidVerifyToken,
@@ -591,3 +593,19 @@ class TestAuthenticate:
         assert user is not None
         assert user.email == "king.arthur@camelot.bt"
         assert update_spy.called is True
+
+
+def test_integer_id_mixin():
+    integer_id_mixin = IntegerIDMixin()
+
+    assert integer_id_mixin.parse_id("123") == 123
+    assert integer_id_mixin.parse_id(123) == 123
+
+    with pytest.raises(InvalidID):
+        integer_id_mixin.parse_id("123.42")
+
+    with pytest.raises(InvalidID):
+        integer_id_mixin.parse_id(123.42)
+
+    with pytest.raises(InvalidID):
+        integer_id_mixin.parse_id("abc")