asyncer integration

Author: felipao-mx

mongoengine_plus/aio/async_document.py

@@ -1,7 +1,7 @@
+from asyncer import asyncify
 from mongoengine import Document
 
 from .async_query_set import AsyncQuerySet
-from .utils import create_awaitable
 
 
 class AsyncDocument(Document):
@@ -11,12 +11,10 @@ class AsyncDocument(Document):
     )
 
     async def async_save(self, *args, **kwargs):
-        return await create_awaitable(self.save, *args, **kwargs)
+        return await asyncify(self.save)(*args, **kwargs)
 
     async def async_reload(self, *fields, **kwargs):
-        return await create_awaitable(self.reload, *fields, **kwargs)
+        return await asyncify(self.reload)(*fields, **kwargs)
 
     async def async_delete(self, signal_kwargs=None, **write_concern):
-        return await create_awaitable(
-            self.delete, signal_kwargs, **write_concern
-        )
+        return await asyncify(self.delete)(signal_kwargs, **write_concern)

mongoengine_plus/aio/async_query_set.py

@@ -1,23 +1,22 @@
+from asyncer import asyncify
 from mongoengine import QuerySet
 
-from .utils import create_awaitable
-
 
 class AsyncQuerySet(QuerySet):
     async def async_first(self):
-        return await create_awaitable(self.first)
+        return await asyncify(self.first)()
 
     async def async_get(self, *q_objs, **query):
-        return await create_awaitable(self.get, *q_objs, **query)
+        return await asyncify(self.get)(*q_objs, **query)
 
     async def async_count(self, with_limit_and_skip=False):
-        return await create_awaitable(self.count, with_limit_and_skip)
+        return await asyncify(self.count)(with_limit_and_skip)
 
     async def async_to_list(self):
-        return await create_awaitable(list, self)
+        return await asyncify(list)(self)
 
     async def async_update(self, *u_objs, **query):
-        return await create_awaitable(self.update, *u_objs, **query)
+        return await asyncify(self.update)(*u_objs, **query)
 
     async def async_insert(
         self,
@@ -26,13 +25,13 @@ async def async_insert(
         write_concern=None,
         signal_kwargs=None,
     ):
-        return await create_awaitable(
-            self.insert, doc_or_docs, load_bulk, write_concern, signal_kwargs
+        return await asyncify(self.insert)(
+            doc_or_docs, load_bulk, write_concern, signal_kwargs
         )
 
     async def async_delete(
         self, write_concern=None, _from_doc_delete=False, cascade_refs=None
     ):
-        return await create_awaitable(
-            self.delete, write_concern, _from_doc_delete, cascade_refs
+        return await asyncify(self.delete)(
+            write_concern, _from_doc_delete, cascade_refs
         )

mongoengine_plus/aio/utils.py

@@ -1 +1 @@
-__version__ = '1.0.1'
+__version__ = '1.1.0.dev0'

mongoengine_plus/version.py

@@ -1,3 +1,4 @@
+asyncer==0.0.8
 blinker==1.9.0
 dnspython==2.7.0
 mongoengine==0.29.1