URL: https://github.com/freeipa/freeipa/pull/142
Author: dkupka
 Title: #142: CheckedIPAddress: Implement __(g|s)etstate__ and to ensure proper 
(un)pickling
Action: opened

PR body:
"""
Missing attributes in instance created by pickle.load cause AttributeError in
second part of ipa-server-install --external-ca.

https://fedorahosted.org/freeipa/ticket/6385
"""

To pull the PR as Git branch:
git remote add ghfreeipa https://github.com/freeipa/freeipa
git fetch ghfreeipa pull/142/head:pr142
git checkout pr142
From 3f096edf847af84077fc0e04fd16437678afaf2f Mon Sep 17 00:00:00 2001
From: David Kupka <dku...@redhat.com>
Date: Thu, 6 Oct 2016 13:31:52 +0200
Subject: [PATCH] CheckedIPAddress: Implement __(g|s)etstate__ and to ensure
 proper (un)pickling

Missing attributes in instance created by pickle.load cause AttributeError in
second part of ipa-server-install --external-ca.

https://fedorahosted.org/freeipa/ticket/6385
---
 ipapython/ipautil.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/ipapython/ipautil.py b/ipapython/ipautil.py
index 41544a1..f8badb0 100644
--- a/ipapython/ipautil.py
+++ b/ipapython/ipautil.py
@@ -205,6 +205,31 @@ def __init__(self, addr, match_local=False, parse_netmask=True,
 
         self.prefixlen = self._net.prefixlen
 
+    def __getstate__(self):
+        state = {
+            '_net': self._net,
+            'prefixlen': self.prefixlen
+        }
+        try:
+            state['super_state'] = super(CheckedIPAddress, self).__getstate__()
+        except AttributeError:
+            # none of base classes implements custom pickling
+            pass
+
+        return state
+
+    def __setstate__(self, state):
+        try:
+            super_state = state.pop('super_state')
+        except KeyError:
+            # no state saved for base classes
+            pass
+        else:
+            super(CheckedIPAddress, self).__setstate__(super_state)
+
+        self._net = state['_net']
+        self.prefixlen = state['prefixlen']
+
     def is_network_addr(self):
         return self == self._net.network
 
-- 
Manage your subscription for the Freeipa-devel mailing list:
https://www.redhat.com/mailman/listinfo/freeipa-devel
Contribute to FreeIPA: http://www.freeipa.org/page/Contribute/Code

Reply via email to