35
35
@param dir the fully-qualified directory to work in.
39
self.ignored_files = [ '.bzr', '.bzrignore' ]
42
def has_authority( self ):
43
"""Check that the directory is under this VCS's control.
46
return os.path.exists( os.path.join( self.dir, '.bzr' ) )
49
def expand_repo_url( self, url ):
50
"""Convert a simple hostname in to an URL that the VCS can use.
53
return 'bzr+ssh://%s/%s/%s' % ( url, the.dir, the.repo.name )
60
44
# the directory shouldn't exist
61
45
os.mkdir( self.dir )
65
self.run( [ 'brz', 'init', '.' ] )
49
self.run( [ 'bzr', 'init', '.' ] )
66
50
except self.VcsError as e:
68
52
# attempt to clean-up dir
83
67
# the directory shouldn't exist
84
68
os.mkdir( self.dir )
88
self.run( [ 'brz', 'checkout', url, '.' ] )
72
self.run( [ 'bzr', 'checkout', url, '.' ] )
89
73
except self.VcsError as e:
91
75
# attempt to clean-up dir
101
85
"""Obtain some sort of revision identifier
105
output = self.run( [ 'brz', 'revno', '--tree' ] )
89
output = self.run( [ 'bzr', 'revno', '--tree' ] )
108
buf = io.StringIO( output )
92
buf = StringIO.StringIO( output )
109
93
return buf.readline().rstrip()
119
output = self.run( [ 'brz', 'status' ] )
120
files = self.parse_file_blocks( output )
122
# remove kind changed files (or they can cause `brz 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 )
130
'failed to parse brz 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 )
139
raise RuntimeError( 'exotic file in repo: %s' % file )
142
self.run( [ 'brz', 'revert', '--no-backup' ] )
145
output = self.run( [ 'brz', 'status' ] )
103
self.run( [ 'bzr', 'revert', '--no-backup' ] )
106
output = self.run( [ 'bzr', 'status' ] )
146
107
files = self.parse_file_blocks( output )
148
109
# remove unknown files
149
110
if 'unknown' in files:
150
111
for file in files[ 'unknown' ]:
151
matches = re.search( r'^(.+?)[/@+]?$', file )
154
'failed to parse brz unknowns: %s' % file )
155
file = matches.group( 1 )
156
if the.verbose >= 2: print("removing (unknown): " + file)
157
112
full_file = os.path.join( self.dir, file )
158
if os.path.isfile( full_file ) or os.path.islink( full_file ):
113
if os.path.isfile( full_file ):
159
114
os.unlink( full_file )
160
elif os.path.isdir( full_file ):
115
elif os.full_file.isdir( full_file ):
161
116
shutil.rmtree( full_file )
163
118
raise RuntimeError( 'exotic file in repo: %s' % file )
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:
120
# if a revision identifyer has been given, update to that
121
if revno is not None:
169
self.run( [ 'brz', 'update', '-r', revno ] )
124
self.run( [ 'bzr', 'update', '-r', revno ] )
172
127
def update( self ):
178
# WARNING: the following might cause brz to ask for your ssh password more than
133
# WARNING: the following might cause bzr to ask for your ssh password more than
179
134
# once during an update!!!
182
137
# revno = self.get_revno()
184
139
# # update to current revision (pull in history without updating tree)
185
# self.run( [ 'brz', 'update', '-r', revno ] )
140
# self.run( [ 'bzr', 'update', '-r', revno ] )
187
142
# # get log output
188
143
# next_revno = str( int( revno ) + 1 )
189
# output = self.run( [ 'brz', 'log', '-r', next_revno + '..' ] )
144
# output = self.run( [ 'bzr', 'log', '-r', next_revno + '..' ] )
192
147
# keep_files = list()
214
169
# rename_from = rename_files[ rename_from ]
215
170
# rename_files[ line[ 4 : ] ] = rename_from
217
# brz update properly
218
output = self.run( [ 'brz', 'update' ] )
172
# bzr update properly
173
output = self.run( [ 'bzr', 'update' ] )
220
175
# parse output (see logic in report() in bzrlib/delta.py)
222
buf = io.StringIO( output )
177
buf = StringIO.StringIO( output )
224
179
if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
225
180
line = line.rstrip()
181
if the.verbose > 1: print ' %s' % line
227
183
# renames show before and after file names
228
184
matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
246
202
raise RuntimeError(
247
'failed to parse brz update output line:\n%s' % line )
203
'failed to parse bzr update output line:\n%' % 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 )
265
208
def has_changes( self ):
266
209
"""Check if the branch has any local modifications.
270
output = self.run( [ 'brz', 'status', '--no-pending' ] )
213
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
273
216
files = self.parse_file_blocks( output )
278
221
"""Return a list of files that have conflicts.
282
output = self.run( [ 'brz', 'status', '--no-pending' ] )
225
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
285
228
files = self.parse_file_blocks( output )
286
229
return files['conflicts'] if 'conflicts' in files else None
289
def add( self, files ):
290
"""Make sure files are added to version control.
291
@param files a list of relative filenames
295
self.run( [ 'brz', 'add', '-N' ] + files )
299
"""Commit changes to the repo.
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 );
314
232
def run( self, cmd ):
315
if the.verbose >= 2: print('exec: %s' % ' '.join( cmd ))
233
if the.verbose > 1: print 'exec: %s' % ' '.join( cmd )
316
234
p = Popen( cmd, cwd = self.dir,
317
235
stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
318
output = p.communicate()[ 0 ].decode()
236
output = p.communicate()[ 0 ]
319
237
if p.returncode > 0:
320
238
raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
322
verbose_output = output.rstrip()
323
if len( verbose_output ):
324
print(re.sub( '(^|\n)', '\\1 : ', verbose_output ))
328
242
def parse_file_blocks( self, output ):
331
buf = io.StringIO( output )
245
buf = StringIO.StringIO( output )
333
247
matches = re.search( '^([a-z ]+):$', line, re.I )
341
255
res[ current ] = list()
342
256
res[ current ].append( matches.group( 1 ) )
344
if re.search( '^[0-9]+ shel(?:f|ves) exists?', line ): continue
258
if re.search( '^[0-9]+ shelf exists', line ): continue
345
259
if re.search( '^working tree is out of date', line ): continue
346
260
raise self.ParseError( "unrecognised line: %s" % line )