From b4f03173f33ed8d214ddc20b4791ec11677f6bb1 Mon Sep 17 00:00:00 2001
From: Gauvain Pocentek <gauvain@pocentek.net>
Date: Mon, 5 Feb 2018 14:21:14 +0100
Subject: Gitlab can be used as context manager

Fixes #371
---
 gitlab/__init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'gitlab')

diff --git a/gitlab/__init__.py b/gitlab/__init__.py
index 8a31a48..69629f8 100644
--- a/gitlab/__init__.py
+++ b/gitlab/__init__.py
@@ -147,6 +147,12 @@ class Gitlab(object):
                     manager = getattr(objects, cls_name)(self)
                     setattr(self, var_name, manager)
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        self.session.close()
+
     def __getstate__(self):
         state = self.__dict__.copy()
         state.pop('_objects')
-- 
cgit v1.2.1