# HG changeset patch # User Durham Goode # Date 2015-05-22 19:47:18 # Node ID 660b178f49c78a66b4b50d8173ffb81bdb3e9425 # Parent 8c8af4d8aca3ac9fe40233d33ee35454fde57ea3 pathutil: add dirname and join functions This adds dirname and join functions to pathutil which are explicitly for handling '/' delimited paths. The implementations are basically copy paste from python's posix os.path.dirname and os.path.join functions. diff --git a/mercurial/pathutil.py b/mercurial/pathutil.py --- a/mercurial/pathutil.py +++ b/mercurial/pathutil.py @@ -187,3 +187,68 @@ def normasprefix(path): return path + os.sep else: return path + +def join(path, *paths): + '''Join two or more pathname components, inserting '/' as needed. + + Based on the posix os.path.join() implementation. + + >>> join('foo', 'bar') + 'foo/bar' + >>> join('/foo', 'bar') + '/foo/bar' + >>> join('foo', '/bar') + '/bar' + >>> join('foo', 'bar/') + 'foo/bar/' + >>> join('foo', 'bar', 'gah') + 'foo/bar/gah' + >>> join('foo') + 'foo' + >>> join('', 'foo') + 'foo' + >>> join('foo/', 'bar') + 'foo/bar' + >>> join('', '', '') + '' + >>> join ('foo', '', '', 'bar') + 'foo/bar' + ''' + sep = '/' + if not paths: + path[:0] + sep #23780: Ensure compatible data type even if p is null. + for piece in paths: + if piece.startswith(sep): + path = piece + elif not path or path.endswith(sep): + path += piece + else: + path += sep + piece + return path + +def dirname(path): + '''returns the directory portion of the given path + + Based on the posix os.path.split() implementation. + + >>> dirname('foo') + '' + >>> dirname('foo/') + 'foo' + >>> dirname('foo/bar') + 'foo' + >>> dirname('/foo') + '/' + >>> dirname('/foo/bar') + '/foo' + >>> dirname('/foo//bar/poo') + '/foo//bar' + >>> dirname('/foo//bar') + '/foo' + ''' + sep = '/' + i = path.rfind(sep) + 1 + dirname = path[:i] + if dirname and dirname != sep * len(dirname): + dirname = dirname.rstrip(sep) + return dirname