diff --git a/export_saved.py b/export_saved.py index 7e4f5db..b390169 100755 --- a/export_saved.py +++ b/export_saved.py @@ -8,7 +8,7 @@ from time import time import argparse -import csv +import unicodecsv as csv import logging import sys @@ -67,7 +67,7 @@ def convert(self): ifile = open(self._html_file, 'wb') try: ifile.write(content) - except TypeError: + except (UnicodeEncodeError, TypeError): ifile.write(content.encode('utf-8', 'ignore')) @@ -251,7 +251,7 @@ def write_csv(csv_rows, file_name=None): csvwriter.writerow(row) except (UnicodeEncodeError, TypeError) as e: with open(file_name, "w") as f: - csvwriter = csv.writer(f, delimiter=delimiter, quoting=csv.QUOTE_MINIMAL) + csvwriter = csv.writer(f, delimiter=delimiter, quoting=csv.QUOTE_MINIMAL, encoding='utf-8') csvwriter.writerow(csv_fields) for row in csv_rows: try: diff --git a/requirements.txt b/requirements.txt index 94ebe0a..1d43adb 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,2 @@ praw==4.4.0 +unicodecsv==0.14.1 \ No newline at end of file