/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-04-17 21:00:42 UTC
  • Revision ID: tim@ed.am-20160417210042-njcd725axg87i20j
renamed tools dir to dev

Show diffs side-by-side

added added

removed removed

lib/stdhome/vcs/bzr.py
1
 
# brz.py
 
1
# bzr.py
2
2
#
3
3
# Copyright (C) 2014 Tim Marston <tim@edm.am>
4
4
#
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 ):
60
59
                # the directory shouldn't exist
61
60
                os.mkdir( self.dir )
62
61
 
63
 
                # brz init
 
62
                # bzr init
64
63
                try:
65
 
                        self.run( [ 'brz', 'init', '.' ] )
 
64
                        self.run( [ 'bzr', 'init', '.' ] )
66
65
                except self.VcsError as e:
67
66
 
68
67
                        # attempt to clean-up dir
83
82
                # the directory shouldn't exist
84
83
                os.mkdir( self.dir )
85
84
 
86
 
                # brz co
 
85
                # bzr co
87
86
                try:
88
 
                        self.run( [ 'brz', 'checkout', url, '.' ] )
 
87
                        self.run( [ 'bzr', 'checkout', url, '.' ] )
89
88
                except self.VcsError as e:
90
89
 
91
90
                        # attempt to clean-up dir
101
100
                """Obtain some sort of revision identifier
102
101
                """
103
102
 
104
 
                # brz revno
105
 
                output = self.run( [ 'brz', 'revno', '--tree' ] )
 
103
                # bzr revno
 
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
 
115
114
                revision.
116
115
                """
117
116
 
118
 
                # brz st
119
 
                output = self.run( [ 'brz', 'status' ] )
 
117
                # bzr st
 
118
                output = self.run( [ 'bzr', 'status' ] )
120
119
                files = self.parse_file_blocks( output )
121
120
 
122
 
                # remove kind changed files (or they can cause `brz revert` to break in
 
121
                # remove kind changed files (or they can cause `bzr revert` to break in
123
122
                # strange situations, like when a directory has been replaced with a
124
123
                # symlink to a non-existant file)
125
124
                if 'kind changed' in files:
127
126
                                matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
128
127
                                if not matches:
129
128
                                        raise RunTimeError(
130
 
                                                'failed to parse brz kind change: %s' % file )
 
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 )
138
137
                                else:
139
138
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
140
139
 
141
 
                # brz revert
142
 
                self.run( [ 'brz', 'revert', '--no-backup' ] )
 
140
                # bzr revert
 
141
                self.run( [ 'bzr', 'revert', '--no-backup' ] )
143
142
 
144
 
                # brz st
145
 
                output = self.run( [ 'brz', 'status' ] )
 
143
                # bzr st
 
144
                output = self.run( [ 'bzr', 'status' ] )
146
145
                files = self.parse_file_blocks( output )
147
146
 
148
147
                # remove unknown files
151
150
                                matches = re.search( r'^(.+?)[/@+]?$', file )
152
151
                                if not matches:
153
152
                                        raise RunTimeError(
154
 
                                                'failed to parse brz unknowns: %s' % file )
 
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 )
165
164
                # if a revision identifier has been given, ensure we're updated to that
166
165
                if revno is not None and self.get_revno() != revno:
167
166
 
168
 
                        # brz update
169
 
                        self.run( [ 'brz', 'update', '-r', revno ] )
 
167
                        # bzr update
 
168
                        self.run( [ 'bzr', 'update', '-r', revno ] )
170
169
 
171
170
 
172
171
        def update( self ):
175
174
                operation.
176
175
                """
177
176
 
178
 
#               WARNING: the following might cause brz to ask for your ssh password more than
 
177
#               WARNING: the following might cause bzr to ask for your ssh password more than
179
178
#               once during an update!!!
180
179
#
181
180
#               # get revno
182
181
#               revno = self.get_revno()
183
182
#
184
183
#               # update to current revision (pull in history without updating tree)
185
 
#               self.run( [ 'brz', 'update', '-r', revno ] )
 
184
#               self.run( [ 'bzr', 'update', '-r', revno ] )
186
185
#
187
186
#               # get log output
188
187
#               next_revno = str( int( revno ) + 1 )
189
 
#               output = self.run( [ 'brz', 'log', '-r', next_revno + '..' ] )
 
188
#               output = self.run( [ 'bzr', 'log', '-r', next_revno + '..' ] )
190
189
#
191
190
#               # parse output
192
191
#               keep_files = list()
214
213
#                                                       rename_from = rename_files[ rename_from ]
215
214
#                                               rename_files[ line[ 4 : ] ] = rename_from
216
215
 
217
 
                # brz update properly
218
 
                output = self.run( [ 'brz', 'update' ] )
 
216
                # bzr update properly
 
217
                output = self.run( [ 'bzr', 'update' ] )
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()
244
243
                                continue
245
244
 
246
245
                        raise RuntimeError(
247
 
                                'failed to parse brz update output line:\n%s' % line )
 
246
                                'failed to parse bzr update output line:\n%s' % line )
248
247
 
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
                """
268
254
 
269
 
                # brz status
270
 
                output = self.run( [ 'brz', 'status', '--no-pending' ] )
 
255
                # bzr status
 
256
                output = self.run( [ 'bzr', 'status', '--no-pending' ] )
271
257
 
272
258
                # parse output
273
259
                files = self.parse_file_blocks( output )
278
264
                """Return a list of files that have conflicts.
279
265
                """
280
266
 
281
 
                # brz status
282
 
                output = self.run( [ 'brz', 'status', '--no-pending' ] )
 
267
                # bzr status
 
268
                output = self.run( [ 'bzr', 'status', '--no-pending' ] )
283
269
 
284
270
                # parse output
285
271
                files = self.parse_file_blocks( output )
291
277
                @param files a list of relative filenames
292
278
                """
293
279
 
294
 
                # brz add
295
 
                self.run( [ 'brz', 'add', '-N' ] + files )
 
280
                # bzr add
 
281
                self.run( [ 'bzr', 'add', '-N' ] + files )
296
282
 
297
283
 
298
284
        def commit( self ):
299
285
                """Commit changes to the repo.
300
286
                """
301
287
 
302
 
                # brz commit
303
 
                try:
304
 
                        self.run( [ 'brz', '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
 
288
                # bzr commit
 
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: