diff options
| author | bourgesl <bourges.laurent@gmail.com> | 2019-07-15 13:45:23 +0200 |
|---|---|---|
| committer | Max Wittig <max.wittig95@gmail.com> | 2019-07-21 21:01:36 +0200 |
| commit | b4b5decb7e49ac16d98d56547a874fb8f9d5492b (patch) | |
| tree | 14fa3b363bb3c8c4c6d08a3dff28997a8728b01f /gitlab | |
| parent | 262b222000dad30fc6dfc63ccf2fa200eba09662 (diff) | |
| download | gitlab-b4b5decb7e49ac16d98d56547a874fb8f9d5492b.tar.gz | |
fix: improve pickle support
Diffstat (limited to 'gitlab')
| -rw-r--r-- | gitlab/base.py | 2 | ||||
| -rw-r--r-- | gitlab/tests/test_base.py | 1 |
2 files changed, 2 insertions, 1 deletions
diff --git a/gitlab/base.py b/gitlab/base.py index d2e44b8..f81d039 100644 --- a/gitlab/base.py +++ b/gitlab/base.py @@ -52,7 +52,7 @@ class RESTObject(object): def __setstate__(self, state): module_name = state.pop("_module_name") self.__dict__.update(state) - self._module = importlib.import_module(module_name) + self.__dict__["_module"] = importlib.import_module(module_name) def __getattr__(self, name): try: diff --git a/gitlab/tests/test_base.py b/gitlab/tests/test_base.py index 2526bee..47eda6c 100644 --- a/gitlab/tests/test_base.py +++ b/gitlab/tests/test_base.py @@ -93,6 +93,7 @@ class TestRESTObject(unittest.TestCase): self.assertIsInstance(unpickled, FakeObject) self.assertTrue(hasattr(unpickled, "_module")) self.assertEqual(unpickled._module, original_obj_module) + pickled2 = pickle.dumps(unpickled) def test_attrs(self): obj = FakeObject(self.manager, {"foo": "bar"}) |