comparison mercurial/registrar.py @ 32377:ec84db232fc2

registrar: switch @command decorator to class

It overrides _funcregistrarbase._doregister() since the structure of the
command table is quite different.

32376:46ba2cdda476

    def _extrasetup(self, name, func):
        """Execute exra setup for registered function, if needed
        """
        pass

def command(table):
    """Returns a function object to be used as a decorator for making commands.

    This function receives a command table as its argument. The table should
    be a dict.

    The returned function can be used as a decorator for adding commands
    to that command table. This function accepts multiple arguments to define
    a command.

    The first argument is the command name.

    The options argument is an iterable of tuples defining command arguments.
    See ``mercurial.fancyopts.fancyopts()`` for the format of each tuple.

    The inferrepo argument defines whether to try to find a repository from the
    command line arguments. If True, arguments will be examined for potential
    repository locations. See ``findrepo()``. If a repository is found, it
    will be used.
    """
    def cmd(name, options=(), synopsis=None, norepo=False, optionalrepo=False,
            inferrepo=False):
        def decorator(func):
            func.norepo = norepo
            func.optionalrepo = optionalrepo
            func.inferrepo = inferrepo
            if synopsis:
                table[name] = func, list(options), synopsis
            else:
                table[name] = func, list(options)
            return func
        return decorator

    return cmd

class revsetpredicate(_funcregistrarbase):
    """Decorator to register revset predicate

    Usage::

32377:ec84db232fc2

    def _extrasetup(self, name, func):
        """Execute exra setup for registered function, if needed
        """
        pass

class command(_funcregistrarbase):
    """Decorator to register a command function to table

    This class receives a command table as its argument. The table should
    be a dict.

    The created object can be used as a decorator for adding commands to
    that command table. This accepts multiple arguments to define a command.

    The first argument is the command name.

    The options argument is an iterable of tuples defining command arguments.
    See ``mercurial.fancyopts.fancyopts()`` for the format of each tuple.

    The inferrepo argument defines whether to try to find a repository from the
    command line arguments. If True, arguments will be examined for potential
    repository locations. See ``findrepo()``. If a repository is found, it
    will be used.
    """

    def _doregister(self, func, name, options=(), synopsis=None,
                    norepo=False, optionalrepo=False, inferrepo=False):
        if True:
            func.norepo = norepo
            func.optionalrepo = optionalrepo
            func.inferrepo = inferrepo
            if synopsis:
                self._table[name] = func, list(options), synopsis
            else:
                self._table[name] = func, list(options)
            return func

class revsetpredicate(_funcregistrarbase):
    """Decorator to register revset predicate

    Usage::