diff options
Diffstat (limited to 'gitlab')
-rw-r--r-- | gitlab/__init__.py | 6 |
1 files changed, 6 insertions, 0 deletions
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') |