File size: 1,464 Bytes
b1bd80d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from __future__ import annotations

import io
import subprocess

import git


class Git(git.Git):
    """
    Git subclassed to never use persistent processes.
    """

    def _get_persistent_cmd(self, attr_name, cmd_name, *args, **kwargs):
        raise NotImplementedError(f"Refusing to use persistent process: {attr_name} ({cmd_name} {args} {kwargs})")

    def get_object_header(self, ref: str | bytes) -> tuple[str, str, int]:
        ret = subprocess.check_output(
            [self.GIT_PYTHON_GIT_EXECUTABLE, "cat-file", "--batch-check"],
            input=self._prepare_ref(ref),
            cwd=self._working_dir,
            timeout=2,
        )
        return self._parse_object_header(ret)

    def stream_object_data(self, ref: str) -> tuple[str, str, int, Git.CatFileContentStream]:
        # Not really streaming, per se; this buffers the entire object in memory.
        # Shouldn't be a problem for our use case, since we're only using this for
        # object headers (commit objects).
        ret = subprocess.check_output(
            [self.GIT_PYTHON_GIT_EXECUTABLE, "cat-file", "--batch"],
            input=self._prepare_ref(ref),
            cwd=self._working_dir,
            timeout=30,
        )
        bio = io.BytesIO(ret)
        hexsha, typename, size = self._parse_object_header(bio.readline())
        return (hexsha, typename, size, self.CatFileContentStream(size, bio))


class Repo(git.Repo):
    GitCommandWrapperType = Git