GitPython/git/objects/submodule/util.py at main · danigm/GitPython · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

import git

from git.exc import InvalidGitRepositoryError

from git.config import GitConfigParser

from io import BytesIO

import weakref

# typing -----------------------------------------------------------------------

from typing import Any, Sequence, TYPE_CHECKING, Union

from git.types import PathLike

if TYPE_CHECKING:

from .base import Submodule

from weakref import ReferenceType

from git.repo import Repo

from git.refs import Head

from git import Remote

from git.refs import RemoteReference

__all__ = (

"sm_section",

"sm_name",

"mkhead",

"find_first_remote_branch",

"SubmoduleConfigParser",

)

# { Utilities

def sm_section(name: str) -> str:

""":return: section title used in .gitmodules configuration file"""

return f'submodule "{name}"'

def sm_name(section: str) -> str:

""":return: name of the submodule as parsed from the section name"""

section = section.strip()

return section[11:-1]

def mkhead(repo: "Repo", path: PathLike) -> "Head":

""":return: New branch/head instance"""

return git.Head(repo, git.Head.to_full_path(path))

def find_first_remote_branch(remotes: Sequence["Remote"], branch_name: str) -> "RemoteReference":

"""Find the remote branch matching the name of the given branch or raise InvalidGitRepositoryError"""

for remote in remotes:

try:

return remote.refs[branch_name]

except IndexError:

continue

# END exception handling

# END for remote

raise InvalidGitRepositoryError("Didn't find remote branch '%r' in any of the given remotes" % branch_name)

# } END utilities

# { Classes

class SubmoduleConfigParser(GitConfigParser):

"""

Catches calls to _write, and updates the .gitmodules blob in the index

with the new data, if we have written into a stream. Otherwise it will

add the local file to the index to make it correspond with the working tree.

Additionally, the cache must be cleared

Please note that no mutating method will work in bare mode

"""

def __init__(self, *args: Any, **kwargs: Any) -> None:

self._smref: Union["ReferenceType[Submodule]", None] = None

self._index = None

self._auto_write = True

super(SubmoduleConfigParser, self).__init__(*args, **kwargs)

# { Interface

def set_submodule(self, submodule: "Submodule") -> None:

"""Set this instance's submodule. It must be called before

the first write operation begins"""

self._smref = weakref.ref(submodule)

def flush_to_index(self) -> None:

"""Flush changes in our configuration file to the index"""

assert self._smref is not None

# should always have a file here

assert not isinstance(self._file_or_files, BytesIO)

sm = self._smref()

if sm is not None:

index = self._index

if index is None:

index = sm.repo.index

# END handle index

index.add([sm.k_modules_file], write=self._auto_write)

sm._clear_cache()

# END handle weakref

# } END interface

# { Overridden Methods

def write(self) -> None: # type: ignore[override]

rval: None = super(SubmoduleConfigParser, self).write()

self.flush_to_index()

return rval

# END overridden methods

# } END classes

GitPython/git/objects/submodule/util.py at main · danigm/GitPython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

util.py

Latest commit

History

Breadcrumbs

util.py

File metadata and controls