64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
"""Module with our own gitdb implementation - it uses the git command"""
|
|
from git.util import bin_to_hex, hex_to_bin
|
|
from gitdb.base import OInfo, OStream
|
|
from gitdb.db import GitDB # @UnusedImport
|
|
from gitdb.db import LooseObjectDB
|
|
|
|
from gitdb.exc import BadObject
|
|
from git.exc import GitCommandError
|
|
|
|
# typing-------------------------------------------------
|
|
|
|
from typing import TYPE_CHECKING
|
|
from git.types import PathLike
|
|
|
|
if TYPE_CHECKING:
|
|
from git.cmd import Git
|
|
|
|
|
|
# --------------------------------------------------------
|
|
|
|
__all__ = ("GitCmdObjectDB", "GitDB")
|
|
|
|
|
|
class GitCmdObjectDB(LooseObjectDB):
|
|
|
|
"""A database representing the default git object store, which includes loose
|
|
objects, pack files and an alternates file
|
|
|
|
It will create objects only in the loose object database.
|
|
:note: for now, we use the git command to do all the lookup, just until he
|
|
have packs and the other implementations
|
|
"""
|
|
|
|
def __init__(self, root_path: PathLike, git: "Git") -> None:
|
|
"""Initialize this instance with the root and a git command"""
|
|
super(GitCmdObjectDB, self).__init__(root_path)
|
|
self._git = git
|
|
|
|
def info(self, binsha: bytes) -> OInfo:
|
|
hexsha, typename, size = self._git.get_object_header(bin_to_hex(binsha))
|
|
return OInfo(hex_to_bin(hexsha), typename, size)
|
|
|
|
def stream(self, binsha: bytes) -> OStream:
|
|
"""For now, all lookup is done by git itself"""
|
|
hexsha, typename, size, stream = self._git.stream_object_data(bin_to_hex(binsha))
|
|
return OStream(hex_to_bin(hexsha), typename, size, stream)
|
|
|
|
# { Interface
|
|
|
|
def partial_to_complete_sha_hex(self, partial_hexsha: str) -> bytes:
|
|
""":return: Full binary 20 byte sha from the given partial hexsha
|
|
:raise AmbiguousObjectName:
|
|
:raise BadObject:
|
|
:note: currently we only raise BadObject as git does not communicate
|
|
AmbiguousObjects separately"""
|
|
try:
|
|
hexsha, _typename, _size = self._git.get_object_header(partial_hexsha)
|
|
return hex_to_bin(hexsha)
|
|
except (GitCommandError, ValueError) as e:
|
|
raise BadObject(partial_hexsha) from e
|
|
# END handle exceptions
|
|
|
|
# } END interface
|