/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-12 22:36:44 UTC
  • Revision ID: tim@ed.am-20140312223644-9pehua402pww0op7
fixed incorrect variable name

Show diffs side-by-side

added added

removed removed

 
1
# copy_out.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 CopyOutWalker( Walker ):
 
28
        """The copy-out walker traverses the repo, copying its content to the home
 
29
        directory.  It is run *after* checking for conflicts that might occur during
 
30
        copy-out, so any conflicts that exist can be assumed to be unimportant and
 
31
        disposable.  In other words, this will happily overwrite stuff in the home
 
32
        directory, without regard to its content.
 
33
 
 
34
        Walker source:       repo
 
35
        Walker destination:  home dir
 
36
        Walker traversing:   repo (or a provided file list, which may include stuff
 
37
                                                        that no longer exists in the repo
 
38
        """
 
39
 
 
40
        def __init__( self, updated_files = None ):
 
41
                self.src_dir = the.repo.full_dir
 
42
                self.dst_dir = the.full_home_dir
 
43
                self.walk_list = updated_files if updated_files is not None else \
 
44
                                                 self.generate_walk_list( the.repo.full_dir )
 
45
 
 
46
 
 
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