Use set_header to replace header's value in-place

[mimedecode.git] / mimedecode.py

diff --git a/mimedecode.py b/mimedecode.py
index e63aedcc8664c82bdf89ae915d3ac59666e2c4a0..01f650563060eb085b6811295a8ee8ce267c2744 100755 (executable)
--- a/mimedecode.py
+++ b/mimedecode.py
@@ -74,9 +74,6 @@ def decode_header(msg, header):
             set_header(msg, header, new_value)
 
 
-def _decode_header_param(s):
-    return recode_if_needed(s[2], s[0])
-
 def decode_header_param(msg, header, param):
     "Decode mail header's parameter (if exists) and put it back, if it was encoded"
 
@@ -84,7 +81,7 @@ def decode_header_param(msg, header, param):
         value = msg.get_param(param, header=header)
         if value:
             if isinstance(value, tuple):
-                new_value = _decode_header_param(value)
+                new_value = recode_if_needed(value[2], value[0])
             else:
                 new_value = _decode_header(value)
             if new_value <> value: # do not bother to touch msg if not changed
@@ -568,7 +565,7 @@ if __name__ == "__main__":
     msg = email.message_from_file(infile)
 
     for header, value in g.set_header_value:
-        msg[header] = value
+        set_header(msg, header, value)
 
     for header, param, value in g.set_header_param:
         if header in msg: