FilesExpand file tree

 master

/

util.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

99 lines (73 loc) · 2.83 KB

 master

/

util.py

Top

File metadata and controls

• Code
• Blame

99 lines (73 loc) · 2.83 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

"""Module containing index utilities"""

from functools import wraps

import os

import struct

import tempfile

from git.compat import is_win

import os.path as osp

__all__ = ('TemporaryFileSwap', 'post_clear_cache', 'default_index', 'git_working_dir')

#{ Aliases

pack = struct.pack

unpack = struct.unpack

#} END aliases

class TemporaryFileSwap(object):

"""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):

self.file_path = file_path

self.tmp_file_path = self.file_path + tempfile.mktemp('', '', '')

# it may be that the source does not exist

try:

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

except OSError:

pass

def __del__(self):

if osp.isfile(self.tmp_file_path):

if is_win and osp.exists(self.file_path):

os.remove(self.file_path)

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

# END temp file exists

#{ Decorators

def post_clear_cache(func):

"""Decorator for functions that alter the index using the git command. This would

invalidate our possibly existing entries dictionary which is why it must be

deleted to allow it to be lazily reread later.

:note:

This decorator will not be required once all functions are implemented

natively which in fact is possible, but probably not feasible performance wise.

"""

@wraps(func)

def post_clear_cache_if_not_raised(self, *args, **kwargs):

rval = func(self, *args, **kwargs)

self._delete_entries_cache()

return rval

# END wrapper method

return post_clear_cache_if_not_raised

def default_index(func):

"""Decorator assuring 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, *args, **kwargs):

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):

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

repository in order to assure relative paths are handled correctly"""

@wraps(func)

def set_git_working_dir(self, *args, **kwargs):

cur_wd = os.getcwd()

os.chdir(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