mercurial/revset.py

28394:dcb4209bd30d

# symbols which can't be used for a DoS attack for any given input
# (e.g. those which accept regexes as plain strings shouldn't be included)
# functions that just return a lot of changesets (like all) don't count here
safesymbols = set()

class predicate(registrar.funcregistrar):
    """Decorator to register revset predicate

    Usage::

        @predicate('mypredicate(arg1, arg2[, arg3])')
        def mypredicatefunc(repo, subset, x):
            '''Explanation of this revset predicate ....
            '''
            pass

    The first string argument of the constructor is used also in
    online help.

    Use 'extpredicate' instead of this to register revset predicate in
    extensions.
    """
    table = symbols
    formatdoc = "``%s``\n    %s"
    getname = registrar.funcregistrar.parsefuncdecl

    def __init__(self, decl, safe=False):
        """'safe' indicates whether a predicate is safe for DoS attack
        """
        super(predicate, self).__init__(decl)
        self.safe = safe

    def extraaction(self, name, func):
        if self.safe:
            safesymbols.add(name)

class extpredicate(registrar.delayregistrar):
    """Decorator to register revset predicate in extensions

    Usage::

    for name, func in registrarobj._table.iteritems():
        symbols[name] = func
        if func._safe:
            safesymbols.add(name)

# tell hggettext to extract docstrings from these functions:
i18nfunctions = symbols.values()

28395:0383f7a5e86c

# symbols which can't be used for a DoS attack for any given input
# (e.g. those which accept regexes as plain strings shouldn't be included)
# functions that just return a lot of changesets (like all) don't count here
safesymbols = set()

predicate = registrar.revsetpredicate()

class extpredicate(registrar.delayregistrar):
    """Decorator to register revset predicate in extensions

    Usage::

    for name, func in registrarobj._table.iteritems():
        symbols[name] = func
        if func._safe:
            safesymbols.add(name)

# load built-in predicates explicitly to setup safesymbols
loadpredicate(None, None, predicate)

# tell hggettext to extract docstrings from these functions:
i18nfunctions = symbols.values()