kallithea Files · rhodecode/lib/vcs/backends/hg/workdir.py

Files @ 61f9aeb2129e

Branch filter:

Location: kallithea/rhodecode/lib/vcs/backends/hg/workdir.py

61f9aeb2129e 657 B text/x-python Show Annotation Show as Raw Download as Raw

Marcin Kuzminski

Added session wrapper, for rc 1.2.X compatibility. Adds backwards compatability
for older RhodeCode sessions, created before 1.3 series.

from rhodecode.lib.vcs.backends.base import BaseWorkdir
from rhodecode.lib.vcs.exceptions import BranchDoesNotExistError

from ...utils.hgcompat import hg_merge


class MercurialWorkdir(BaseWorkdir):

    def get_branch(self):
        return self.repository._repo.dirstate.branch()

    def get_changeset(self):
        return self.repository.get_changeset()

    def checkout_branch(self, branch=None):
        if branch is None:
            branch = self.repository.DEFAULT_BRANCH_NAME
        if branch not in self.repository.branches:
            raise BranchDoesNotExistError

        hg_merge.update(self.repository._repo, branch, False, False, None)