changeset 249:a065bf61caa1

encoding: fix encoding swap to work with demandimport enabled (#67)
author Daniel Tang <dytang@cs.purdue.edu>
date Fri, 10 Apr 2009 15:43:47 -0500
parents a9134fa28d15
children 79349fd04836
files util.py
diffstat 1 files changed, 5 insertions(+), 8 deletions(-) [+]
line wrap: on
line diff
--- a/util.py
+++ b/util.py
@@ -4,10 +4,6 @@ import shutil
 from mercurial import hg
 from mercurial import node
 from mercurial import util
-try:
-    from mercurial import encoding
-except ImportError:
-    encoding = None
 
 svn_subcommands = { }
 def register_subcommand(name):
@@ -143,10 +139,11 @@ def describe_commit(ui, h, b):
 
 def swap_out_encoding(new_encoding="UTF-8"):
     """ Utility for mercurial incompatibility changes, can be removed after 1.3"""
-    if encoding is None:
-        old = util._encoding
-        util._encoding = new_encoding
-    else:
+    try:
+        from mercurial import encoding
         old = encoding.encoding
         encoding.encoding = new_encoding
+    except ImportError:
+        old = util._encoding
+        util._encoding = new_encoding
     return old