hgext/largefiles/overrides.py

50026:3550e4a88ccd

# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.

'''Overridden Mercurial commands and functions for the largefiles extension'''

import copy
import os

from mercurial.i18n import _

from mercurial import (
    archival,
    cmdutil,
    copies as copiesmod,
    error,
    exchange,
    extensions,
    exthelper,
    filemerge,

        removelargefiles(
            ui, repo, False, matcher, uipathfn, dryrun, after=after, force=force
        )
        or result
    )


@eh.wrapfunction(subrepo.hgsubrepo, b'status')
def overridestatusfn(orig, repo, rev2, **opts):
    with lfstatus(repo._repo):

50027:0b4a6912292e

# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.

'''Overridden Mercurial commands and functions for the largefiles extension'''

import contextlib
import copy
import os

from mercurial.i18n import _

from mercurial import (
    archival,
    cmdutil,
    copies as copiesmod,
    dirstate,
    error,
    exchange,
    extensions,
    exthelper,
    filemerge,

        removelargefiles(
            ui, repo, False, matcher, uipathfn, dryrun, after=after, force=force
        )
        or result
    )


@eh.wrapfunction(dirstate.dirstate, b'_changing')
@contextlib.contextmanager
def _changing(orig, self, repo, change_type):
    pre = sub_dirstate = getattr(self, '_sub_dirstate', None)
    try:
        lfd = getattr(self, '_large_file_dirstate', False)
        if sub_dirstate is None and not lfd:
            sub_dirstate = lfutil.openlfdirstate(repo.ui, repo)
            self._sub_dirstate = sub_dirstate
        if not lfd:
            assert self._sub_dirstate is not None
        with orig(self, repo, change_type):
            if sub_dirstate is None:
                yield
            else:
                with sub_dirstate._changing(repo, change_type):
                    yield
    finally:
        self._sub_dirstate = pre


@eh.wrapfunction(subrepo.hgsubrepo, b'status')
def overridestatusfn(orig, repo, rev2, **opts):
    with lfstatus(repo._repo):