/stdhome

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

  • Committer: Tim Marston
  • Date: 2016-05-22 16:40:51 UTC
  • Revision ID: tim@ed.am-20160522164051-h2n54oaxg25c7x7c
updated todo

Show diffs side-by-side

added added

removed removed

lib/stdhome/vcs/bzr.py
26
26
from stdhome import the
27
27
 
28
28
 
29
 
class VcsBzr( Vcs ):
 
29
class BzrVcs( Vcs ):
30
30
 
31
31
 
32
32
        def __init__( self, dir ):
34
34
 
35
35
                @param dir the fully-qualified directory to work in.
36
36
                """
 
37
 
37
38
                self.dir = dir
38
39
 
39
40
 
 
41
        def has_authority( self ):
 
42
                """Check that the directory is under this VCS's control.
 
43
                """
 
44
 
 
45
                return os.path.exists( os.path.join( self.dir, '.bzr' ) )
 
46
 
 
47
 
 
48
        def expand_repo_url( self, url ):
 
49
                """Convert a simple hostname in to an URL that the VCS can use.
 
50
                """
 
51
 
 
52
                return 'bzr+ssh://%s/%s/%s' % ( url, the.dir, the.repo.name )
 
53
 
 
54
 
40
55
        def init( self ):
41
56
                """Create a new, empty branch
42
57
                """
45
60
                os.mkdir( self.dir )
46
61
 
47
62
                # bzr init
48
 
                p = Popen( [ 'bzr', 'init', '.' ], cwd = self.dir,
49
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
50
 
                output = p.communicate()[ 0 ]
51
 
                if p.returncode > 0:
 
63
                try:
 
64
                        self.run( [ 'bzr', 'init', '.' ] )
 
65
                except self.VcsError as e:
52
66
 
53
67
                        # attempt to clean-up dir
54
68
                        try:
56
70
                        except OSError:
57
71
                                pass
58
72
 
59
 
                        raise self.VcsError( 'bzr init failed', output )
 
73
                        raise
60
74
 
61
75
 
62
76
        def checkout( self, url ):
69
83
                os.mkdir( self.dir )
70
84
 
71
85
                # bzr co
72
 
                p = Popen( [ 'bzr', 'co', url, '.' ], cwd = self.dir,
73
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
74
 
                output = p.communicate()[ 0 ]
75
 
                if p.returncode > 0:
 
86
                try:
 
87
                        self.run( [ 'bzr', 'checkout', url, '.' ] )
 
88
                except self.VcsError as e:
76
89
 
77
90
                        # attempt to clean-up dir
78
91
                        try:
80
93
                        except OSError:
81
94
                                pass
82
95
 
83
 
                        raise self.VcsError( 'bzr checkout failed', output )
84
 
 
85
 
 
86
 
        def revert( self ):
 
96
                        raise
 
97
 
 
98
 
 
99
        def get_revno( self ):
 
100
                """Obtain some sort of revision identifier
 
101
                """
 
102
 
 
103
                # bzr revno
 
104
                output = self.run( [ 'bzr', 'revno', '--tree' ] )
 
105
 
 
106
                # parse revno
 
107
                buf = StringIO.StringIO( output )
 
108
                return buf.readline().rstrip()
 
109
 
 
110
 
 
111
        def revert( self, revno = None ):
87
112
                """Revert the branch so that there are no outstanding changes or unknown files.
 
113
                If a revno is supplied, then the repository is reverted to that
 
114
                revision.
