19
19
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
import subprocess, os, re, shutil
23
27
from subprocess import Popen
26
29
from stdhome import the
29
34
class BzrVcs( Vcs ):
35
40
@param dir the fully-qualified directory to work in.
44
self.ignored_files = [ '.bzr', '.bzrignore' ]
47
def has_authority( self ):
48
"""Check that the directory is under this VCS's control.
51
return os.path.exists( os.path.join( self.dir, '.bzr' ) )
54
def expand_repo_url( self, url ):
55
"""Convert a simple hostname in to an URL that the VCS can use.
58
return 'bzr+ssh://%s/%s/%s' % ( url, the.dir, the.repo.name )
85
106
"""Obtain some sort of revision identifier
89
110
output = self.run( [ 'bzr', 'revno', '--tree' ] )
92
buf = StringIO.StringIO( output )
113
buf = io.StringIO( output )
93
114
return buf.readline().rstrip()
108
129
# symlink to a non-existant file)
109
130
if 'kind changed' in files:
110
131
for file in files[ 'kind changed' ]:
111
matches = re.match( r'(.+?)[/@+]? \([^)]+\)', file )
132
matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
113
134
raise RunTimeError(
114
135
'failed to parse bzr kind change: %s' % file )
115
136
file = matches.group( 1 )
116
if the.verbose >= 2: print "removing (kind changed): " + file
137
if the.verbose >= 2: print("removing (kind changed): " + file)
117
138
full_file = os.path.join( self.dir, file )
118
139
if os.path.isfile( full_file ) or os.path.islink( full_file ):
119
140
os.unlink( full_file )
132
153
# remove unknown files
133
154
if 'unknown' in files:
134
155
for file in files[ 'unknown' ]:
135
if the.verbose >= 2: print "removing (unknown): " + file
156
matches = re.search( r'^(.+?)[/@+]?$', file )
159
'failed to parse bzr unknowns: %s' % file )
160
file = matches.group( 1 )
161
if the.verbose >= 2: print("removing (unknown): " + file)
136
162
full_file = os.path.join( self.dir, file )
137
if os.path.isfile( full_file ):
163
if os.path.isfile( full_file ) or os.path.islink( full_file ):
138
164
os.unlink( full_file )
139
165
elif os.path.isdir( full_file ):
140
166
shutil.rmtree( full_file )
142
168
raise RuntimeError( 'exotic file in repo: %s' % file )
144
# if a revision identifyer has been given, update to that
145
if revno is not None:
170
# if a revision identifier has been given, ensure we're updated to that
171
if revno is not None and self.get_revno() != revno:
148
174
self.run( [ 'bzr', 'update', '-r', revno ] )
199
225
# parse output (see logic in report() in bzrlib/delta.py)
201
buf = StringIO.StringIO( output )
227
buf = io.StringIO( output )
203
229
if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
204
230
line = line.rstrip()
205
if the.verbose >= 2: print ' %s' % line
207
232
# renames show before and after file names
208
233
matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', 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 )
232
270
def has_changes( self ):
233
271
"""Check if the branch has any local modifications.
270
self.run( [ 'bzr', 'commit', '-m', '' ] )
309
self.run( [ 'bzr', '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 );
273
319
def run( self, cmd ):
274
if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
320
if the.verbose >= 2: print('exec: %s' % ' '.join( cmd ))
275
321
p = Popen( cmd, cwd = self.dir,
276
322
stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
277
output = p.communicate()[ 0 ]
323
output = p.communicate()[ 0 ].decode()
278
324
if p.returncode > 0:
279
325
raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
327
verbose_output = output.rstrip()
328
if len( verbose_output ):
329
print(re.sub( '(^|\n)', '\\1 : ', verbose_output ))
283
333
def parse_file_blocks( self, output ):
286
buf = StringIO.StringIO( output )
336
buf = io.StringIO( output )
288
338
matches = re.search( '^([a-z ]+):$', line, re.I )