From 0183639ce55b276a97e3ffae433fd84d17cf1aa7 Mon Sep 17 00:00:00 2001
From: shmookoff <shmookoff@gmail.com>
Date: Fri, 15 Apr 2022 19:16:14 +0500
Subject: [PATCH] Nesting context raise exception.

---
 aiogoogle/client.py | 28 +++++++++++++++++-----------
 1 file changed, 17 insertions(+), 11 deletions(-)

diff --git a/aiogoogle/client.py b/aiogoogle/client.py
index 036f947..f5dba8f 100644
--- a/aiogoogle/client.py
+++ b/aiogoogle/client.py
@@ -380,24 +380,30 @@ async def as_anon(self, *requests, timeout=None, full_res=False, raise_for_statu
             session_factory=self.session_factory
         )
 
-    def _ensure_session_set(self):
-        session = self.session_context.get()
-        if session is None:
-            session = self.session_factory()
-            self.session_context.set(session)
+    def _get_session(self):
+        return self.session_context.get()
+
+    def _set_session(self):
+        session = self.session_factory()
+        self.session_context.set(session)
         return session
 
     async def send(self, *args, **kwargs):
-        active_session = self._ensure_session_set()
-        return await active_session.send(*args, **kwargs)
+        session = self._get_session()
+        if session is None:
+            session = self._set_session()
+        return await session.send(*args, **kwargs)
 
     async def __aenter__(self):
-        session = self._ensure_session_set()
-        await session.__aenter__()
-        return self
+        session = self._get_session()
+        if session is None:
+            session = self._set_session()
+            await session.__aenter__()
+            return self
+        raise RuntimeError("Nesting context managers using the same Aiogoogle object is not allowed.")
 
     async def __aexit__(self, *args):
-        session = self.session_context.get()
+        session = self._get_session()
         await session.__aexit__(*args)
         # Had to add this because there's no use of keeping a closed session
         # Closed sessions cannot be reopened, so it's better to just get rid of the object