19
19
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
import subprocess, os, re, shutil
27
23
from subprocess import Popen
29
26
from stdhome import the
37
32
def __init__( self, dir ):
65
59
# the directory shouldn't exist
66
60
os.mkdir( self.dir )
70
self.run( [ 'brz', 'init', '.' ] )
64
self.run( [ 'bzr', 'init', '.' ] )
71
65
except self.VcsError as e:
73
67
# attempt to clean-up dir
88
82
# the directory shouldn't exist
89
83
os.mkdir( self.dir )
93
self.run( [ 'brz', 'checkout', url, '.' ] )
87
self.run( [ 'bzr', 'checkout', url, '.' ] )
94
88
except self.VcsError as e:
96
90
# attempt to clean-up dir
106
100
"""Obtain some sort of revision identifier
110
output = self.run( [ 'brz', 'revno', '--tree' ] )
104
output = self.run( [ 'bzr', 'revno', '--tree' ] )
113
buf = io.StringIO( output )
107
buf = StringIO.StringIO( output )
114
108
return buf.readline().rstrip()
124
output = self.run( [ 'brz', 'status' ] )
118
output = self.run( [ 'bzr', 'status' ] )
125
119
files = self.parse_file_blocks( output )
127
# 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
128
122
# strange situations, like when a directory has been replaced with a
129
123
# symlink to a non-existant file)
130
124
if 'kind changed' in files:
132
126
matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
134
128
raise RunTimeError(
135
'failed to parse brz kind change: %s' % file )
129
'failed to parse bzr kind change: %s' % file )
136
130
file = matches.group( 1 )
137
if the.verbose >= 2: print("removing (kind changed): " + file)
131
if the.verbose >= 2: print "removing (kind changed): " + file
138
132
full_file = os.path.join( self.dir, file )
139
133
if os.path.isfile( full_file ) or os.path.islink( full_file ):
140
134
os.unlink( full_file )
144
138
raise RuntimeError( 'exotic file in repo: %s' % file )
147
self.run( [ 'brz', 'revert', '--no-backup' ] )
141
self.run( [ 'bzr', 'revert', '--no-backup' ] )
150
output = self.run( [ 'brz', 'status' ] )
144
output = self.run( [ 'bzr', 'status' ] )
151
145
files = self.parse_file_blocks( output )
153
147
# remove unknown files
156
150
matches = re.search( r'^(.+?)[/@+]?$', file )
158
152
raise RunTimeError(
159
'failed to parse brz unknowns: %s' % file )
153
'failed to parse bzr unknowns: %s' % file )
160
154
file = matches.group( 1 )
161
if the.verbose >= 2: print("removing (unknown): " + file)
155
if the.verbose >= 2: print "removing (unknown): " + file
162
156
full_file = os.path.join( self.dir, file )
163
157
if os.path.isfile( full_file ) or os.path.islink( full_file ):
164
158
os.unlink( full_file )
170
164
# if a revision identifier has been given, ensure we're updated to that
171
165
if revno is not None and self.get_revno() != revno:
174
self.run( [ 'brz', 'update', '-r', revno ] )
168
self.run( [ 'bzr', 'update', '-r', revno ] )
177
171
def update( self ):
183
# 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
184
178
# once during an update!!!
187
181
# revno = self.get_revno()
189
183
# # update to current revision (pull in history without updating tree)
190
# self.run( [ 'brz', 'update', '-r', revno ] )
184
# self.run( [ 'bzr', 'update', '-r', revno ] )
192
186
# # get log output
193
187
# next_revno = str( int( revno ) + 1 )
194
# output = self.run( [ 'brz', 'log', '-r', next_revno + '..' ] )
188
# output = self.run( [ 'bzr', 'log', '-r', next_revno + '..' ] )
197
191
# keep_files = list()
219
213
# rename_from = rename_files[ rename_from ]
220
214
# rename_files[ line[ 4 : ] ] = rename_from
222
# brz update properly
223
output = self.run( [ 'brz', 'update' ] )
216
# bzr update properly
217
output = self.run( [ 'bzr', 'update' ] )
225
219
# parse output (see logic in report() in bzrlib/delta.py)
227
buf = io.StringIO( output )
221
buf = StringIO.StringIO( output )
229
223
if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
230
224
line = line.rstrip()
251
245
raise RuntimeError(
252
'failed to parse brz update output line:\n%s' % line )
246
'failed to parse bzr update output line:\n%s' % line )
258
"""Get a list of any local modifications. This method returns a list of files
264
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
267
return self.parse_file_blocks( output )
270
251
def has_changes( self ):
271
252
"""Check if the branch has any local modifications.
275
output = self.run( [ 'brz', 'status', '--no-pending' ] )
256
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
278
259
files = self.parse_file_blocks( output )
283
264
"""Return a list of files that have conflicts.
287
output = self.run( [ 'brz', 'status', '--no-pending' ] )
268
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
290
271
files = self.parse_file_blocks( output )
296
277
@param files a list of relative filenames
300
self.run( [ 'brz', 'add', '-N' ] + files )
281
self.run( [ 'bzr', 'add', '-N' ] + files )
303
284
def commit( self ):
304
285
"""Commit changes to the repo.
309
self.run( [ 'brz', 'commit', '-m', '' ] )
310
except self.VcsError as e:
311
if re.search( 'Working tree is out of date', e.output ):
312
raise the.program.FatalError(
313
'you must update your files first.\n' +
314
'Hint: see "%s update --help"' % the.program.name );
289
self.run( [ 'bzr', 'commit', '-m', '' ] )
319
292
def run( self, cmd ):
320
if the.verbose >= 2: print('exec: %s' % ' '.join( cmd ))
293
if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
321
294
p = Popen( cmd, cwd = self.dir,
322
295
stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
323
output = p.communicate()[ 0 ].decode()
296
output = p.communicate()[ 0 ]
324
297
if p.returncode > 0:
325
298
raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
326
299
if the.verbose >= 2:
327
300
verbose_output = output.rstrip()
328
301
if len( verbose_output ):
329
print(re.sub( '(^|\n)', '\\1 : ', verbose_output ))
302
print re.sub( '(^|\n)', '\\1 : ', verbose_output )
333
306
def parse_file_blocks( self, output ):
336
buf = io.StringIO( output )
309
buf = StringIO.StringIO( output )
338
311
matches = re.search( '^([a-z ]+):$', line, re.I )