/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: 2016-05-22 16:22:07 UTC
  • Revision ID: tim@ed.am-20160522162207-4b2i6gfejzev6o8g
add symlink in ~/bin to dev version in dev script

Show diffs side-by-side

added added

removed removed

lib/stdhome/vcs/bzr.py
21
21
 
22
22
import subprocess, os, re, shutil
23
23
from subprocess import Popen
24
 
import io
25
 
from .vcs import Vcs
 
24
import StringIO
 
25
from vcs import Vcs
26
26
from stdhome import the
27
27
 
28
28
 
36
36
                """
37
37
 
38
38
                self.dir = dir
39
 
                self.ignored_files = [ '.bzr', '.bzrignore' ]
40
39
 
41
40
 
42
41
        def has_authority( self ):
105
104
                output = self.run( [ 'bzr', 'revno', '--tree' ] )
106
105
 
107
106
                # parse revno
108
 
                buf = io.StringIO( output )
 
107
                buf = StringIO.StringIO( output )
109
108
                return buf.readline().rstrip()
110
109
 
111
110
 
129
128
                                        raise RunTimeError(
130
129
                                                'failed to parse bzr kind change: %s' % file )
131
130
                                file = matches.group( 1 )
132
 
                                if the.verbose >= 2: print("removing (kind changed): " + file)
 
131
                                if the.verbose >= 2: print "removing (kind changed): " + file
133
132
                                full_file = os.path.join( self.dir, file )
134
133
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
135
134
                                        os.unlink( full_file )
153
152
                                        raise RunTimeError(
154
153
                                                'failed to parse bzr unknowns: %s' % file )
155
154
                                file = matches.group( 1 )
156
 
                                if the.verbose >= 2: print("removing (unknown): " + file)
 
155
                                if the.verbose >= 2: print "removing (unknown): " + file
157
156
                                full_file = os.path.join( self.dir, file )
158
157
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
159
158
                                        os.unlink( full_file )
219
218
 
220
219
                # parse output (see logic in report() in bzrlib/delta.py)
221
220
                files = list()
222
 
                buf = io.StringIO( output )
 
221
                buf = StringIO.StringIO( output )
223
222
                for line in buf:
224
223
                        if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
225
224
                        line = line.rstrip()
249
248
                return files
250
249
 
251
250
 
252
 
        def status( self ):
253
 
                """Get a list of any local modifications.  This method returns a list of files
254
 
                which are modified.
255
 
 
256
 
                """
257
 
 
258
 
                # bzr status
259
 
                output = self.run( [ 'bzr', 'status', '--no-pending' ] )
260
 
 
261
 
                # parse output
262
 
                return self.parse_file_blocks( output )
263
 
 
264
 
 
265
251
        def has_changes( self ):
266
252
                """Check if the branch has any local modifications.
267
253
                """
300
286
                """
301
287
 
302
288
                # bzr commit
303
 
                try:
304
 
                        self.run( [ 'bzr', 'commit', '-m', '' ] )
305
 
                except self.VcsError as e:
306
 
                        if re.search( 'Working tree is out of date', e.output ):
307
 
                                raise the.program.FatalError(
308
 
                                        'you must update your files first.\n' +
309
 
                                        'Hint: see "%s update --help"' % the.program.name );
310
 
                        else:
311
 
                                raise e
 
289
                self.run( [ 'bzr', 'commit', '-m', '' ] )
312
290
 
313
291
 
314
292
        def run( self, cmd ):
315
 
                if the.verbose >= 2: print('exec: %s' % ' '.join( cmd ))
 
293
                if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
316
294
                p = Popen( cmd, cwd = self.dir,
317
295
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
318
 
                output = p.communicate()[ 0 ].decode()
 
296
                output = p.communicate()[ 0 ]
319
297
                if p.returncode > 0:
320
298
                        raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
321
299
                if the.verbose >= 2:
322
300
                        verbose_output = output.rstrip()
323
301
                        if len( verbose_output ):
324
 
                                print(re.sub( '(^|\n)', '\\1  : ', verbose_output ))
 
302
                                print re.sub( '(^|\n)', '\\1  : ', verbose_output )
325
303
                return output
326
304
 
327
305
 
328
306
        def parse_file_blocks( self, output ):
329
307
                res = dict()
330
308
                current = None
331
 
                buf = io.StringIO( output )
 
309
                buf = StringIO.StringIO( output )
332
310
                for line in buf:
333
311
                        matches = re.search( '^([a-z ]+):$', line, re.I )
334
312
                        if matches: