Implemented Context Manager for Requests

mozilla · Feb 3, 2020 · a444cc5 · a444cc5
1 parent 9f44e8f
commit a444cc5
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/agithub/base.py b/agithub/base.py
@@ -430,3 +430,19 @@ def _filterEmptyHeaders(self, headers):
                 newHeaders[header] = headers[header]
 
         return newHeaders
+
+class connectionManager(object):
+    '''
+    Context manager for handling connections in client.request
+    '''
+    def __init__(self, client):
+        self.conn = client.get_connection()
+
+    def __enter__(self):
+        return self.conn
+
+    def __exit__(self, exc_type, exc_value, traceback): 
+        self.conn.close()
+
+
+