59
60
# the directory shouldn't exist
60
61
os.mkdir( self.dir )
64
self.run( [ 'bzr', 'init', '.' ] )
65
self.run( [ 'brz', 'init', '.' ] )
65
66
except self.VcsError as e:
67
68
# attempt to clean-up dir
82
83
# the directory shouldn't exist
83
84
os.mkdir( self.dir )
87
self.run( [ 'bzr', 'checkout', url, '.' ] )
88
self.run( [ 'brz', 'checkout', url, '.' ] )
88
89
except self.VcsError as e:
90
91
# attempt to clean-up dir
100
101
"""Obtain some sort of revision identifier
104
output = self.run( [ 'bzr', 'revno', '--tree' ] )
105
output = self.run( [ 'brz', 'revno', '--tree' ] )
107
buf = StringIO.StringIO( output )
108
buf = io.StringIO( output )
108
109
return buf.readline().rstrip()
118
output = self.run( [ 'bzr', 'status' ] )
119
output = self.run( [ 'brz', 'status' ] )
119
120
files = self.parse_file_blocks( output )
121
# remove kind changed files (or they can cause `bzr revert` to break in
122
# remove kind changed files (or they can cause `brz revert` to break in
122
123
# strange situations, like when a directory has been replaced with a
123
124
# symlink to a non-existant file)
124
125
if 'kind changed' in files:
126
127
matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
128
129
raise RunTimeError(
129
'failed to parse bzr kind change: %s' % file )
130
'failed to parse brz kind change: %s' % file )
130
131
file = matches.group( 1 )
131
if the.verbose >= 2: print "removing (kind changed): " + file
132
if the.verbose >= 2: print("removing (kind changed): " + file)
132
133
full_file = os.path.join( self.dir, file )
133
134
if os.path.isfile( full_file ) or os.path.islink( full_file ):
134
135
os.unlink( full_file )
138
139
raise RuntimeError( 'exotic file in repo: %s' % file )
141
self.run( [ 'bzr', 'revert', '--no-backup' ] )
142
self.run( [ 'brz', 'revert', '--no-backup' ] )
144
output = self.run( [ 'bzr', 'status' ] )
145
output = self.run( [ 'brz', 'status' ] )
145
146
files = self.parse_file_blocks( output )
147
148
# remove unknown files
150
151
matches = re.search( r'^(.+?)[/@+]?$', file )
152
153
raise RunTimeError(
153
'failed to parse bzr unknowns: %s' % file )
154
'failed to parse brz unknowns: %s' % file )
154
155
file = matches.group( 1 )
155
if the.verbose >= 2: print "removing (unknown): " + file
156
if the.verbose >= 2: print("removing (unknown): " + file)
156
157
full_file = os.path.join( self.dir, file )
157
158
if os.path.isfile( full_file ) or os.path.islink( full_file ):
158
159
os.unlink( full_file )
164
165
# if a revision identifier has been given, ensure we're updated to that
165
166
if revno is not None and self.get_revno() != revno:
168
self.run( [ 'bzr', 'update', '-r', revno ] )
169
self.run( [ 'brz', 'update', '-r', revno ] )
171
172
def update( self ):
177
# WARNING: the following might cause bzr to ask for your ssh password more than
178
# WARNING: the following might cause brz to ask for your ssh password more than
178
179
# once during an update!!!
181
182
# revno = self.get_revno()
183
184
# # update to current revision (pull in history without updating tree)
184
# self.run( [ 'bzr', 'update', '-r', revno ] )
185
# self.run( [ 'brz', 'update', '-r', revno ] )
186
187
# # get log output
187
188
# next_revno = str( int( revno ) + 1 )
188
# output = self.run( [ 'bzr', 'log', '-r', next_revno + '..' ] )
189
# output = self.run( [ 'brz', 'log', '-r', next_revno + '..' ] )
191
192
# keep_files = list()
213
214
# rename_from = rename_files[ rename_from ]
214
215
# rename_files[ line[ 4 : ] ] = rename_from
216
# bzr update properly
217
output = self.run( [ 'bzr', 'update' ] )
217
# brz update properly
218
output = self.run( [ 'brz', 'update' ] )
219
220
# parse output (see logic in report() in bzrlib/delta.py)
221
buf = StringIO.StringIO( output )
222
buf = io.StringIO( output )
223
224
if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
224
225
line = line.rstrip()
245
246
raise RuntimeError(
246
'failed to parse bzr update output line:\n%s' % line )
247
'failed to parse brz update output line:\n%s' % line )
253
"""Get a list of any local modifications. This method returns a list of files
259
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
262
return self.parse_file_blocks( output )
251
265
def has_changes( self ):
252
266
"""Check if the branch has any local modifications.
256
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
270
output = self.run( [ 'brz', 'status', '--no-pending' ] )
259
273
files = self.parse_file_blocks( output )
264
278
"""Return a list of files that have conflicts.
268
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
282
output = self.run( [ 'brz', 'status', '--no-pending' ] )
271
285
files = self.parse_file_blocks( output )
277
291
@param files a list of relative filenames
281
self.run( [ 'bzr', 'add', '-N' ] + files )
295
self.run( [ 'brz', 'add', '-N' ] + files )
284
298
def commit( self ):
285
299
"""Commit changes to the repo.
289
self.run( [ 'bzr', 'commit', '-m', '' ] )
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 );
292
314
def run( self, cmd ):
293
if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
315
if the.verbose >= 2: print('exec: %s' % ' '.join( cmd ))
294
316
p = Popen( cmd, cwd = self.dir,
295
317
stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
296
output = p.communicate()[ 0 ]
318
output = p.communicate()[ 0 ].decode()
297
319
if p.returncode > 0:
298
320
raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
299
321
if the.verbose >= 2:
300
322
verbose_output = output.rstrip()
301
323
if len( verbose_output ):
302
print re.sub( '(^|\n)', '\\1 : ', verbose_output )
324
print(re.sub( '(^|\n)', '\\1 : ', verbose_output ))
306
328
def parse_file_blocks( self, output ):
309
buf = StringIO.StringIO( output )
331
buf = io.StringIO( output )
311
333
matches = re.search( '^([a-z ]+):$', line, re.I )