comparison mercurial/shelve.py @ 46285:e79f8ae0901b

shelve: move method for creating backup to new shelf class

Differential Revision: https://phab.mercurial-scm.org/D9712

46284:d7f763c8c58e

    the vfs layer"""

    def __init__(self, repo, name, filetype=None):
        self.name = name
        self.vfs = vfsmod.vfs(repo.vfs.join(shelvedir))
        self.backupvfs = vfsmod.vfs(repo.vfs.join(backupdir))
        if filetype:
            self.fname = name + b'.' + filetype
        else:
            self.fname = name

    def exists(self):
        return self.vfs.exists(self.fname)

    def backupfilename(self):
        def gennames(base):
            yield base
            base, ext = base.rsplit(b'.', 1)
            for i in itertools.count(1):
                yield b'%s-%d.%s' % (base, i, ext)

        for n in gennames(self.fname):
            if not self.backupvfs.exists(n):
                return self.backupvfs.join(n)

    def movetobackup(self):
        if not self.backupvfs.isdir():
            self.backupvfs.makedir()
        util.rename(self.vfs.join(self.fname), self.backupfilename())


class Shelf(object):
    """Represents a shelf, including possibly multiple files storing it.

    Old shelves will have a .patch and a .hg file. Newer shelves will

    def __init__(self, repo, name):
        self.repo = repo
        self.name = name
        self.vfs = vfsmod.vfs(repo.vfs.join(shelvedir))

    def exists(self):
        return self.vfs.exists(self.name + b'.' + patchextension)

    def mtime(self):

        finally:
            fp.close()

    def open_patch(self, mode=b'rb'):
        return self.vfs(self.name + b'.patch', mode)


class shelvedstate(object):
    """Handle persistence during unshelving operations.

def cleanupcmd(ui, repo):
    """subcommand that deletes all shelves"""

    with repo.wlock():
        for _mtime, name in listshelves(repo):
            for suffix in shelvefileextensions:
                shfile = shelvedfile(repo, name, suffix)
                if shfile.exists():
                    shfile.movetobackup()
            cleanupoldbackups(repo)


def deletecmd(ui, repo, pats):
    """subcommand that deletes a specific shelve"""

        for name in pats:
            if not Shelf(repo, name).exists():
                raise error.InputError(
                    _(b"shelved change '%s' not found") % name
                )
            for suffix in shelvefileextensions:
                shfile = shelvedfile(repo, name, suffix)
                if shfile.exists():
                    shfile.movetobackup()
            cleanupoldbackups(repo)


def listshelves(repo):
    """return all shelves in repo as list of (time, name)"""

def unshelvecleanup(ui, repo, name, opts):
    """remove related files after an unshelve"""
    if not opts.get(b'keep'):
        for filetype in shelvefileextensions:
            shfile = shelvedfile(repo, name, filetype)
            if shfile.exists():
                shfile.movetobackup()
        cleanupoldbackups(repo)


def unshelvecontinue(ui, repo, state, opts):
    """subcommand to continue an in-progress unshelve"""

46285:e79f8ae0901b

    the vfs layer"""

    def __init__(self, repo, name, filetype=None):
        self.name = name
        self.vfs = vfsmod.vfs(repo.vfs.join(shelvedir))
        if filetype:
            self.fname = name + b'.' + filetype
        else:
            self.fname = name

    def exists(self):
        return self.vfs.exists(self.fname)


class Shelf(object):
    """Represents a shelf, including possibly multiple files storing it.

    Old shelves will have a .patch and a .hg file. Newer shelves will

    def __init__(self, repo, name):
        self.repo = repo
        self.name = name
        self.vfs = vfsmod.vfs(repo.vfs.join(shelvedir))
        self.backupvfs = vfsmod.vfs(repo.vfs.join(backupdir))

    def exists(self):
        return self.vfs.exists(self.name + b'.' + patchextension)

    def mtime(self):

        finally:
            fp.close()

    def open_patch(self, mode=b'rb'):
        return self.vfs(self.name + b'.patch', mode)

    def _backupfilename(self, filename):
        def gennames(base):
            yield base
            base, ext = base.rsplit(b'.', 1)
            for i in itertools.count(1):
                yield b'%s-%d.%s' % (base, i, ext)

        for n in gennames(filename):
            if not self.backupvfs.exists(n):
                return self.backupvfs.join(n)

    def movetobackup(self):
        if not self.backupvfs.isdir():
            self.backupvfs.makedir()
        for suffix in shelvefileextensions:
            filename = self.name + b'.' + suffix
            if self.vfs.exists(filename):
                util.rename(
                    self.vfs.join(filename), self._backupfilename(filename)
                )


class shelvedstate(object):
    """Handle persistence during unshelving operations.

def cleanupcmd(ui, repo):
    """subcommand that deletes all shelves"""

    with repo.wlock():
        for _mtime, name in listshelves(repo):
            Shelf(repo, name).movetobackup()
            cleanupoldbackups(repo)


def deletecmd(ui, repo, pats):
    """subcommand that deletes a specific shelve"""

        for name in pats:
            if not Shelf(repo, name).exists():
                raise error.InputError(
                    _(b"shelved change '%s' not found") % name
                )
            Shelf(repo, name).movetobackup()
            cleanupoldbackups(repo)


def listshelves(repo):
    """return all shelves in repo as list of (time, name)"""

def unshelvecleanup(ui, repo, name, opts):
    """remove related files after an unshelve"""
    if not opts.get(b'keep'):
        Shelf(repo, name).movetobackup()
        cleanupoldbackups(repo)


def unshelvecontinue(ui, repo, state, opts):
    """subcommand to continue an in-progress unshelve"""