hgext/win32mbcs.py

7876:53c72ba36c2b

import os
from mercurial.i18n import _
from mercurial import util

def decode(arg):
   if isinstance(arg, str):
       uarg = arg.decode(util._encoding)
       if arg == uarg.encode(util._encoding):
           return uarg
       raise UnicodeError("Not local encoding")
   elif isinstance(arg, tuple):
       return tuple(map(decode, arg))
   elif isinstance(arg, list):
       return map(decode, arg)
   return arg

def encode(arg):
   if isinstance(arg, unicode):
       return arg.encode(util._encoding)
   elif isinstance(arg, tuple):
       return tuple(map(encode, arg))
   elif isinstance(arg, list):
       return map(encode, arg)
   return arg

def wrapper(func, args):
   # check argument is unicode, then call original
   for arg in args:
       if isinstance(arg, unicode):
           return func(*args)

   try:
       # convert arguments to unicode, call func, then convert back
       return encode(func(*decode(args)))
   except UnicodeError:
       # If not encoded with util._encoding, report it then
       # continue with calling original function.
      raise util.Abort(_("[win32mbcs] filename conversion fail with"
                         " %s encoding\n") % (util._encoding))

def wrapname(name):
   idx = name.rfind('.')
   module = name[:idx]
   name = name[idx+1:]
   module = eval(module)
   func = getattr(module, name)
   def f(*args):
       return wrapper(func, args)
   try:
      f.__name__ = func.__name__                # fail with python23
   except Exception:
      pass
   setattr(module, name, f)

# List of functions to be wrapped.
# NOTE: os.path.dirname() and os.path.basename() are safe because
#       they use result of os.path.split()
funcs = '''os.path.join os.path.split os.path.splitext

 hkscs cp932 932 ms932 mskanji ms-kanji shift_jis csshiftjis shiftjis
 sjis s_jis shift_jis_2004 shiftjis2004 sjis_2004 sjis2004
 shift_jisx0213 shiftjisx0213 sjisx0213 s_jisx0213'''

def reposetup(ui, repo):
   # TODO: decide use of config section for this extension
   if not os.path.supports_unicode_filenames:
       ui.warn(_("[win32mbcs] cannot activate on this platform.\n"))
       return

   # fake is only for relevant environment.
   if util._encoding.lower() in problematic_encodings.split():
       for f in funcs.split():
           wrapname(f)
       ui.debug(_("[win32mbcs] activated with encoding: %s\n") % util._encoding)

7877:eba7f12b0c51

import os
from mercurial.i18n import _
from mercurial import util

def decode(arg):
    if isinstance(arg, str):
        uarg = arg.decode(util._encoding)
        if arg == uarg.encode(util._encoding):
            return uarg
        raise UnicodeError("Not local encoding")
    elif isinstance(arg, tuple):
        return tuple(map(decode, arg))
    elif isinstance(arg, list):
        return map(decode, arg)
    return arg

def encode(arg):
    if isinstance(arg, unicode):
        return arg.encode(util._encoding)
    elif isinstance(arg, tuple):
        return tuple(map(encode, arg))
    elif isinstance(arg, list):
        return map(encode, arg)
    return arg

def wrapper(func, args):
    # check argument is unicode, then call original
    for arg in args:
        if isinstance(arg, unicode):
            return func(*args)

    try:
        # convert arguments to unicode, call func, then convert back
        return encode(func(*decode(args)))
    except UnicodeError:
        # If not encoded with util._encoding, report it then
        # continue with calling original function.
        raise util.Abort(_("[win32mbcs] filename conversion fail with"
                         " %s encoding\n") % (util._encoding))

def wrapname(name):
    idx = name.rfind('.')
    module = name[:idx]
    name = name[idx+1:]
    module = eval(module)
    func = getattr(module, name)
    def f(*args):
        return wrapper(func, args)
    try:
        f.__name__ = func.__name__                # fail with python23
    except Exception:
        pass
    setattr(module, name, f)

# List of functions to be wrapped.
# NOTE: os.path.dirname() and os.path.basename() are safe because
#       they use result of os.path.split()
funcs = '''os.path.join os.path.split os.path.splitext

 hkscs cp932 932 ms932 mskanji ms-kanji shift_jis csshiftjis shiftjis
 sjis s_jis shift_jis_2004 shiftjis2004 sjis_2004 sjis2004
 shift_jisx0213 shiftjisx0213 sjisx0213 s_jisx0213'''

def reposetup(ui, repo):
    # TODO: decide use of config section for this extension
    if not os.path.supports_unicode_filenames:
        ui.warn(_("[win32mbcs] cannot activate on this platform.\n"))
        return

    # fake is only for relevant environment.
    if util._encoding.lower() in problematic_encodings.split():
        for f in funcs.split():
            wrapname(f)
        ui.debug(_("[win32mbcs] activated with encoding: %s\n") % util._encoding)