comparison mercurial/cmdutil.py @ 50093:ce60c8d4ac87

typing: add type hints to argument checking functions in cmdutil

These might be surprising, since they can take strings instead of bytes.  The
way `AnyStr` works is that it must be all bytes or all str for any given
invocation.

The wildcard here will be the `opts` that get passed in- if the type is unknown
and defaults to `Any`, there's no enforcement that the dict key type matches the
additional args.  But a lot of uses should be using `**opts` from the command
method, which has a str key.  The uses of these methods in this module are now
typed because their internals force a specific type, and it can't just be
inferred from the caller.

50091:a8d71a6ba205

import copy as copymod
import errno
import os
import re

from .i18n import _
from .node import (
    hex,
    nullrev,

from .revlogutils import (
    constants as revlog_constants,
)

if pycompat.TYPE_CHECKING:
    from typing import (
        Any,
        Dict,
    )

    for t in (Any, Dict):
        assert t

stringio = util.stringio

# templates of common command options

# special string such that everything below this line will be ingored in the
# editor text
_linebelow = b"^HG: ------------------------ >8 ------------------------$"


def check_at_most_one_arg(opts, *args):
    """abort if more than one of the arguments are in opts

    Returns the unique argument or None if none of them were specified.
    """

    def to_display(name):
        return pycompat.sysbytes(name).replace(b'_', b'-')

    previous = None
    for x in args:
        if opts.get(x):

                )
            previous = x
    return previous


def check_incompatible_arguments(opts, first, others):
    """abort if the first argument is given along with any of the others

    Unlike check_at_most_one_arg(), `others` are not mutually exclusive
    among themselves, and they're passed as a single collection.
    """
    for other in others:
        check_at_most_one_arg(opts, first, other)


def resolve_commit_options(ui, opts):
    """modify commit options dict to handle related options

    The return value indicates that ``rewrite.update-timestamp`` is the reason
    the ``date`` option is set.
    """

        opts['user'] = ui.username()

    return datemaydiffer


def check_note_size(opts):
    """make sure note is of valid format"""

    note = opts.get('note')
    if not note:
        return

    ctx = repo[None]
    for s in sorted(ctx.substate):
        ctx.sub(s).bailifchanged(hint=hint)


def logmessage(ui, opts):
    """get the log message according to -m and -l option"""

    check_at_most_one_arg(opts, b'message', b'logfile')

    message = opts.get(b'message')
    logfile = opts.get(b'logfile')

    if not message and logfile:
        try:
            if isstdiofilename(logfile):

    commands.
    """
    return openstorage(repo, cmd, file_, opts, returnrevlog=True)


def copy(ui, repo, pats, opts, rename=False):
    check_incompatible_arguments(opts, b'forget', [b'dry_run'])

    # called with the repo lock held
    #
    # hgsep => pathname that uses "/" to separate directories

            if not sub.cat(
                submatch,
                basefm,
                fntemplate,
                subprefix,
                **pycompat.strkwargs(opts)
            ):
                err = 0
        except error.RepoLookupError:
            ui.status(
                _(b"skipping missing subrepository: %s\n") % uipathfn(subpath)

            return False
    else:
        return f not in ctx2.manifest()


def amend(ui, repo, old, extra, pats, opts):
    # avoid cycle context -> subrepo -> cmdutil
    from . import context

    # amend will reuse the existing user if not specified, but the obsolete
    # marker creation requires that the current user's name is specified.

50093:ce60c8d4ac87

import copy as copymod
import errno
import os
import re

from typing import (
    Any,
    AnyStr,
    Dict,
    Iterable,
    Optional,
    cast,
)

from .i18n import _
from .node import (
    hex,
    nullrev,

from .revlogutils import (
    constants as revlog_constants,
)

if pycompat.TYPE_CHECKING:
    from . import (
        ui as uimod,
    )

stringio = util.stringio

# templates of common command options

# special string such that everything below this line will be ingored in the
# editor text
_linebelow = b"^HG: ------------------------ >8 ------------------------$"


def check_at_most_one_arg(
    opts: Dict[AnyStr, Any],
    *args: AnyStr,
) -> Optional[AnyStr]:
    """abort if more than one of the arguments are in opts

    Returns the unique argument or None if none of them were specified.
    """

    def to_display(name: AnyStr) -> bytes:
        return pycompat.sysbytes(name).replace(b'_', b'-')

    previous = None
    for x in args:
        if opts.get(x):

                )
            previous = x
    return previous


def check_incompatible_arguments(
    opts: Dict[AnyStr, Any],
    first: AnyStr,
    others: Iterable[AnyStr],
) -> None:
    """abort if the first argument is given along with any of the others

    Unlike check_at_most_one_arg(), `others` are not mutually exclusive
    among themselves, and they're passed as a single collection.
    """
    for other in others:
        check_at_most_one_arg(opts, first, other)


def resolve_commit_options(ui: "uimod.ui", opts: Dict[str, Any]) -> bool:
    """modify commit options dict to handle related options

    The return value indicates that ``rewrite.update-timestamp`` is the reason
    the ``date`` option is set.
    """

        opts['user'] = ui.username()

    return datemaydiffer


def check_note_size(opts: Dict[str, Any]) -> None:
    """make sure note is of valid format"""

    note = opts.get('note')
    if not note:
        return

    ctx = repo[None]
    for s in sorted(ctx.substate):
        ctx.sub(s).bailifchanged(hint=hint)


def logmessage(ui: "uimod.ui", opts: Dict[bytes, Any]) -> Optional[bytes]:
    """get the log message according to -m and -l option"""

    check_at_most_one_arg(opts, b'message', b'logfile')

    message = cast(Optional[bytes], opts.get(b'message'))
    logfile = opts.get(b'logfile')

    if not message and logfile:
        try:
            if isstdiofilename(logfile):

    commands.
    """
    return openstorage(repo, cmd, file_, opts, returnrevlog=True)


def copy(ui, repo, pats, opts: Dict[bytes, Any], rename=False):
    check_incompatible_arguments(opts, b'forget', [b'dry_run'])

    # called with the repo lock held
    #
    # hgsep => pathname that uses "/" to separate directories

            if not sub.cat(
                submatch,
                basefm,
                fntemplate,
                subprefix,
                **pycompat.strkwargs(opts),
            ):
                err = 0
        except error.RepoLookupError:
            ui.status(
                _(b"skipping missing subrepository: %s\n") % uipathfn(subpath)

            return False
    else:
        return f not in ctx2.manifest()


def amend(ui, repo, old, extra, pats, opts: Dict[str, Any]):
    # avoid cycle context -> subrepo -> cmdutil
    from . import context

    # amend will reuse the existing user if not specified, but the obsolete
    # marker creation requires that the current user's name is specified.