update9to10: make unpickling recurrence rules more robust

agateau · Aug 16, 2024 · 011f413 · 011f413
1 parent 0b058f6
commit 011f413
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/yokadi/update/update9to10.py b/yokadi/update/update9to10.py
@@ -35,7 +35,11 @@ def createByweekdayValue(rule):
 
 
 def createJsonStringFromRule(pickledRule):
-    rule = pickle.loads(pickledRule)
+    try:
+        rule = pickle.loads(pickledRule)
+    except UnicodeDecodeError:
+        # Some rules fails to unpickle if we don't set encoding to latin1
+        rule = pickle.loads(pickledRule, encoding="latin1")
     dct = {}
     dct["freq"] = rule._freq
     dct["bymonth"] = tuplify(rule._bymonth)