Encode filename to system locale

Author: ids1024

pafy/backend_shared.py

@@ -22,6 +22,7 @@
 from . import __version__, g
 from .pafy import call_gdata
 from .playlist import get_playlist2
+from .util import xenc
 
 dbg = logging.debug
 
@@ -454,7 +455,7 @@ def generate_filename(self, meta=False, max_length=None):
                 filename = filename[:max_length-3] + '...'
 
         filename += "." + self.extension
-        return filename
+        return xenc(filename)
 
     @property
     def rawbitrate(self):

pafy/util.py

@@ -1,5 +1,6 @@
 import json
 import sys
+import os
 
 if sys.version_info[:2] >= (3, 0):
     # pylint: disable=E0611,F0401,I0011
@@ -12,6 +13,11 @@
 
 from . import g
 
+
+mswin = os.name == "nt"
+not_utf8_environment = mswin or "UTF-8" not in sys.stdout.encoding
+
+
 class GdataError(Exception):
     """Gdata query failed."""
     pass
@@ -35,3 +41,22 @@ def call_gdata(api, qs):
         raise GdataError(errmsg)
 
     return json.loads(data)
+
+
+def utf8_replace(txt):
+    """
+    Replace unsupported characters in unicode string.
+
+    :param txt: text to filter
+    :type txt: str
+    :returns: Unicode text without any characters unsupported by locale
+    :rtype: str
+    """
+    sse = sys.stdout.encoding
+    txt = txt.encode(sse, "replace").decode(sse)
+    return txt
+
+
+def xenc(stuff):
+    """ Replace unsupported characters. """
+    return utf8_replace(stuff) if not_utf8_environment else stuff