Fix error handling

smarbal · smarbal · commit 37cfdbbc09eb · 2022-01-19T16:41:04.000+01:00
diff --git a/codext/languages/tap.py b/codext/languages/tap.py
@@ -23,28 +23,33 @@ def build_encmap(map) :
             dict[map[i]] = "" + col * "." + " " + row * "."
             i += 1
     dict['k'] = dict['c']
-    dict[' '] = ''
     return dict
 
 def encode_tap(text, errors = 'strict') : 
     map = 'abcdefghijlmnopqrstuvwxyz'
     ENCMAP = build_encmap(map)
     encoded = ""
     for i, letter in enumerate(text) :
-        encoded += ENCMAP[letter.lower()]
+        try : 
+            encoded += ENCMAP[letter.lower()]
+        except KeyError : 
+            pass
         if i != len(text) - 1 and letter != ' ': 
             encoded += '  '
     return encoded, len(text)    
 
 
-def decode_tap(text, errors = 'strict') : 
+def decode_tap(text, errors = 'ignore') : 
     map = 'abcdefghijlmnopqrstuvwxyz'
     ENCMAP = build_encmap(map)
     decoded = ""
     for elem in text.split("  ") :
-        decoded += next(key for key, value in ENCMAP.items() if value == elem)
+        try : 
+            decoded += next(key for key, value in ENCMAP.items() if value == elem)
+        except StopIteration : 
+            print("Invalid character(s) in the input. This is what could be decoded :")
     return decoded, len(text)    
   
 
-add("tap", encode_tap, decode_tap, ignore_case="encode")
+add("tap", encode_tap, decode_tap, ignore_case="both")
 
