diff env/lib/python3.7/site-packages/galaxy/util/aliaspickler.py @ 2:6af9afd405e9 draft

"planemo upload commit 0a63dd5f4d38a1f6944587f52a8cd79874177fc1"
author shellac
date Thu, 14 May 2020 14:56:58 -0400
parents 26e78fe6e8c4
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/env/lib/python3.7/site-packages/galaxy/util/aliaspickler.py	Thu May 14 14:56:58 2020 -0400
@@ -0,0 +1,31 @@
+import pickle
+
+from six.moves import cStringIO as StringIO
+
+
+class AliasUnpickler(pickle.Unpickler):
+    def __init__(self, aliases, *args, **kw):
+        pickle.Unpickler.__init__(self, *args, **kw)
+        self.aliases = aliases
+
+    def find_class(self, module, name):
+        module, name = self.aliases.get((module, name), (module, name))
+        return pickle.Unpickler.find_class(self, module, name)
+
+
+class AliasPickleModule(object):
+    def __init__(self, aliases):
+        self.aliases = aliases
+
+    def dump(self, obj, fileobj, protocol=0):
+        return pickle.dump(obj, fileobj, protocol)
+
+    def dumps(self, obj, protocol=0):
+        return pickle.dumps(obj, protocol)
+
+    def load(self, fileobj):
+        return AliasUnpickler(self.aliases, fileobj).load()
+
+    def loads(self, string):
+        fileobj = StringIO(string)
+        return AliasUnpickler(self.aliases, fileobj).load()