adding __await__ magic method to Query

Author: dantownsend

piccolo/query/base.py

@@ -87,6 +87,13 @@ def _validate(self):
         """
         pass
 
+    def __await__(self):
+        """
+        If the user doesn't explicity call .run(), proxy to it as a
+        convenience.
+        """
+        return self.run().__await__()
+
     async def run(self, in_pool=True):
         self._validate()
 

tests/query/test_await.py

@@ -0,0 +1,18 @@
+import asyncio
+
+from ..base import DBTestCase
+from ..example_project.tables import Band
+
+
+class TestAwait(DBTestCase):
+    def test_await(self):
+        """
+        Test awaiting a query directly - it should proxy to Query.run().
+        """
+
+        async def get_all():
+            return await Band.select()
+
+        response = asyncio.run(get_all())
+
+        self.assertIsInstance(response, list)