mirror of https://github.com/morpheus65535/bazarr
248 lines
8.5 KiB
Python
248 lines
8.5 KiB
Python
![]() |
from git.config import SectionConstraint
|
||
|
from git.util import join_path
|
||
|
from git.exc import GitCommandError
|
||
|
|
||
|
from .symbolic import SymbolicReference
|
||
|
from .reference import Reference
|
||
|
|
||
|
__all__ = ["HEAD", "Head"]
|
||
|
|
||
|
|
||
|
def strip_quotes(string):
|
||
|
if string.startswith('"') and string.endswith('"'):
|
||
|
return string[1:-1]
|
||
|
return string
|
||
|
|
||
|
|
||
|
class HEAD(SymbolicReference):
|
||
|
|
||
|
"""Special case of a Symbolic Reference as it represents the repository's
|
||
|
HEAD reference."""
|
||
|
_HEAD_NAME = 'HEAD'
|
||
|
_ORIG_HEAD_NAME = 'ORIG_HEAD'
|
||
|
__slots__ = ()
|
||
|
|
||
|
def __init__(self, repo, path=_HEAD_NAME):
|
||
|
if path != self._HEAD_NAME:
|
||
|
raise ValueError("HEAD instance must point to %r, got %r" % (self._HEAD_NAME, path))
|
||
|
super(HEAD, self).__init__(repo, path)
|
||
|
|
||
|
def orig_head(self):
|
||
|
"""
|
||
|
:return: SymbolicReference pointing at the ORIG_HEAD, which is maintained
|
||
|
to contain the previous value of HEAD"""
|
||
|
return SymbolicReference(self.repo, self._ORIG_HEAD_NAME)
|
||
|
|
||
|
def reset(self, commit='HEAD', index=True, working_tree=False,
|
||
|
paths=None, **kwargs):
|
||
|
"""Reset our HEAD to the given commit optionally synchronizing
|
||
|
the index and working tree. The reference we refer to will be set to
|
||
|
commit as well.
|
||
|
|
||
|
:param commit:
|
||
|
Commit object, Reference Object or string identifying a revision we
|
||
|
should reset HEAD to.
|
||
|
|
||
|
:param index:
|
||
|
If True, the index will be set to match the given commit. Otherwise
|
||
|
it will not be touched.
|
||
|
|
||
|
:param working_tree:
|
||
|
If True, the working tree will be forcefully adjusted to match the given
|
||
|
commit, possibly overwriting uncommitted changes without warning.
|
||
|
If working_tree is True, index must be true as well
|
||
|
|
||
|
:param paths:
|
||
|
Single path or list of paths relative to the git root directory
|
||
|
that are to be reset. This allows to partially reset individual files.
|
||
|
|
||
|
:param kwargs:
|
||
|
Additional arguments passed to git-reset.
|
||
|
|
||
|
:return: self"""
|
||
|
mode = "--soft"
|
||
|
if index:
|
||
|
mode = "--mixed"
|
||
|
|
||
|
# it appears, some git-versions declare mixed and paths deprecated
|
||
|
# see http://github.com/Byron/GitPython/issues#issue/2
|
||
|
if paths:
|
||
|
mode = None
|
||
|
# END special case
|
||
|
# END handle index
|
||
|
|
||
|
if working_tree:
|
||
|
mode = "--hard"
|
||
|
if not index:
|
||
|
raise ValueError("Cannot reset the working tree if the index is not reset as well")
|
||
|
|
||
|
# END working tree handling
|
||
|
|
||
|
try:
|
||
|
self.repo.git.reset(mode, commit, '--', paths, **kwargs)
|
||
|
except GitCommandError as e:
|
||
|
# git nowadays may use 1 as status to indicate there are still unstaged
|
||
|
# modifications after the reset
|
||
|
if e.status != 1:
|
||
|
raise
|
||
|
# END handle exception
|
||
|
|
||
|
return self
|
||
|
|
||
|
|
||
|
class Head(Reference):
|
||
|
|
||
|
"""A Head is a named reference to a Commit. Every Head instance contains a name
|
||
|
and a Commit object.
|
||
|
|
||
|
Examples::
|
||
|
|
||
|
>>> repo = Repo("/path/to/repo")
|
||
|
>>> head = repo.heads[0]
|
||
|
|
||
|
>>> head.name
|
||
|
'master'
|
||
|
|
||
|
>>> head.commit
|
||
|
<git.Commit "1c09f116cbc2cb4100fb6935bb162daa4723f455">
|
||
|
|
||
|
>>> head.commit.hexsha
|
||
|
'1c09f116cbc2cb4100fb6935bb162daa4723f455'"""
|
||
|
_common_path_default = "refs/heads"
|
||
|
k_config_remote = "remote"
|
||
|
k_config_remote_ref = "merge" # branch to merge from remote
|
||
|
|
||
|
@classmethod
|
||
|
def delete(cls, repo, *heads, **kwargs):
|
||
|
"""Delete the given heads
|
||
|
|
||
|
:param force:
|
||
|
If True, the heads will be deleted even if they are not yet merged into
|
||
|
the main development stream.
|
||
|
Default False"""
|
||
|
force = kwargs.get("force", False)
|
||
|
flag = "-d"
|
||
|
if force:
|
||
|
flag = "-D"
|
||
|
repo.git.branch(flag, *heads)
|
||
|
|
||
|
def set_tracking_branch(self, remote_reference):
|
||
|
"""
|
||
|
Configure this branch to track the given remote reference. This will alter
|
||
|
this branch's configuration accordingly.
|
||
|
|
||
|
:param remote_reference: The remote reference to track or None to untrack
|
||
|
any references
|
||
|
:return: self"""
|
||
|
from .remote import RemoteReference
|
||
|
if remote_reference is not None and not isinstance(remote_reference, RemoteReference):
|
||
|
raise ValueError("Incorrect parameter type: %r" % remote_reference)
|
||
|
# END handle type
|
||
|
|
||
|
with self.config_writer() as writer:
|
||
|
if remote_reference is None:
|
||
|
writer.remove_option(self.k_config_remote)
|
||
|
writer.remove_option(self.k_config_remote_ref)
|
||
|
if len(writer.options()) == 0:
|
||
|
writer.remove_section()
|
||
|
else:
|
||
|
writer.set_value(self.k_config_remote, remote_reference.remote_name)
|
||
|
writer.set_value(self.k_config_remote_ref, Head.to_full_path(remote_reference.remote_head))
|
||
|
|
||
|
return self
|
||
|
|
||
|
def tracking_branch(self):
|
||
|
"""
|
||
|
:return: The remote_reference we are tracking, or None if we are
|
||
|
not a tracking branch"""
|
||
|
from .remote import RemoteReference
|
||
|
reader = self.config_reader()
|
||
|
if reader.has_option(self.k_config_remote) and reader.has_option(self.k_config_remote_ref):
|
||
|
ref = Head(self.repo, Head.to_full_path(strip_quotes(reader.get_value(self.k_config_remote_ref))))
|
||
|
remote_refpath = RemoteReference.to_full_path(join_path(reader.get_value(self.k_config_remote), ref.name))
|
||
|
return RemoteReference(self.repo, remote_refpath)
|
||
|
# END handle have tracking branch
|
||
|
|
||
|
# we are not a tracking branch
|
||
|
return None
|
||
|
|
||
|
def rename(self, new_path, force=False):
|
||
|
"""Rename self to a new path
|
||
|
|
||
|
:param new_path:
|
||
|
Either a simple name or a path, i.e. new_name or features/new_name.
|
||
|
The prefix refs/heads is implied
|
||
|
|
||
|
:param force:
|
||
|
If True, the rename will succeed even if a head with the target name
|
||
|
already exists.
|
||
|
|
||
|
:return: self
|
||
|
:note: respects the ref log as git commands are used"""
|
||
|
flag = "-m"
|
||
|
if force:
|
||
|
flag = "-M"
|
||
|
|
||
|
self.repo.git.branch(flag, self, new_path)
|
||
|
self.path = "%s/%s" % (self._common_path_default, new_path)
|
||
|
return self
|
||
|
|
||
|
def checkout(self, force=False, **kwargs):
|
||
|
"""Checkout this head by setting the HEAD to this reference, by updating the index
|
||
|
to reflect the tree we point to and by updating the working tree to reflect
|
||
|
the latest index.
|
||
|
|
||
|
The command will fail if changed working tree files would be overwritten.
|
||
|
|
||
|
:param force:
|
||
|
If True, changes to the index and the working tree will be discarded.
|
||
|
If False, GitCommandError will be raised in that situation.
|
||
|
|
||
|
:param kwargs:
|
||
|
Additional keyword arguments to be passed to git checkout, i.e.
|
||
|
b='new_branch' to create a new branch at the given spot.
|
||
|
|
||
|
:return:
|
||
|
The active branch after the checkout operation, usually self unless
|
||
|
a new branch has been created.
|
||
|
If there is no active branch, as the HEAD is now detached, the HEAD
|
||
|
reference will be returned instead.
|
||
|
|
||
|
:note:
|
||
|
By default it is only allowed to checkout heads - everything else
|
||
|
will leave the HEAD detached which is allowed and possible, but remains
|
||
|
a special state that some tools might not be able to handle."""
|
||
|
kwargs['f'] = force
|
||
|
if kwargs['f'] is False:
|
||
|
kwargs.pop('f')
|
||
|
|
||
|
self.repo.git.checkout(self, **kwargs)
|
||
|
if self.repo.head.is_detached:
|
||
|
return self.repo.head
|
||
|
else:
|
||
|
return self.repo.active_branch
|
||
|
|
||
|
#{ Configuration
|
||
|
def _config_parser(self, read_only):
|
||
|
if read_only:
|
||
|
parser = self.repo.config_reader()
|
||
|
else:
|
||
|
parser = self.repo.config_writer()
|
||
|
# END handle parser instance
|
||
|
|
||
|
return SectionConstraint(parser, 'branch "%s"' % self.name)
|
||
|
|
||
|
def config_reader(self):
|
||
|
"""
|
||
|
:return: A configuration parser instance constrained to only read
|
||
|
this instance's values"""
|
||
|
return self._config_parser(read_only=True)
|
||
|
|
||
|
def config_writer(self):
|
||
|
"""
|
||
|
:return: A configuration writer instance with read-and write access
|
||
|
to options of this head"""
|
||
|
return self._config_parser(read_only=False)
|
||
|
|
||
|
#} END configuration
|