async_pymongo: Allow accessing database and collection as an attribute (

#10) Signed-off-by: wulan17 <wulan17@nusantararom.org>
Mayuri-Chan · Oct 15, 2024 · 41f1d2b · 41f1d2b
1 parent dff9bd2
commit 41f1d2b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/async_pymongo/client.py b/async_pymongo/client.py
@@ -69,6 +69,9 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         # Propagate initialization to base
         super().__init__(dispatch)
 
+    def __getattr__(self, name: str) -> AsyncDatabase:
+        return AsyncDatabase(self, self.dispatch[name])
+
     def __getitem__(self, name: str) -> AsyncDatabase:
         return AsyncDatabase(self, self.dispatch[name])
 

diff --git a/async_pymongo/db.py b/async_pymongo/db.py
@@ -58,6 +58,9 @@ def __init__(self, client: "AsyncClient", database: Database) -> None:
     def __bool__(self) -> bool:
         return self.dispatch is not None
 
+    def __getattr__(self, name) -> AsyncCollection:
+        return AsyncCollection(self, name)
+
     def __getitem__(self, name) -> AsyncCollection:
         return AsyncCollection(self, name)