fix: [yaml] add one more workaround for python 3

ssato · Mar 12, 2017 · f4beedc · f4beedc
1 parent ba6f92e
commit f4beedc
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/anyconfig/backend/yaml.py b/anyconfig/backend/yaml.py
@@ -108,12 +108,14 @@ def container_representer(dumper, data, mapping_tag=_MAPPING_TAG):
         """
         return dumper.represent_mapping(mapping_tag, data.items())
 
-    if not anyconfig.compat.IS_PYTHON_3:
-        def ustr_representer(dumper, data):
-            tag = "tag:yaml.org,2002:python/unicode"
-            return dumper.represent_scalar(tag, data)
+    def ustr_representer(dumper, data):
+        tag = "tag:yaml.org,2002:python/unicode"
+        return dumper.represent_scalar(tag, data)
 
+    try:
         dumper.add_representer(unicode, ustr_representer)
+    except NameError:
+        pass
 
     if type(container) != dict:
         dumper.add_representer(container, container_representer)