GitPython/git/index/util.py at main · gitpython-developers/GitPython · GitHub

Repository navigation

History

# This module is part of GitPython and is released under the

# 3-Clause BSD License: https://opensource.org/license/bsd-3-clause/

"""Index utilities."""

__all__ = ["TemporaryFileSwap", "default_index", "git_working_dir"]

import contextlib

from functools import wraps

import os

import os.path as osp

import tempfile

from types import TracebackType

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

from typing import Any, Callable, TYPE_CHECKING, Optional, Type, cast

from git.types import Literal, PathLike, _T

if TYPE_CHECKING:

from git.index import IndexFile

# ---------------------------------------------------------------------------------

class TemporaryFileSwap:

"""Utility class moving a file to a temporary location within the same directory and

moving it back on to where on object deletion."""

__slots__ = ("file_path", "tmp_file_path")

def __init__(self, file_path: PathLike) -> None:

self.file_path = file_path

dirname, basename = osp.split(file_path)

fd, self.tmp_file_path = tempfile.mkstemp(prefix=basename, dir=dirname)

os.close(fd)

with contextlib.suppress(OSError): # It may be that the source does not exist.

os.replace(self.file_path, self.tmp_file_path)

def __enter__(self) -> "TemporaryFileSwap":

return self

def __exit__(

self,

exc_type: Optional[Type[BaseException]],

exc_val: Optional[BaseException],

exc_tb: Optional[TracebackType],

) -> Literal[False]:

if osp.isfile(self.tmp_file_path):

os.replace(self.tmp_file_path, self.file_path)

return False

# { Decorators

def default_index(func: Callable[..., _T]) -> Callable[..., _T]:

"""Decorator ensuring the wrapped method may only run if we are the default

repository index.

This is as we rely on git commands that operate on that index only.

"""

@wraps(func)

def check_default_index(self: "IndexFile", *args: Any, **kwargs: Any) -> _T:

if self._file_path != self._index_path():

raise AssertionError(

"Cannot call %r on indices that do not represent the default git index" % func.__name__

)

return func(self, *args, **kwargs)

# END wrapper method

return check_default_index

def git_working_dir(func: Callable[..., _T]) -> Callable[..., _T]:

"""Decorator which changes the current working dir to the one of the git

repository in order to ensure relative paths are handled correctly."""

@wraps(func)

def set_git_working_dir(self: "IndexFile", *args: Any, **kwargs: Any) -> _T:

cur_wd = os.getcwd()

os.chdir(cast(PathLike, self.repo.working_tree_dir))

try:

return func(self, *args, **kwargs)

finally:

os.chdir(cur_wd)

# END handle working dir

# END wrapper

return set_git_working_dir

# } END decorators

GitPython/git/index/util.py at main · gitpython-developers/GitPython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

util.py

Latest commit

History

Breadcrumbs

util.py

File metadata and controls