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
- browse files at revision 21
- compare with another revision
- download diff
- download tarball
- view history from revision 21
- Committer: Tim Marston
- Date: 2014-04-05 19:20:06 UTC
- Revision ID: tim@ed.am-20140405192006-fv0toip4b8m5pc88
fixed bug preventing skipping of subdirectories in walker; also added
walker.File.__str__()
walker.File.__str__()
- files removed:
- dev
- files renamed:
- VERSION => version
- files modified:
- .bzrignore
added
removed
lib/stdhome/vcs/bzr.py
19
19
# along with this program. If not, see <http://www.gnu.org/licenses/>.
20
20
21
21
22
import io
23
import os
24
import re
25
import shutil
26
import subprocess
22
import subprocess, os, re, shutil
27
23
from subprocess import Popen
28
24
import StringIO
25
from vcs import Vcs
29
26
from stdhome import the
30
27
31
from .vcs import Vcs
32
33
28
34
29
class BzrVcs( Vcs ):
35
30
39
34
40
35
@param dir the fully-qualified directory to work in.
41
36
"""
42
43
37
self.dir = dir
44
self.ignored_files = [ '.bzr', '.bzrignore' ]
45
46
47
def has_authority( self ):
48
"""Check that the directory is under this VCS's control.
49
"""
50
51
return os.path.exists( os.path.join( self.dir, '.bzr' ) )
52
53
54
def expand_repo_url( self, url ):
55
"""Convert a simple hostname in to an URL that the VCS can use.
56
"""
57
58
return 'bzr+ssh://%s/%s/%s' % ( url, the.dir, the.repo.name )
59
38
60
39
61
40
def init( self ):
106
85
"""Obtain some sort of revision identifier
107
86
"""
108
87
109
# bzr revno
88
# bzr revert
110
89
output = self.run( [ 'bzr', 'revno', '--tree' ] )
111
90
112
91
# parse revno
113
buf = io.StringIO( output )
92
buf = StringIO.StringIO( output )
114
93
return buf.readline().rstrip()
115
94
116
95
120
99
revision.
121
100
"""
122
101
123
# bzr st
124
output = self.run( [ 'bzr', 'status' ] )
125
files = self.parse_file_blocks( output )
126
127
# remove kind changed files (or they can cause `bzr revert` to break in
128
# strange situations, like when a directory has been replaced with a
129
# symlink to a non-existant file)
130
if 'kind changed' in files:
131
for file in files[ 'kind changed' ]:
132
matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
133
if not matches:
134
raise RunTimeError(
135
'failed to parse bzr kind change: %s' % file )
136
file = matches.group( 1 )
137
if the.verbose >= 2: print("removing (kind changed): " + file)
138
full_file = os.path.join( self.dir, file )
139
if os.path.isfile( full_file ) or os.path.islink( full_file ):
140
os.unlink( full_file )
141
elif os.path.isdir( full_file ):
142
shutil.rmtree( full_file )
143
else:
144
raise RuntimeError( 'exotic file in repo: %s' % file )
145
146
102
# bzr revert
147
103
self.run( [ 'bzr', 'revert', '--no-backup' ] )
148
104
153
109
# remove unknown files
154
110
if 'unknown' in files:
155
111
for file in files[ 'unknown' ]:
156
matches = re.search( r'^(.+?)[/@+]?$', file )
157
if not matches:
158
raise RunTimeError(
159
'failed to parse bzr unknowns: %s' % file )
160
file = matches.group( 1 )
161
if the.verbose >= 2: print("removing (unknown): " + file)
162
112
full_file = os.path.join( self.dir, file )
163
if os.path.isfile( full_file ) or os.path.islink( full_file ):
113
if os.path.isfile( full_file ):
164
114
os.unlink( full_file )
165
elif os.path.isdir( full_file ):
115
elif os.full_file.isdir( full_file ):
166
116
shutil.rmtree( full_file )
167
117
else:
168
118
raise RuntimeError( 'exotic file in repo: %s' % file )
169
119
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:
120
# if a revision identifyer has been given, update to that
121
if revno is not None:
172
122
173
123
# bzr update
174
124
self.run( [ 'bzr', 'update', '-r', revno ] )
224
174
225
175
# parse output (see logic in report() in bzrlib/delta.py)
226
176
files = list()
227
buf = io.StringIO( output )
177
buf = StringIO.StringIO( output )
228
178
for line in buf:
229
179
if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
230
180
line = line.rstrip()
181
if the.verbose > 1: print ' %s' % line
231
182
232
183
# renames show before and after file names
233
184
matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
249
200
continue
250
201
251
202
raise RuntimeError(
252
'failed to parse bzr update output line:\n%s' % line )
203
'failed to parse bzr update output line:\n%' % line )
253
204
254
205
return files
255
206
256
207
257
def status( self ):
258
"""Get a list of any local modifications. This method returns a list of files
259
which are modified.
260
261
"""
262
263
# bzr status
264
output = self.run( [ 'bzr', 'status', '--no-pending' ] )
265
266
# parse output
267
return self.parse_file_blocks( output )
268
269
270
208
def has_changes( self ):
271
209
"""Check if the branch has any local modifications.
272
210
"""
291
229
return files['conflicts'] if 'conflicts' in files else None
292
230
293
231
294
def add( self, files ):
295
"""Make sure files are added to version control.
296
@param files a list of relative filenames
297
"""
298
299
# bzr add
300
self.run( [ 'bzr', 'add', '-N' ] + files )
301
302
303
def commit( self ):
304
"""Commit changes to the repo.
305
"""
306
307
# bzr commit
308
try:
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 );
315
else:
316
raise e
317
318
319
232
def run( self, cmd ):
320
if the.verbose >= 2: print('exec: %s' % ' '.join( cmd ))
233
if the.verbose > 1: print 'exec: %s' % ' '.join( cmd )
321
234
p = Popen( cmd, cwd = self.dir,
322
235
stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
323
output = p.communicate()[ 0 ].decode()
236
output = p.communicate()[ 0 ]
324
237
if p.returncode > 0:
325
238
raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
326
if the.verbose >= 2:
327
verbose_output = output.rstrip()
328
if len( verbose_output ):
329
print(re.sub( '(^|\n)', '\\1 : ', verbose_output ))
330
239
return output
331
240
332
241
333
242
def parse_file_blocks( self, output ):
334
243
res = dict()
335
244
current = None
336
buf = io.StringIO( output )
245
buf = StringIO.StringIO( output )
337
246
for line in buf:
338
247
matches = re.search( '^([a-z ]+):$', line, re.I )
339
248
if matches:
346
255
res[ current ] = list()
347
256
res[ current ].append( matches.group( 1 ) )
348
257
continue
349
if re.search( '^[0-9]+ shel(?:f|ves) exists?', line ): continue
258
if re.search( '^[0-9]+ shelf exists', line ): continue
350
259
if re.search( '^working tree is out of date', line ): continue
351
260
raise self.ParseError( "unrecognised line: %s" % line )
352
261
return res
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0