Changeset 955:d98b211d32fa for src/mlx/pirep.py

Timestamp:

05/12/19 07:18:47 (5 years ago)

Author:

István Váradi <ivaradi@…>

Branch:

python3

hg-Phase:

(<MercurialRepository 1 'hg:/home/ivaradi/mlx/hg' '/'>, 'public')

Message:

Fixed pickling of PIREPs saved with Python 2 (re #347).

File:

• src/mlx/pirep.py (modified) (3 diffs)

src/mlx/pirep.py

@@ -2 +2 @@
 from .util import utf2unicode
 from .flight import Flight
+from .common import fixUnpickled
 
 from . import const
@@ -114 +115 @@
         try:
             with open(path, "rb") as f:
-                pirep = pickle.load(f)
+                pirep = pickle.load(f, fix_imports = True, encoding = "bytes")
                 if "numCrew" not in dir(pirep):
                     pirep.numCrew = pirep.bookedFlight.numCrew
@@ -412 +413 @@
 
         return ([], attrs)
+
+    def __setstate__(self, state):
+        """Set the state from the given unpickled dictionary."""
+        self.__dict__.update(fixUnpickled(state))