pickle isnt needed thats py2

NorthIsUp · Apr 5, 2022 · 8ed832f · 8ed832f
1 parent 611d12b
commit 8ed832f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 12 deletions.
diff --git a/workflow/background.py b/workflow/background.py
@@ -18,12 +18,7 @@
 
 
 import os
-
-try:
-    import cPickle as pickle
-except:
-    import pickle
-
+import pickle
 import signal
 import subprocess
 import sys

diff --git a/workflow/workflow.py b/workflow/workflow.py
@@ -25,12 +25,7 @@
 import logging
 import logging.handlers
 import os
-
-try:
-    import cPickle as pickle
-except:
-    import pickle
-
+import pickle
 import plistlib
 import re
 import shutil
@@ -1605,6 +1600,8 @@ def stored_data(self, name):
                 "to load this data.".format(serializer_name)
             )
 
+        self.logger.debug("data `%s` stored as `%s`", name, serializer_name)
+
         filename = "{0}.{1}".format(name, serializer_name)
         data_path = self.datafile(filename)
 
@@ -1618,6 +1615,8 @@ def stored_data(self, name):
         with open(data_path, "rb") as file_obj:
             data = serializer.load(file_obj)
 
+            self.logger.debug("stored data loaded: %s", data_path)
+
         return data
 
     def store_data(self, name, data, serializer=None):
@@ -1687,6 +1686,8 @@ def _store():
 
         _store()
 
+        self.logger.debug("saved data: %s", data_path)
+
     def cached_data(self, name, data_func=None, max_age=60):
         """Return cached data if younger than ``max_age`` seconds.