/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-04 22:35:04 UTC
  • Revision ID: tim@ed.am-20140404223504-z91mtjgeljckqyg8
updated comment

Show diffs side-by-side

added added

removed removed

Lines of Context:
34
34
 
35
35
                @param dir the fully-qualified directory to work in.
36
36
                """
37
 
 
38
37
                self.dir = dir
39
38
 
40
39
 
86
85
                """Obtain some sort of revision identifier
87
86
                """
88
87
 
89
 
                # bzr revno
 
88
                # bzr revert
90
89
                output = self.run( [ 'bzr', 'revno', '--tree' ] )
91
90
 
92
91
                # parse revno
100
99
                revision.
101
100
                """
102
101
 
103
 
                # bzr st
104
 
                output = self.run( [ 'bzr', 'status' ] )
105
 
                files = self.parse_file_blocks( output )
106
 
 
107
 
                # remove kind changed files (or they can cause `bzr revert` to break in
108
 
                # strange situations, like when a directory has been replaced with a
109
 
                # symlink to a non-existant file)
110
 
                if 'kind changed' in files:
111
 
                        for file in files[ 'kind changed' ]:
112
 
                                matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
113
 
                                if not matches:
114
 
                                        raise RunTimeError(
115
 
                                                'failed to parse bzr kind change: %s' % file )
116
 
                                file = matches.group( 1 )
117
 
                                if the.verbose >= 2: print "removing (kind changed): " + file
118
 
                                full_file = os.path.join( self.dir, file )
119
 
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
120
 
                                        os.unlink( full_file )
121
 
                                elif os.path.isdir( full_file ):
122
 
                                        shutil.rmtree( full_file )
123
 
                                else:
124
 
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
125
 
 
126
102
                # bzr revert
127
103
                self.run( [ 'bzr', 'revert', '--no-backup' ] )
128
104
 
133
109
                # remove unknown files
134
110
                if 'unknown' in files:
135
111
                        for file in files[ 'unknown' ]:
136
 
                                matches = re.search( r'^(.+?)[/@+]?$', file )
137
 
                                if not matches:
138
 
                                        raise RunTimeError(
139
 
                                                'failed to parse bzr unknowns: %s' % file )
140
 
                                file = matches.group( 1 )
141
 
                                if the.verbose >= 2: print "removing (unknown): " + file
142
112
                                full_file = os.path.join( self.dir, file )
143
 
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
 
113
                                if os.path.isfile( full_file ):
144
114
                                        os.unlink( full_file )
145
 
                                elif os.path.isdir( full_file ):
 
115
                                elif os.full_file.isdir( full_file ):
146
116
                                        shutil.rmtree( full_file )
147
117
                                else:
148
118
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
149
119
 
150
 
                # if a revision identifier has been given, ensure we're updated to that
151
 
                if revno is not None and self.get_revno() != revno:
 
120
                # if a revision identifyer has been given, update to that
 
121
                if revno is not None:
152
122
 
153
123
                        # bzr update
154
124
                        self.run( [ 'bzr', 'update', '-r', revno ] )
208
178
                for line in buf:
209
179
                        if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
210
180
                        line = line.rstrip()
 
181
                        if the.verbose > 1: print '  %s' % line
211
182
 
212
183
                        # renames show before and after file names
213
184
                        matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
229
200
                                continue
230
201
 
231
202
                        raise RuntimeError(
232
 
                                'failed to parse bzr update output line:\n%s' % line )
 
203
                                'failed to parse bzr update output line:\n%' % line )
233
204
 
234
205
                return files
235
206
 
258
229
                return files['conflicts'] if 'conflicts' in files else None
259
230
 
260
231
 
261
 
        def add( self, files ):
262
 
                """Make sure files are added to version control.
263
 
                @param files a list of relative filenames
264
 
                """
265
 
 
266
 
                # bzr add
267
 
                self.run( [ 'bzr', 'add', '-N' ] + files )
268
 
 
269
 
 
270
 
        def commit( self ):
271
 
                """Commit changes to the repo.
272
 
                """
273
 
 
274
 
                # bzr commit
275
 
                self.run( [ 'bzr', 'commit', '-m', '' ] )
276
 
 
277
 
 
278
232
        def run( self, cmd ):
279
 
                if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
 
233
                if the.verbose > 1: print 'exec: %s' % ' '.join( cmd )
280
234
                p = Popen( cmd, cwd = self.dir,
281
235
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
282
236
                output = p.communicate()[ 0 ]
283
237
                if p.returncode > 0:
284
238
                        raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
285
 
                if the.verbose >= 2:
286
 
                        print re.sub( '(^|\n)', '\\1  > ', output.rstrip() )
287
239
                return output
288
240
 
289
241
 
303
255
                                                res[ current ] = list()
304
256
                                        res[ current ].append( matches.group( 1 ) )
305
257
                                        continue
306
 
                        if re.search( '^[0-9]+ shel(?:f|ves) exists?', line ): continue
 
258
                        if re.search( '^[0-9]+ shelf exists', line ): continue
307
259
                        if re.search( '^working tree is out of date', line ): continue
308
260
                        raise self.ParseError( "unrecognised line: %s" % line )
309
261
                return res