/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-05 19:22:21 UTC
  • Revision ID: tim@ed.am-20140405192221-yl8xgy3qorbahlfw
implemented CopyInWalker in terms of CopyBaseWalker, changed implementation of
the verbose operation print function for readability

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/stdhome/vcs/bzr.py
34
34
 
35
35
                @param dir the fully-qualified directory to work in.
36
36
                """
37
 
 
38
37
                self.dir = dir
39
 
                self.ignored_files = [ '.bzr', '.bzrignore' ]
40
 
 
41
 
 
42
 
        def has_authority( self ):
43
 
                """Check that the directory is under this VCS's control.
44
 
                """
45
 
 
46
 
                return os.path.exists( os.path.join( self.dir, '.bzr' ) )
47
 
 
48
 
 
49
 
        def expand_repo_url( self, url ):
50
 
                """Convert a simple hostname in to an URL that the VCS can use.
51
 
                """
52
 
 
53
 
                return 'bzr+ssh://%s/%s/%s' % ( url, the.dir, the.repo.name )
54
38
 
55
39
 
56
40
        def init( self ):
101
85
                """Obtain some sort of revision identifier
102
86
                """
103
87
 
104
 
                # bzr revno
 
88
                # bzr revert
105
89
                output = self.run( [ 'bzr', 'revno', '--tree' ] )
106
90
 
107
91
                # parse revno
115
99
                revision.
116
100
                """
117
101
 
118
 
                # bzr st
119
 
                output = self.run( [ 'bzr', 'status' ] )
120
 
                files = self.parse_file_blocks( output )
121
 
 
122
 
                # remove kind changed files (or they can cause `bzr revert` to break in
123
 
                # strange situations, like when a directory has been replaced with a
124
 
                # symlink to a non-existant file)
125
 
                if 'kind changed' in files:
126
 
                        for file in files[ 'kind changed' ]:
127
 
                                matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
128
 
                                if not matches:
129
 
                                        raise RunTimeError(
130
 
                                                'failed to parse bzr kind change: %s' % file )
131
 
                                file = matches.group( 1 )
132
 
                                if the.verbose >= 2: print "removing (kind changed): " + file
133
 
                                full_file = os.path.join( self.dir, file )
134
 
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
135
 
                                        os.unlink( full_file )
136
 
                                elif os.path.isdir( full_file ):
137
 
                                        shutil.rmtree( full_file )
138
 
                                else:
139
 
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
140
 
 
141
 
                # bzr revert
 
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' ] )
142
105
                self.run( [ 'bzr', 'revert', '--no-backup' ] )
143
106
 
144
107
                # bzr st
148
111
                # remove unknown files
149
112
                if 'unknown' in files:
150
113
                        for file in files[ 'unknown' ]:
151
 
                                matches = re.search( r'^(.+?)[/@+]?$', file )
152
 
                                if not matches:
153
 
                                        raise RunTimeError(
154
 
                                                'failed to parse bzr unknowns: %s' % file )
155
 
                                file = matches.group( 1 )
156
 
                                if the.verbose >= 2: print "removing (unknown): " + file
 
114
                                if the.verbose > 1: print "removing unknown: " + file
157
115
                                full_file = os.path.join( self.dir, file )
158
 
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
 
116
                                if os.path.isfile( full_file ):
159
117
                                        os.unlink( full_file )
160
118
                                elif os.path.isdir( full_file ):
161
119
                                        shutil.rmtree( full_file )
162
120
                                else:
163
121
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
164
122
 
165
 
                # if a revision identifier has been given, ensure we're updated to that
166
 
                if revno is not None and self.get_revno() != revno:
 
123
                # if a revision identifyer has been given, update to that
 
124
                if revno is not None:
167
125
 
168
126
                        # bzr update
169
127
                        self.run( [ 'bzr', 'update', '-r', revno ] )
223
181
                for line in buf:
224
182
                        if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
225
183
                        line = line.rstrip()
 
184
                        if the.verbose > 1: print '  %s' % line
226
185
 
227
186
                        # renames show before and after file names
228
187
                        matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
244
203
                                continue
245
204
 
246
205
                        raise RuntimeError(
247
 
                                'failed to parse bzr update output line:\n%s' % line )
 
206
                                'failed to parse bzr update output line:\n%' % line )
248
207
 
249
208
                return files
250
209
 
273
232
                return files['conflicts'] if 'conflicts' in files else None
274
233
 
275
234
 
276
 
        def add( self, files ):
277
 
                """Make sure files are added to version control.
278
 
                @param files a list of relative filenames
279
 
                """
280
 
 
281
 
                # bzr add
282
 
                self.run( [ 'bzr', 'add', '-N' ] + files )
283
 
 
284
 
 
285
 
        def commit( self ):
286
 
                """Commit changes to the repo.
287
 
                """
288
 
 
289
 
                # bzr commit
290
 
                try:
291
 
                        self.run( [ 'bzr', 'commit', '-m', '' ] )
292
 
                except self.VcsError as e:
293
 
                        if re.search( '/Working tree is out of date, please/', e.output ):
294
 
                                raise the.program.FatalError(
295
 
                                        'you must update your files first.\n' +
296
 
                                        'Hint: see "%s update --help"' % the.program.name );
297
 
 
298
 
 
299
235
        def run( self, cmd ):
300
 
                if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
 
236
                if the.verbose > 1: print 'exec: %s' % ' '.join( cmd )
301
237
                p = Popen( cmd, cwd = self.dir,
302
238
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
303
239
                output = p.communicate()[ 0 ]
304
240
                if p.returncode > 0:
305
241
                        raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
306
 
                if the.verbose >= 2:
307
 
                        verbose_output = output.rstrip()
308
 
                        if len( verbose_output ):
309
 
                                print re.sub( '(^|\n)', '\\1  : ', verbose_output )
310
242
                return output
311
243
 
312
244
 
326
258
                                                res[ current ] = list()
327
259
                                        res[ current ].append( matches.group( 1 ) )
328
260
                                        continue
329
 
                        if re.search( '^[0-9]+ shel(?:f|ves) exists?', line ): continue
 
261
                        if re.search( '^[0-9]+ shelf exists', line ): continue
330
262
                        if re.search( '^working tree is out of date', line ): continue
331
263
                        raise self.ParseError( "unrecognised line: %s" % line )
332
264
                return res