88
115
                """
89
116
 
 
117
                # bzr st
 
118
                output = self.run( [ 'bzr', 'status' ] )
 
119
                files = self.parse_file_blocks( output )
 
120
 
 
121
                # remove kind changed files (or they can cause `bzr revert` to break in
 
122
                # strange situations, like when a directory has been replaced with a
 
123
                # symlink to a non-existant file)
 
124
                if 'kind changed' in files:
 
125
                        for file in files[ 'kind changed' ]:
 
126
                                matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
 
127
                                if not matches:
 
128
                                        raise RunTimeError(
 
129
                                                'failed to parse bzr kind change: %s' % file )
 
130
                                file = matches.group( 1 )
 
131
                                if the.verbose >= 2: print "removing (kind changed): " + file
 
132
                                full_file = os.path.join( self.dir, file )
 
133
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
 
134
                                        os.unlink( full_file )
 
135
                                elif os.path.isdir( full_file ):
 
136
                                        shutil.rmtree( full_file )
 
137
                                else:
 
138
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
 
139
 
90
140
                # bzr revert
91
 
                p = Popen( [ 'bzr', 'revert', '--no-backup' ], cwd = self.dir,
92
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
93
 
                output = p.communicate()[ 0 ]
94
 
                if p.returncode > 0:
95
 
                        raise self.VcsError( 'bzr revert failed', output )
 
141
                self.run( [ 'bzr', 'revert', '--no-backup' ] )
96
142
 
97
143
                # bzr st
98
 
                p = Popen( [ 'bzr', 'st' ], cwd = self.dir,
99
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
100
 
                output = p.communicate()[ 0 ]
101
 
                if p.returncode > 0:
102
 
                        raise self.VcsError( 'bzr status failed', output )
 
144
                output = self.run( [ 'bzr', 'status' ] )
103
145
                files = self.parse_file_blocks( output )
104
146
 
105
147
                # remove unknown files
106
148
                if 'unknown' in files:
107
149
                        for file in files[ 'unknown' ]:
 
150
                                matches = re.search( r'^(.+?)[/@+]?$', file )
 
151
                                if not matches:
 
152
                                        raise RunTimeError(
 
153
                                                'failed to parse bzr unknowns: %s' % file )
 
154
                                file = matches.group( 1 )
 
155
                                if the.verbose >= 2: print "removing (unknown): " + file
108
156
                                full_file = os.path.join( self.dir, file )
109
 
                                if os.path.isfile( full_file ):
 
157
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
110
158
                                        os.unlink( full_file )
111
 
                                elif os.full_file.isdir( full_file ):
 
159
                                elif os.path.isdir( full_file ):
112
160
                                        shutil.rmtree( full_file )
113
161
                                else:
114
162
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
115
163
 
 
164
                # if a revision identifier has been given, ensure we're updated to that
 
165
                if revno is not None and self.get_revno() != revno:
 
166
 
 
167
                        # bzr update
 
168
                        self.run( [ 'bzr', 'update', '-r', revno ] )
 
169
 
116
170
 
117
171
        def update( self ):
118
172
                """Update the branch, pulling down any upstream changes and merging them.  This
120
174
                operation.
121
175
                """
122
176
 
123
 
                # bzr update
124
 
                p = Popen( [ 'bzr', 'update' ], cwd = self.dir,
125
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
126
 
                output = p.communicate()[ 0 ]
127
 
                if p.returncode > 0:
128
 
                        raise self.VcsError( 'bzr update failed', output )
 
177
#               WARNING: the following might cause bzr to ask for your ssh password more than
 
178
#               once during an update!!!
 
179
#
 
180
#               # get revno
 
181
#               revno = self.get_revno()
 
182
#
 
183
#               # update to current revision (pull in history without updating tree)
 
184
#               self.run( [ 'bzr', 'update', '-r', revno ] )
 
185
#
 
186
#               # get log output
 
187
#               next_revno = str( int( revno ) + 1 )
 
188
#               output = self.run( [ 'bzr', 'log', '-r', next_revno + '..' ] )
 
189
#
 
190
#               # parse output
 
191
#               keep_files = list()
 
192
#               buf = StringIO.StringIO( output )
 
193
#               in_message = False
 
194
#               for line in buf:
 
195
#                       line = line.rstrip( '\n' )
 
196
#                       if line.lower() == 'message:':
 
197
#                               in_message = True
 
198
#                       elif in_message:
 
199
#                               if line[ : 2 ] != '  ':
 
200
#                                       in_message = False
 
201
#                               else:
 
202
#                                       line = line[ 2 : ]
 
203
#
 
204
#                                       # process directives
 
205
#                                       if line[ : 6 ].lower() == 'keep: ':
 
206
#                                               file = line[ 6 : ]
 
207
#                                               if file in rename_files: file = rename_files[ file ]
 
208
#                                               keep_files.append( file )
 
209
#                                       elif line[ : 8 ].lower() == 'rename: ':
 
210
#                                               rename_from = line[ 8 : ]
 
211
#                                       elif line[ : 4 ].lower() == 'to: ':
 
212
#                                               if rename_from in rename_files:
 
213
#                                                       rename_from = rename_files[ rename_from ]
 
214
#                                               rename_files[ line[ 4 : ] ] = rename_from
 
215
 
 
216
                # bzr update properly
 
217
                output = self.run( [ 'bzr', 'update' ] )
129
218
 
130
219
                # parse output (see logic in report() in bzrlib/delta.py)
131
220
                files = list()
133
222
                for line in buf:
134
223
                        if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
135
224
                        line = line.rstrip()
136
 
                        if the.verbose: print '  %s' % line
137
225
 
138
226
                        # renames show before and after file names
139
227
                        matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
155
243
                                continue
156
244
 
157
245
                        raise RuntimeError(
158
 
                                'failed to parse bzr update output line:\n%' % line )
 
246
                                'failed to parse bzr update output line:\n%s' % line )
159
247
 
160
248
                return files
161
249
 
165
253
                """
