comparison mercurial/fileset.py @ 27462:470ea34ba593

fileset: use decorator to mark a predicate as "existing caller"

This can localize changes for adding (or removing) an "existing
caller" predicate function in source code.

27461:afa76585c955

symbols = {}

# filesets using matchctx.status()
_statuscallers = []

def predicate(decl, callstatus=False):
    """Return a decorator for fileset predicate function

    'decl' argument is the declaration (including argument list like
    'adds(pattern)') or the name (for internal use only) of predicate.

    Optional 'callstatus' argument indicates whether predicate implies
    'matchctx.status()' at runtime or not (False, by default).
    """
    def decorator(func):
        i = decl.find('(')
        if i > 0:
            name = decl[:i]
        else:
            name = decl
        symbols[name] = func
        if callstatus:
            _statuscallers.append(name)
        if func.__doc__:
            func.__doc__ = "``%s``\n    %s" % (decl, func.__doc__.strip())
        return func
    return decorator

    l = getlist(x)
    if len(l) < min or len(l) > max:
        raise error.ParseError(err)
    return l

@predicate('binary()')
def binary(mctx, x):
    """File that appears to be binary (contains NUL bytes).
    """
    # i18n: "binary" is a keyword
    getargs(x, 0, 0, _("binary takes no arguments"))
    return [f for f in mctx.existing() if util.binary(mctx.ctx[f].data())]

@predicate('exec()')
def exec_(mctx, x):
    """File that is marked as executable.
    """
    # i18n: "exec" is a keyword
    getargs(x, 0, 0, _("exec takes no arguments"))
    return [f for f in mctx.existing() if mctx.ctx.flags(f) == 'x']

@predicate('symlink()')
def symlink(mctx, x):
    """File that is marked as a symlink.
    """
    # i18n: "symlink" is a keyword
    getargs(x, 0, 0, _("symlink takes no arguments"))

    # i18n: "portable" is a keyword
    getargs(x, 0, 0, _("portable takes no arguments"))
    checkwinfilename = util.checkwinfilename
    return [f for f in mctx.subset if checkwinfilename(f) is None]

@predicate('grep(regex)')
def grep(mctx, x):
    """File contains the given regular expression.
    """
    try:
        # i18n: "grep" is a keyword

        # no extension, this is a precise value
        return int(s)
    except ValueError:
        raise error.ParseError(_("couldn't parse size: %s") % s)

@predicate('size(expression)')
def size(mctx, x):
    """File size matches the given expression. Examples:

    - 1k (files from 1024 to 2047 bytes)
    - < 20k (files less than 20480 bytes)

    else:
        raise error.ParseError(_("couldn't parse size: %s") % expr)

    return [f for f in mctx.existing() if m(mctx.ctx[f].size())]

@predicate('encoding(name)')
def encoding(mctx, x):
    """File can be successfully decoded with the given character
    encoding. May not be useful for encodings other than ASCII and
    UTF-8.
    """

            continue
        s.append(f)

    return s

@predicate('eol(style)')
def eol(mctx, x):
    """File contains newlines of the given style (dos, unix, mac). Binary
    files are excluded, files with mixed line endings match multiple
    styles.
    """

        for s in tree[1:]:
            if _intree(funcs, s):
                return True
    return False

# filesets using matchctx.existing()
_existingcallers = [
    'binary',
    'encoding',
    'eol',
    'exec',
    'grep',
    'size',
    'symlink',
]

def getfileset(ctx, expr):
    tree = parse(expr)

    # do we need status info?
    if (_intree(_statuscallers, tree) or

27462:470ea34ba593

symbols = {}

# filesets using matchctx.status()
_statuscallers = []

# filesets using matchctx.existing()
_existingcallers = []

def predicate(decl, callstatus=False, callexisting=False):
    """Return a decorator for fileset predicate function

    'decl' argument is the declaration (including argument list like
    'adds(pattern)') or the name (for internal use only) of predicate.

    Optional 'callstatus' argument indicates whether predicate implies
    'matchctx.status()' at runtime or not (False, by default).

    Optional 'callexisting' argument indicates whether predicate
    implies 'matchctx.existing()' at runtime or not (False, by
    default).
    """
    def decorator(func):
        i = decl.find('(')
        if i > 0:
            name = decl[:i]
        else:
            name = decl
        symbols[name] = func
        if callstatus:
            _statuscallers.append(name)
        if callexisting:
            _existingcallers.append(name)
        if func.__doc__:
            func.__doc__ = "``%s``\n    %s" % (decl, func.__doc__.strip())
        return func
    return decorator

    l = getlist(x)
    if len(l) < min or len(l) > max:
        raise error.ParseError(err)
    return l

@predicate('binary()', callexisting=True)
def binary(mctx, x):
    """File that appears to be binary (contains NUL bytes).
    """
    # i18n: "binary" is a keyword
    getargs(x, 0, 0, _("binary takes no arguments"))
    return [f for f in mctx.existing() if util.binary(mctx.ctx[f].data())]

@predicate('exec()', callexisting=True)
def exec_(mctx, x):
    """File that is marked as executable.
    """
    # i18n: "exec" is a keyword
    getargs(x, 0, 0, _("exec takes no arguments"))
    return [f for f in mctx.existing() if mctx.ctx.flags(f) == 'x']

@predicate('symlink()', callexisting=True)
def symlink(mctx, x):
    """File that is marked as a symlink.
    """
    # i18n: "symlink" is a keyword
    getargs(x, 0, 0, _("symlink takes no arguments"))

    # i18n: "portable" is a keyword
    getargs(x, 0, 0, _("portable takes no arguments"))
    checkwinfilename = util.checkwinfilename
    return [f for f in mctx.subset if checkwinfilename(f) is None]

@predicate('grep(regex)', callexisting=True)
def grep(mctx, x):
    """File contains the given regular expression.
    """
    try:
        # i18n: "grep" is a keyword

        # no extension, this is a precise value
        return int(s)
    except ValueError:
        raise error.ParseError(_("couldn't parse size: %s") % s)

@predicate('size(expression)', callexisting=True)
def size(mctx, x):
    """File size matches the given expression. Examples:

    - 1k (files from 1024 to 2047 bytes)
    - < 20k (files less than 20480 bytes)

    else:
        raise error.ParseError(_("couldn't parse size: %s") % expr)

    return [f for f in mctx.existing() if m(mctx.ctx[f].size())]

@predicate('encoding(name)', callexisting=True)
def encoding(mctx, x):
    """File can be successfully decoded with the given character
    encoding. May not be useful for encodings other than ASCII and
    UTF-8.
    """

            continue
        s.append(f)

    return s

@predicate('eol(style)', callexisting=True)
def eol(mctx, x):
    """File contains newlines of the given style (dos, unix, mac). Binary
    files are excluded, files with mixed line endings match multiple
    styles.
    """

        for s in tree[1:]:
            if _intree(funcs, s):
                return True
    return False

def getfileset(ctx, expr):
    tree = parse(expr)

    # do we need status info?
    if (_intree(_statuscallers, tree) or