/stdhome

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_out.py

  • Committer: Tim Marston
  • Date: 2014-03-19 20:02:10 UTC
  • Revision ID: tim@ed.am-20140319200210-b6nm63rpktfmw0l3
changed working of output

Show diffs side-by-side

added added

removed removed

19
19
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
20
 
21
21
 
22
 
import re
23
 
from copy_base import CopyBaseWalker
 
22
import filecmp, os, shutil
 
23
from walker import Walker
24
24
import stdhome.the as the
25
25
 
26
26
 
27
 
class CopyOutWalker( CopyBaseWalker ):
 
27
class CopyOutWalker( Walker ):
28
28
        """The copy-out walker traverses the repo, copying its content to the home
29
29
        directory.  It is run *after* checking for conflicts that might occur during
30
30
        copy-out, so any conflicts that exist can be assumed to be unimportant and
44
44
                                                 self.generate_walk_list( the.repo.full_dir )
45
45
 
46
46
 
47
 
        def print_op( self, rel_file, operation ):
48
 
                operation = re.sub( r'(.)(.)(.)', r'\3\2\1', operation )
49
 
                operation = re.sub( r'\>', r'<', operation )
50
 
                print "  %s %s" % ( operation, rel_file )
 
47
        def process( self, rel_file, src_file, src_type, dst_file, dst_type ):
 
48
 
 
49
                # directory (in repo)
 
50
                if src_type == 'd':
 
51
 
 
52
                        # if entity doesn't exist in home dir, create directory (no need to
 
53
                        # recurse, since we're copying the whole directory)
 
54
                        if dst_type == '_':
 
55
                                if the.verbose > 1: print "  _<d " + rel_file
 
56
                                os.mkdir( dst_file )
 
57
                                shutil.copystat( src_file, dst_file )
 
58
                                return False
 
59
 
 
60
                        # if entity is a file or symlink in home dir, replace it with
 
61
                        # directory (no need to recurse, since we're copying the whole
 
62
                        # directory)
 
63
                        elif dst_type == 'f' or dst_type == 'l':
 
64
                                if the.verbose > 1: print "  %s<d %s" % ( dst_type, rel_file )
 
65
                                os.unlink( dst_file )
 
66
                                os.mkdir( dst_file )
 
67
                                shutil.copystat( src_file, dst_file )
 
68
                                return False
 
69
 
 
70
                        # if entity is a directory in home dir, copy permissions, as
 
71
                        # required (and recurse)
 
72
                        elif dst_type == 'd':
 
73
                                # TODO: should check permission and only do as necessary
 
74
                                if the.verbose > 1: print "  d<d " + rel_file
 
75
                                shutil.copystat( src_file, dst_file )
 
76
                                return True
 
77
 
 
78
                        else:
 
79
                                raise NotImplementedError()
 
80
 
 
81
                # file (in repo)
 
82
                if src_type == 'f':
 
83
 
 
84
                        # if entity doesn't exist in home dir, copy file
 
85
                        if dst_type == '_':
 
86
                                if the.verbose > 1: print "  _<f " + rel_file
 
87
                                shutil.copy( src_file, dst_file )
 
88
                                shutil.copystat( src_file, dst_file )
 
89
 
 
90
                        # if entity is a symlink in home dir, replace it with file
 
91
                        elif dst_type == 'l':
 
92
                                if the.verbose > 1: print "  l<f " + rel_file
 
93
                                os.unlink( dst_file )
 
94
                                shutil.copy( src_file, dst_file )
 
95
                                shutil.copystat( src_file, dst_file )
 
96
 
 
97
                        # if entity is a file in home dir, replace it only if it differs
 
98
                        elif dst_type == 'f':
 
99
                                if not filecmp.cmp( src_file, dst_file ):
 
100
                                        if the.verbose > 1: print "  f<f " + rel_file
 
101
                                        os.unlink( dst_file )
 
102
                                        shutil.copy( src_file, dst_file )
 
103
                                        shutil.copystat( src_file, dst_file )
 
104
                                else:
 
105
                                        if the.verbose > 1: print "  f=f " + rel_file
 
106
 
 
107
                        # if entity is a directory in home dir, replace it with file
 
108
                        elif dst_type == 'd':
 
109
                                if the.verbose > 1: print "  d<f " + rel_file
 
110
                                shutil.rmtree( dst_file )
 
111
                                shutil.copy( src_file, dst_file )
 
112
                                shutil.copystat( src_file, dst_file )
 
113
 
 
114
                        else:
 
115
                                raise NotImplementedError()
 
116
 
 
117
                # link (in repo)
 
118
                if src_type == 'l':
 
119
 
 
120
                        # if entity doesn't exist in home dir, copy symlink
 
121
                        if dst_type == '_':
 
122
                                if the.verbose > 1: print "  _<l " + rel_file
 
123
                                os.symlink( os.readlink( src_file ), dst_file )
 
124
 
 
125
                        # if entity is a symlink in home dir, replace it only if it differs
 
126
                        elif dst_type == 'l':
 
127
                                if os.readlink( src_file ) != os.readlink( dst_file ):
 
128
                                        if the.verbose > 1: print "  l<l " + rel_file
 
129
                                        os.unlink( dst_file )
 
130
                                        os.symlink( os.readlink( src_file ), dst_file )
 
131
                                else:
 
132
                                        if the.verbose > 1: print "  l=l " + rel_file
 
133
 
 
134
                        # if entity is a file in home dir, replace it with file
 
135
                        elif dst_type == 'f':
 
136
                                if the.verbose > 1: print "  f<l " + rel_file
 
137
                                os.unlink( dst_file )
 
138
                                os.symlink( os.readlink( src_file ), dst_file )
 
139
 
 
140
                        # if entity is a directory in home dir, replace it with file
 
141
                        elif dst_type == 'd':
 
142
                                if the.verbose > 1: print "  d<l " + rel_file
 
143
                                shutil.rmtree( dst_file )
 
144
                                os.symlink( os.readlink( src_file ), dst_file )
 
145
 
 
146
                        else:
 
147
                                raise NotImplementedError()
 
148
 
 
149
                # deleted file
 
150
                if src_type == '_':
 
151
 
 
152
                        # if entity doesn't exist in home dir, we're good
 
153
                        if dst_type == '_':
 
154
                                pass;
 
155
 
 
156
                        # if entity is a file or symlink in home dir, delete it
 
157
                        if dst_type == 'f' or dst_type == 'l':
 
158
                                if the.verbose > 1: print "  %s<_ %s" % ( dst_type, rel_file )
 
159
                                os.unlink( dst_file )
 
160
 
 
161
                        # if entity is a directory in home dir, delete it
 
162
                        elif dst_type == 'd':
 
163
                                if the.verbose > 1: print "  d<_ " + rel_file
 
164
                                shutil.rmtree( dst_file )
 
165
 
 
166
                        else:
 
167
                                raise NotImplementedError()
 
168
 
 
169
                # non-directories can not be recursed in to
 
170
                return False