166
254
 
167
255
                # bzr status
168
 
                p = Popen( [ 'bzr', 'status', '--no-pending' ], cwd = self.dir,
169
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
170
 
                output = p.communicate()[ 0 ]
171
 
                if p.returncode > 0:
172
 
                        raise self.VcsError( 'bzr status failed', output )
 
256
                output = self.run( [ 'bzr', 'status', '--no-pending' ] )
 
257
 
 
258
                # parse output
173
259
                files = self.parse_file_blocks( output )
174
260
                return True if len( files ) else False
175
261
 
179
265
                """
180
266
 
181
267
                # bzr status
182
 
                p = Popen( [ 'bzr', 'status', '--no-pending' ], cwd = self.dir,
183
 
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
184
 
                output = p.communicate()[ 0 ]
185
 
                if p.returncode > 0:
186
 
                        raise self.VcsError( 'bzr status failed', output )
 
268
                output = self.run( [ 'bzr', 'status', '--no-pending' ] )
 
269
 
 
270
                # parse output
187
271
                files = self.parse_file_blocks( output )
188
272
                return files['conflicts'] if 'conflicts' in files else None
189
273
 
190
274
 
 
275
        def add( self, files ):
 
276
                """Make sure files are added to version control.
 
277
                @param files a list of relative filenames
 
278
                """
 
279
 
 
280
                # bzr add
 
281
                self.run( [ 'bzr', 'add', '-N' ] + files )
 
282
 
 
283
 
 
284
        def commit( self ):
 
285
                """Commit changes to the repo.
 
286
                """
 
287
 
 
288
                # bzr commit
 
289
                self.run( [ 'bzr', 'commit', '-m', '' ] )
 
290
 
 
291
 
 
292
        def run( self, cmd ):
 
293
                if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
 
294
                p = Popen( cmd, cwd = self.dir,
 
295
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
 
296
                output = p.communicate()[ 0 ]
 
297
                if p.returncode > 0:
 
298
                        raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
 
299
                if the.verbose >= 2:
 
300
                        verbose_output = output.rstrip()
 
301
                        if len( verbose_output ):
 
302
                                print re.sub( '(^|\n)', '\\1  : ', verbose_output )
 
303
                return output
 
304
 
 
305
 
191
306
        def parse_file_blocks( self, output ):
192
307
                res = dict()
193
308
                current = None
204
319
                                                res[ current ] = list()
205
320
                                        res[ current ].append( matches.group( 1 ) )
206
321
                                        continue
207
 
                        if re.search( '^[0-9]+ shelf exists', line ): continue
 
322
                        if re.search( '^[0-9]+ shel(?:f|ves) exists?', line ): continue
208
323
                        if re.search( '^working tree is out of date', line ): continue
209
324
                        raise self.ParseError( "unrecognised line: %s" % line )
210
325
                return res