28
29
"""The copy-base walker traverses a walklist ruthlessly mirroring src to dst.
29
30
It is designed to be the base class of both the copy-in and copy-out walker,
30
31
both of which are specialisations of this purpose. See them for more
31
information. The print_op method, derived in those classes, takes, in
32
addition to the relative filename, a source file type, an operation, and a
33
sestination file type. Valid file types are f (file), l (symlink), d
34
(directory) and _ (non-existant). Valid operations are * (modify), = (skip:
35
same), @ (skip: symlink substitute) and # (skip: ignored).
40
self.check_src_symlinks = False
41
self.check_dst_symlinks = False
42
self.check_dst_ignores = False
35
def __init__( self, updated_files = None ):
36
self.accept_list = FileMatcher()
45
39
def process( self, rel_file, src, dst ):
48
if self.check_dst_ignores and the.config.ignores.matches( rel_file ):
49
self.print_op( rel_file, src.type, '#', dst.type )
52
41
# src entity is directory
53
42
if src.type == 'd':
55
44
# if dst entity doesn't exist, create directory (no need to recurse,
56
45
# since we're copying the whole directory)
57
46
if dst.type == '_':
58
self.print_op( rel_file, 'd', '*', '_' )
59
shutil.copytree( src.file, dst.file, True )
47
if the.verbose > 1: self.print_op( rel_file, 'd>_' )
49
shutil.copystat( src.file, dst.file )
61
52
# if dst entity is a directory, copy permissions, as required (and
63
54
elif dst.type == 'd':
64
55
# TODO: should check permission and only do as necessary
65
self.print_op( rel_file, 'd', '*', 'd' )
56
if the.verbose > 1: self.print_op( rel_file, 'd>d' )
66
57
shutil.copystat( src.file, dst.file )
69
# if dst entity is a symlink to a directory, and this is an
70
# acceptable substitute, just recurse
71
elif self.check_dst_symlinks and dst.link_type == 'd' and \
72
the.config.symlinks.matches( rel_file ):
73
self.print_op( rel_file, 'd', '@', 'd' )
60
# TODO: if dst entity is a symlink to a directory, and rel_file is
61
# matched by the acceptable symlinks file matcher, just recurse
62
elif dst.link_type == 'd' and self.accept_list.match( rel_file ):
63
if the.verbose > 1: self.print_op( rel_file, 'd@d' )
76
66
# if dst entity is a file or symlink in home dir, replace it with
77
67
# directory (no need to recurse, since we're copying the whole
79
69
elif dst.type == 'f' or dst.type == 'l':
80
self.print_op( rel_file, 'd', '*', dst.type )
70
if the.verbose > 1: self.print_op( rel_file, 'd>' + dst.type )
81
71
os.unlink( dst.file )
82
shutil.copytree( src.file, dst.file, True )
73
shutil.copystat( src.file, dst.file )
85
77
raise NotImplementedError()
87
79
# src entity is file
90
82
# if dst entity doesn't exist, copy file
91
83
if dst.type == '_':
92
self.print_op( rel_file, 'f', '*', '_' )
84
if the.verbose > 1: self.print_op( rel_file, 'f>_' )
93
85
shutil.copy( src.file, dst.file )
94
86
shutil.copystat( src.file, dst.file )
96
88
# if dst entity is a file, replace it only if it differs
97
89
elif dst.type == 'f':
98
90
if not filecmp.cmp( src.file, dst.file ):
99
self.print_op( rel_file, 'f', '*', 'f' )
91
if the.verbose > 1: self.print_op( rel_file, 'f>f' )
100
92
os.unlink( dst.file )
101
93
shutil.copy( src.file, dst.file )
102
94
shutil.copystat( src.file, dst.file )
104
self.print_op( rel_file, 'f', '=', 'f' )
96
if the.verbose > 1: self.print_op( rel_file, 'f=f' )
106
98
# if dst entity is a directory, replace it with file
107
99
elif dst.type == 'd':
108
self.print_op( rel_file, 'f', '*', 'd' )
100
if the.verbose > 1: self.print_op( rel_file, 'f>d' )
109
101
shutil.rmtree( dst.file )
110
102
shutil.copy( src.file, dst.file )
111
103
shutil.copystat( src.file, dst.file )
113
105
# if dst entity is a symlink, replace it with file
114
106
elif dst.type == 'l':
115
self.print_op( rel_file, 'f', '*', 'l' )
107
if the.verbose > 1: self.print_op( rel_file, 'f>l' )
116
108
os.unlink( dst.file )
117
109
shutil.copy( src.file, dst.file )
118
110
shutil.copystat( src.file, dst.file )
121
113
raise NotImplementedError()
123
115
# src entity is a symlink
124
elif src.type == 'l':
126
118
# if dst entity doesn't exist, copy symlink
127
119
if dst.type == '_':
128
self.print_op( rel_file, 'l', '*', '_' )
120
if the.verbose > 1: self.print_op( rel_file, 'l>_' )
129
121
os.symlink( os.readlink( src.file ), dst.file )
131
123
# if dst entity is a symlink, replace it only if it differs
132
124
elif dst.type == 'l':
133
125
if os.readlink( src.file ) != os.readlink( dst.file ):
134
self.print_op( rel_file, 'l', '*', 'l' )
126
if the.verbose > 1: self.print_op( rel_file, 'l>l' )
135
127
os.unlink( dst.file )
136
128
os.symlink( os.readlink( src.file ), dst.file )
138
self.print_op( rel_file, 'l', '=', 'l' )
130
if the.verbose > 1: self.print_op( rel_file, 'l=l' )
140
132
# if dst entity is a file, replace it with symlink
141
133
elif dst.type == 'f':
142
self.print_op( rel_file, 'l', '*', 'f' )
134
if the.verbose > 1: self.print_op( rel_file, 'l>f' )
143
135
os.unlink( dst.file )
144
136
os.symlink( os.readlink( src.file ), dst.file )
146
# if dst entity is a directory...
138
# if dst entity is a directory, replace it with symlink
147
139
elif dst.type == 'd':
149
# if src entity is a symlink to a directory, and this is an
150
# acceptable substitute, just recurse
151
if self.check_src_symlinks and src.link_type == 'd' and \
152
the.config.symlinks.matches( rel_file ):
153
self.print_op( rel_file, 'd', '@', 'd' )
156
# else replace it with a symlink
157
self.print_op( rel_file, 'l', '*', 'd' )
140
if the.verbose > 1: self.print_op( rel_file, 'l>d' )
158
141
shutil.rmtree( dst.file )
159
142
os.symlink( os.readlink( src.file ), dst.file )