fix User model's manager + bumping version

Author: rednaks

README.md

@@ -114,6 +114,7 @@ This is an on going projects, not all model methods are ported.
 | `Model.objects.async_ordered`            | ✅ |  |
 | `__aiter__`                              | ✅ |  |
 | `__repr__`                               | ✅ |  |
+| `Model.objects.async_iterator            | ❌ |  |
 
 
 ### Model:
@@ -126,11 +127,12 @@ This is an on going projects, not all model methods are ported.
 | `...`                                   | ❌ |  |
 
 
-### User Model 
+### User Model / Manager
 | methods                    | supported  | comments |
 |----------------------------|------------|----------|
-| `UserModel.is_authenticated`            | ❌ |  |
-| `UserModel.is_super_user`               | ❌ |  |
+| `UserModel.is_authenticated`            | ✅ |  |
+| `UserModel.is_super_user`               | ✅ |  |
+| `UserModel.objects.async_create_user`   | ❌ |  |
 | `...`                                   | ❌ |  |
 
 

django_async_orm/apps.py

@@ -1,7 +1,7 @@
 import logging
 from django.apps import AppConfig, apps
 
-from django_async_orm.manager import AsyncManager
+from django_async_orm.utils import patch_manager
 
 
 class AsyncOrmConfig(AppConfig):
@@ -10,5 +10,4 @@ class AsyncOrmConfig(AppConfig):
     def ready(self):
         logging.info('AsyncORM: patching models')
         for model in apps.get_models(include_auto_created=True):
-            model.objects = AsyncManager()
-            model.objects.model = model
+            patch_manager(model)

django_async_orm/utils.py

@@ -1,5 +1,8 @@
 import asyncio
 
+from django_async_orm.manager import AsyncManager
+
+
 class AsyncIter:
     def __init__(self, iterable):
         self._iter = iter(iterable)
@@ -14,3 +17,22 @@ async def __anext__(self):
             raise StopAsyncIteration
         await asyncio.sleep(0)
         return element
+
+
+def async_user_manager_factory():
+    from django.contrib.auth.models import UserManager
+
+    class AsyncUserManager(UserManager, AsyncManager):
+        pass
+
+    return AsyncUserManager
+
+
+def patch_manager(model):
+    from django.contrib.auth.models import UserManager
+    async_manager_cls = AsyncManager
+    if isinstance(model.objects, UserManager):
+        async_manager_cls = async_user_manager_factory()
+
+    model.objects = async_manager_cls()
+    model.objects.model = model

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-async-orm"
-version = "0.1.2"
+version = "0.1.3"
 description = "Bringing async capabilities to django ORM"
 authors = ["SkanderBM <[email protected]>"]
 license = "MIT"