/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/vcs/bzr.py

  • Committer: Tim Marston
  • Date: 2014-04-18 14:32:26 UTC
  • Revision ID: tim@ed.am-20140418143226-d3ih1aafs6mg9ndw
made revert require --all to act on all files

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/stdhome/vcs/bzr.py
99
99
                revision.
100
100
                """
101
101
 
102
 
                # bzr revert (run twice to handle a bug in bzr where reverting a
103
 
                # directory from a symlink can cause conflicts during initial revert)
104
 
                self.run( [ 'bzr', 'revert', '--no-backup' ] )
 
102
                # bzr st
 
103
                output = self.run( [ 'bzr', 'status' ] )
 
104
                files = self.parse_file_blocks( output )
 
105
 
 
106
                # remove kind changed files (or they can cause `bzr revert` to break in
 
107
                # strange situations, like when a directory has been replaced with a
 
108
                # symlink to a non-existant file)
 
109
                if 'kind changed' in files:
 
110
                        for file in files[ 'kind changed' ]:
 
111
                                matches = re.match( r'(.+?)[/@+]? \([^)]+\)', file )
 
112
                                if not matches:
 
113
                                        raise RunTimeError(
 
114
                                                'failed to parse bzr kind change: %s' % file )
 
115
                                file = matches.group( 1 )
 
116
                                if the.verbose >= 2: print "removing (kind changed): " + file
 
117
                                full_file = os.path.join( self.dir, file )
 
118
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
 
119
                                        os.unlink( full_file )
 
120
                                elif os.path.isdir( full_file ):
 
121
                                        shutil.rmtree( full_file )
 
122
                                else:
 
123
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
 
124
 
 
125
                # bzr revert
105
126
                self.run( [ 'bzr', 'revert', '--no-backup' ] )
106
127
 
107
128
                # bzr st
111
132
                # remove unknown files
112
133
                if 'unknown' in files:
113
134
                        for file in files[ 'unknown' ]:
114
 
                                if the.verbose > 1: print "removing unknown: " + file
 
135
                                if the.verbose >= 2: print "removing (unknown): " + file
115
136
                                full_file = os.path.join( self.dir, file )
116
137
                                if os.path.isfile( full_file ):
117
138
                                        os.unlink( full_file )
181
202
                for line in buf:
182
203
                        if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
183
204
                        line = line.rstrip()
184
 
                        if the.verbose > 1: print '  %s' % line
 
205
                        if the.verbose >= 2: print '  %s' % line
185
206
 
186
207
                        # renames show before and after file names
187
208
                        matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
203
224
                                continue
204
225
 
205
226
                        raise RuntimeError(
206
 
                                'failed to parse bzr update output line:\n%' % line )
 
227
                                'failed to parse bzr update output line:\n%s' % line )
207
228
 
208
229
                return files
209
230
 
233
254
 
234
255
 
235
256
        def run( self, cmd ):
236
 
                if the.verbose > 1: print 'exec: %s' % ' '.join( cmd )
 
257
                if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
237
258
                p = Popen( cmd, cwd = self.dir,
238
259
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
239
260
                output = p.communicate()[ 0 ]
258
279
                                                res[ current ] = list()
259
280
                                        res[ current ].append( matches.group( 1 ) )
260
281
                                        continue
261
 
                        if re.search( '^[0-9]+ shelf exists', line ): continue
 
282
                        if re.search( '^[0-9]+ shel(?:f|ves) exists?', line ): continue
262
283
                        if re.search( '^working tree is out of date', line ): continue
263
284
                        raise self.ParseError( "unrecognised line: %s" % line )
264
285
                return res