To get this branch, use:
bzr branch
http://bzr.ed.am/stdhome
« back to all changes in this revision
Viewing changes to lib/stdhome/walker/copy_base.py
- browse files at revision 13
- compare with another revision
- download diff
- download tarball
- view history from revision 13
- Committer: Tim Marston
- Date: 2014-03-19 20:02:10 UTC
- Revision ID: tim@ed.am-20140319200210-b6nm63rpktfmw0l3
changed working of output
- files removed:
- lib/stdhome/command/revert.py
- files modified:
- lib/stdhome/command/diff.py
added
removed
lib/stdhome/walker/copy_base.py
1
# copy_base.py
2
#
3
# Copyright (C) 2013 to 2014 Tim Marston <tim@edm.am>
4
#
5
# This file is part of stdhome (hereafter referred to as "this program").
6
# See http://ed.am/dev/stdhome for more information.
7
#
8
# This program is free software: you can redistribute it and/or modify
9
# it under the terms of the GNU General Public License as published by
10
# the Free Software Foundation, either version 3 of the License, or
11
# (at your option) any later version.
12
#
13
# This program is distributed in the hope that it will be useful,
14
# but WITHOUT ANY WARRANTY; without even the implied warranty of
15
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16
# GNU General Public License for more details.
17
#
18
# You should have received a copy of the GNU General Public License
19
# along with this program. If not, see <http://www.gnu.org/licenses/>.
20
21
22
import filecmp, os, shutil
23
from walker import Walker
24
import stdhome.the as the
25
26
27
class CopyBaseWalker( Walker ):
28
"""The copy-base walker traverses a walklist ruthlessly mirroring src to dst.
29
It is designed to be the base class of both the copy-in and copy-out walker,
30
both of which are specialisations of this purpose. See them for more
31
information.
32
"""
33
34
accept_list = None
35
36
37
def process( self, rel_file, src, dst ):
38
39
# src entity is directory
40
if src.type == 'd':
41
42
# if dst entity doesn't exist, create directory (no need to recurse,
43
# since we're copying the whole directory)
44
if dst.type == '_':
45
if the.verbose > 1: self.print_op( rel_file, 'd>_' )
46
shutil.copytree( src.file, dst.file, True )
47
48
# if dst entity is a directory, copy permissions, as required (and
49
# recurse)
50
elif dst.type == 'd':
51
# TODO: should check permission and only do as necessary
52
if the.verbose > 1: self.print_op( rel_file, 'd>d' )
53
shutil.copystat( src.file, dst.file )
54
return True
55
56
# if dst entity is a symlink to a directory, and this is an
57
# acceptable substitute, just recurse
58
elif dst.link_type == 'd' and \
59
self.accept_list and self.accept_list.match( rel_file ):
60
if the.verbose > 1: self.print_op( rel_file, 'd@d' )
61
return True
62
63
# if dst entity is a file or symlink in home dir, replace it with
64
# directory (no need to recurse, since we're copying the whole
65
# directory)
66
elif dst.type == 'f' or dst.type == 'l':
67
if the.verbose > 1: self.print_op( rel_file, 'd>' + dst.type )
68
os.unlink( dst.file )
69
shutil.copytree( src.file, dst.file, True )
70
71
else:
72
raise NotImplementedError()
73
74
# src entity is file
75
elif src.type == 'f':
76
77
# if dst entity doesn't exist, copy file
78
if dst.type == '_':
79
if the.verbose > 1: self.print_op( rel_file, 'f>_' )
80
shutil.copy( src.file, dst.file )
81
shutil.copystat( src.file, dst.file )
82
83
# if dst entity is a file, replace it only if it differs
84
elif dst.type == 'f':
85
if not filecmp.cmp( src.file, dst.file ):
86
if the.verbose > 1: self.print_op( rel_file, 'f>f' )
87
os.unlink( dst.file )
88
shutil.copy( src.file, dst.file )
89
shutil.copystat( src.file, dst.file )
90
else:
91
if the.verbose > 1: self.print_op( rel_file, 'f=f' )
92
93
# if dst entity is a directory, replace it with file
94
elif dst.type == 'd':
95
if the.verbose > 1: self.print_op( rel_file, 'f>d' )
96
shutil.rmtree( dst.file )
97
shutil.copy( src.file, dst.file )
98
shutil.copystat( src.file, dst.file )
99
100
# if dst entity is a symlink, replace it with file
101
elif dst.type == 'l':
102
if the.verbose > 1: self.print_op( rel_file, 'f>l' )
103
os.unlink( dst.file )
104
shutil.copy( src.file, dst.file )
105
shutil.copystat( src.file, dst.file )
106
107
else:
108
raise NotImplementedError()
109
110
# src entity is a symlink
111
elif src.type == 'l':
112
113
# if dst entity doesn't exist, copy symlink
114
if dst.type == '_':
115
if the.verbose > 1: self.print_op( rel_file, 'l>_' )
116
os.symlink( os.readlink( src.file ), dst.file )
117
118
# if dst entity is a symlink, replace it only if it differs
119
elif dst.type == 'l':
120
if os.readlink( src.file ) != os.readlink( dst.file ):
121
if the.verbose > 1: self.print_op( rel_file, 'l>l' )
122
os.unlink( dst.file )
123
os.symlink( os.readlink( src.file ), dst.file )
124
else:
125
if the.verbose > 1: self.print_op( rel_file, 'l=l' )
126
127
# if dst entity is a file, replace it with symlink
128
elif dst.type == 'f':
129
if the.verbose > 1: self.print_op( rel_file, 'l>f' )
130
os.unlink( dst.file )
131
os.symlink( os.readlink( src.file ), dst.file )
132
133
# if dst entity is a directory, and src entity is a symlink to a
134
# directory, and this is an acceptable substitute, just recurse
135
elif dst.type == 'd' and src.link_type == 'd' and \
136
self.accept_list and self.accept_list.match( rel_file ):
137
if the.verbose > 1: self.print_op( rel_file, 'd@d' )
138
return True
139
140
# if dst entity is a directory, replace it with symlink
141
elif dst.type == 'd':
142
if the.verbose > 1: self.print_op( rel_file, 'l>d' )
143
shutil.rmtree( dst.file )
144
os.symlink( os.readlink( src.file ), dst.file )
145
146
else:
147
raise NotImplementedError()
148
149
# src entity is missing
150
elif src.type == '_':
151
152
# if dst entity doesn't exist, we're good
153
if dst.type == '_':
154
pass;
155
156
# if dst entity is a file or symlink, delete it
157
elif dst.type == 'f' or dst.type == 'l':
158
if the.verbose > 1: self.print_op( rel_file, '_>' + dst.type )
159
os.unlink( dst.file )
160
161
# if dst entity is a directory, delete it
162
elif dst.type == 'd':
163
if the.verbose > 1: self.print_op( rel_file, '_>d' )
164
shutil.rmtree( dst.file )
165
166
else:
167
raise NotImplementedError()
168
169
# if we got here, we don't want to recurse...
170
return False
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0