Added BE_INPUT_ENCODING and BE_OUTPUT_ENCODING to allow charset for stdin and stdout...
authorNiall Douglas (s [underscore] sourceforge {at} nedprod [dot] com) <spam@spamtrap.com>
Mon, 20 Feb 2012 15:34:59 +0000 (15:34 +0000)
committerW. Trevor King <wking@drexel.edu>
Sat, 25 Feb 2012 22:05:58 +0000 (17:05 -0500)
libbe/util/encoding.py

index ee51993a2cdace943028e0d90b7a65450761a534..a33678f84ad3cdab9d7a24251357bad668fe78ec 100644 (file)
@@ -34,6 +34,10 @@ if libbe.TESTING == True:
 
 ENCODING = os.environ.get('BE_ENCODING', None)
 "override get_encoding() output"
+INPUT_ENCODING = os.environ.get('BE_INPUT_ENCODING', None)
+"override get_input_encoding() output"
+OUTPUT_ENCODING = os.environ.get('BE_OUTPUT_ENCODING', None)
+"override get_output_encoding() output"
 
 def get_encoding():
     """
@@ -46,9 +50,13 @@ def get_encoding():
     return encoding
 
 def get_input_encoding():
+    if INPUT_ENCODING != None:
+        return INPUT_ENCODING
     return sys.__stdin__.encoding or get_encoding()
 
 def get_output_encoding():
+    if OUTPUT_ENCODING != None:
+        return OUTPUT_ENCODING
     return sys.__stdout__.encoding or get_encoding()
 
 def get_text_file_encoding():