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
100
101
102
103
104
105
106
107
108
|
import py
from click import echo
from operator import attrgetter
from .dotfile import Dotfile
from .exceptions import DotfileException, TargetIgnored
from .exceptions import NotRootedInHome, InRepository, IsDirectory
class Repository(object):
"""A repository is a directory that contains dotfiles.
:param repodir: the location of the repository directory
:param homedir: the location of the home directory (primarily for testing)
:param ignore: a list of targets to ignore
"""
homedir = py.path.local('~/', expanduser=True)
ignore = ['.git', '.hg']
def __init__(self, repodir, homedir=homedir, ignore=ignore, dot=True):
self.repodir = repodir.ensure(dir=1)
self.homedir = homedir
self.ignore = ignore
self.dot = dot
def __str__(self):
"""Return human-readable repository contents."""
return ''.join('%s\n' % item for item in self.contents()).rstrip()
def __repr__(self):
return '<Repository %r>' % self.repodir
def _target_to_name(self, target):
"""Return the expected symlink for the given repository target."""
relpath = self.repodir.bestrelpath(target)
if self.dot:
return self.homedir.join(relpath)
else:
return self.homedir.join('.%s' % relpath)
def _name_to_target(self, name):
"""Return the expected repository target for the given symlink."""
relpath = self.homedir.bestrelpath(name)
if self.dot:
return self.repodir.join(relpath)
else:
return self.repodir.join(relpath[1:])
def _dotfile(self, name):
"""Return a valid dotfile for the given path."""
target = self._name_to_target(name)
if not name.fnmatch('%s/*' % self.homedir):
raise NotRootedInHome(name)
if name.fnmatch('%s/*' % self.repodir):
raise InRepository(name)
if target.basename in self.ignore:
raise TargetIgnored(name)
if name.check(dir=1):
raise IsDirectory(name)
return Dotfile(name, target)
def _contents(self, dir):
def filter(node):
return node.check(dir=0) and node.basename not in self.ignore
def recurse(node):
return node.basename not in self.ignore
return dir.visit(filter, recurse)
def dotfiles(self, paths):
paths = list(set(map(py.path.local, paths)))
for path in paths:
if path.check(dir=1):
paths.extend(self._contents(path))
paths.remove(path)
def construct(path):
try:
return self._dotfile(path)
except DotfileException as err:
echo(err)
return None
return [d for d in map(construct, paths) if d is not None]
def contents(self):
def construct(target):
return Dotfile(self._target_to_name(target), target)
contents = self._contents(self.repodir)
return sorted(map(construct, contents), key=attrgetter('name'))
def prune(self):
"""Remove any empty directories in the repository."""
def filter(node):
return node.check(dir=1) and node.basename not in self.ignore
def recurse(node):
return node.basename not in self.ignore
for dir in self.repodir.visit(filter, recurse):
if not len(dir.listdir()):
dir.remove()
|
