/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: 2014-04-05 22:30:23 UTC
  • Revision ID: tim@ed.am-20140405223023-3co87mbvnxwk1l65
added revert command

Show diffs side-by-side

added added

removed removed

Lines of Context:
34
34
 
35
35
                @param dir the fully-qualified directory to work in.
36
36
                """
37
 
 
38
37
                self.dir = dir
39
 
                self.ignored_files = [ '.bzr', '.bzrignore' ]
40
 
 
41
 
 
42
 
        def has_authority( self ):
43
 
                """Check that the directory is under this VCS's control.
44
 
                """
45
 
 
46
 
                return os.path.exists( os.path.join( self.dir, '.bzr' ) )
47
 
 
48
 
 
49
 
        def expand_repo_url( self, url ):
50
 
                """Convert a simple hostname in to an URL that the VCS can use.
51
 
                """
52
 
 
53
 
                return 'bzr+ssh://%s/%s/%s' % ( url, the.dir, the.repo.name )
54
38
 
55
39
 
56
40
        def init( self ):
101
85
                """Obtain some sort of revision identifier
102
86
                """
103
87
 
104
 
                # bzr revno
 
88
                # bzr revert
105
89
                output = self.run( [ 'bzr', 'revno', '--tree' ] )
106
90
 
107
91
                # parse revno
124
108
                # symlink to a non-existant file)
125
109
                if 'kind changed' in files:
126
110
                        for file in files[ 'kind changed' ]:
127
 
                                matches = re.search( '^(.+?)[/@+]? \([^)]+\)$', file )
 
111
                                matches = re.match( r'(.+?)[/@+]? \([^)]+\)', file )
128
112
                                if not matches:
129
113
                                        raise RunTimeError(
130
114
                                                'failed to parse bzr kind change: %s' % file )
131
115
                                file = matches.group( 1 )
132
 
                                if the.verbose >= 2: print "removing (kind changed): " + file
 
116
                                if the.verbose > 1: print "removing (kind changed): " + file
133
117
                                full_file = os.path.join( self.dir, file )
134
118
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
135
119
                                        os.unlink( full_file )
148
132
                # remove unknown files
149
133
                if 'unknown' in files:
150
134
                        for file in files[ 'unknown' ]:
151
 
                                matches = re.search( r'^(.+?)[/@+]?$', file )
152
 
                                if not matches:
153
 
                                        raise RunTimeError(
154
 
                                                'failed to parse bzr unknowns: %s' % file )
155
 
                                file = matches.group( 1 )
156
 
                                if the.verbose >= 2: print "removing (unknown): " + file
 
135
                                if the.verbose > 1: print "removing (unknown): " + file
157
136
                                full_file = os.path.join( self.dir, file )
158
 
                                if os.path.isfile( full_file ) or os.path.islink( full_file ):
 
137
                                if os.path.isfile( full_file ):
159
138
                                        os.unlink( full_file )
160
139
                                elif os.path.isdir( full_file ):
161
140
                                        shutil.rmtree( full_file )
162
141
                                else:
163
142
                                        raise RuntimeError( 'exotic file in repo: %s' % file )
164
143
 
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:
 
144
                # if a revision identifyer has been given, update to that
 
145
                if revno is not None:
167
146
 
168
147
                        # bzr update
169
148
                        self.run( [ 'bzr', 'update', '-r', revno ] )
223
202
                for line in buf:
224
203
                        if not re.search( '^[-R+ ?][K NMD!][* ] ', line ): continue
225
204
                        line = line.rstrip()
 
205
                        if the.verbose > 1: print '  %s' % line
226
206
 
227
207
                        # renames show before and after file names
228
208
                        matches = re.search( '^R.. (.*?)[/@+]? => (.*?)[/@+]?$', line )
273
253
                return files['conflicts'] if 'conflicts' in files else None
274
254
 
275
255
 
276
 
        def add( self, files ):
277
 
                """Make sure files are added to version control.
278
 
                @param files a list of relative filenames
279
 
                """
280
 
 
281
 
                # bzr add
282
 
                self.run( [ 'bzr', 'add', '-N' ] + files )
283
 
 
284
 
 
285
 
        def commit( self ):
286
 
                """Commit changes to the repo.
287
 
                """
288
 
 
289
 
                # bzr commit
290
 
                try:
291
 
                        self.run( [ 'bzr', 'commit', '-m', '' ] )
292
 
                except self.VcsError as e:
293
 
                        if re.search( 'Working tree is out of date', e.output ):
294
 
                                raise the.program.FatalError(
295
 
                                        'you must update your files first.\n' +
296
 
                                        'Hint: see "%s update --help"' % the.program.name );
297
 
                        else:
298
 
                                raise e
299
 
 
300
 
 
301
256
        def run( self, cmd ):
302
 
                if the.verbose >= 2: print 'exec: %s' % ' '.join( cmd )
 
257
                if the.verbose > 1: print 'exec: %s' % ' '.join( cmd )
303
258
                p = Popen( cmd, cwd = self.dir,
304
259
                                   stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
305
260
                output = p.communicate()[ 0 ]
306
261
                if p.returncode > 0:
307
262
                        raise self.VcsError( ' '.join( cmd[ : 2 ] ), output )
308
 
                if the.verbose >= 2:
309
 
                        verbose_output = output.rstrip()
310
 
                        if len( verbose_output ):
311
 
                                print re.sub( '(^|\n)', '\\1  : ', verbose_output )
312
263
                return output
313
264
 
314
265