diff options
author | Gauvain Pocentek <gauvain@pocentek.net> | 2018-02-05 14:21:14 +0100 |
---|---|---|
committer | Gauvain Pocentek <gauvain@pocentek.net> | 2018-02-05 14:22:40 +0100 |
commit | b4f03173f33ed8d214ddc20b4791ec11677f6bb1 (patch) | |
tree | 2bf8d58fff2068ace593c250e854eb9b4916f5cc /gitlab | |
parent | 29bd81336828b72a47673c76862cb4b532401766 (diff) | |
download | gitlab-b4f03173f33ed8d214ddc20b4791ec11677f6bb1.tar.gz |
Gitlab can be used as context manager
Fixes #371
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') |