Added first_or_404

Author: fantix

gino/ext/sanic.py

@@ -1,8 +1,10 @@
 # noinspection PyPackageRequirements
 from sanic.exceptions import NotFound
 
-from ..api import Gino as _Gino
+from ..api import Gino as _Gino, GinoExecutor as _Executor
 from ..local import enable_task_local, disable_task_local
+from ..connection import GinoConnection as _Connection
+from ..pool import GinoPool as _Pool
 
 
 class SanicModelMixin:
@@ -15,6 +17,33 @@ async def get_or_404(cls, *args, **kwargs):
         return rv
 
 
+# noinspection PyClassHasNoInit
+class GinoExecutor(_Executor):
+    async def first_or_404(self, *args, **kwargs):
+        rv = await self.first(*args, **kwargs)
+        if rv is None:
+            raise NotFound('No such data')
+        return rv
+
+
+# noinspection PyClassHasNoInit
+class GinoConnection(_Connection):
+    async def first_or_404(self, *args, **kwargs):
+        rv = await self.first(*args, **kwargs)
+        if rv is None:
+            raise NotFound('No such data')
+        return rv
+
+
+# noinspection PyClassHasNoInit
+class GinoPool(_Pool):
+    async def first_or_404(self, *args, **kwargs):
+        rv = await self.first(*args, **kwargs)
+        if rv is None:
+            raise NotFound('No such data')
+        return rv
+
+
 # noinspection PyClassHasNoInit
 class Gino(_Gino):
     """Support Sanic web server.
@@ -34,7 +63,10 @@ class Gino(_Gino):
     Here `request['connection']` is a :class:`LazyConnection` object, see its
     doc string for more information.
     """
-    default_model_classes = _Gino.default_model_classes + (SanicModelMixin,)
+    model_base_classes = _Gino.model_base_classes + (SanicModelMixin,)
+    query_executor = GinoExecutor
+    connection_cls = GinoConnection
+    pool_cls = GinoPool
 
     def init_app(self, app):
         task_local_enabled = [False]
@@ -75,3 +107,9 @@ async def after_server_stop(_, loop):
             if task_local_enabled[0]:
                 disable_task_local(loop)
                 task_local_enabled[0] = False
+
+    async def first_or_404(self, *args, **kwargs):
+        rv = await self.first(*args, **kwargs)
+        if rv is None:
+            raise NotFound('No such data')
+        return rv