formatters: use locale.getpreferredencoding() instead of utf-8

Signed-off-by: Matthieu Weber <mweber+weboob@free.fr>
Signed-off-by: Romain Bignon <romain@symlink.me>
This commit is contained in:
Matthieu Weber 2015-05-13 21:16:06 +03:00 committed by Romain Bignon
commit 95cd12fd22
2 changed files with 5 additions and 3 deletions

View file

@ -22,6 +22,7 @@ from __future__ import print_function
import os import os
import sys import sys
import locale
import subprocess import subprocess
try: try:
@ -121,7 +122,7 @@ class IFormatter(object):
def output(self, formatted): def output(self, formatted):
if self.outfile != sys.stdout: if self.outfile != sys.stdout:
with open(self.outfile, "a+") as outfile: with open(self.outfile, "a+") as outfile:
outfile.write(formatted.encode('utf-8') + os.linesep) outfile.write(formatted.encode(locale.getpreferredencoding(), errors='replace') + os.linesep)
else: else:
for line in formatted.split('\n'): for line in formatted.split('\n'):
@ -133,7 +134,7 @@ class IFormatter(object):
self.print_lines = 0 self.print_lines = 0
if isinstance(line, unicode): if isinstance(line, unicode):
line = line.encode('utf-8') line = line.encode(locale.getpreferredencoding(), errors='replace')
print(line) print(line)
self.print_lines += 1 self.print_lines += 1

View file

@ -17,6 +17,7 @@
# You should have received a copy of the GNU Affero General Public License # You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>. # along with weboob. If not, see <http://www.gnu.org/licenses/>.
import locale
from prettytable import PrettyTable from prettytable import PrettyTable
@ -40,7 +41,7 @@ class TableFormatter(IFormatter):
def flush(self): def flush(self):
s = self.get_formatted_table() s = self.get_formatted_table()
if s is not None: if s is not None:
self.output(s.encode('utf-8')) self.output(s.encode(locale.getpreferredencoding(), errors='replace'))
def get_formatted_table(self): def get_formatted_table(self):
if len(self.queue) == 0: if len(self.queue) == 0: