/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
329 by Martin Pool
- refactor command functions into command classes
1
# Copyright (C) 2004, 2005 by Canonical Ltd
1 by mbp at sourcefrog
import from baz patch-364
2
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
17
18
338 by Martin Pool
- cleanup of some imports
19
import sys, os, time, os.path
1 by mbp at sourcefrog
import from baz patch-364
20
21
import bzrlib
22
from bzrlib.trace import mutter, note, log_error
329 by Martin Pool
- refactor command functions into command classes
23
from bzrlib.errors import bailout, BzrError, BzrCheckError, BzrCommandError
1 by mbp at sourcefrog
import from baz patch-364
24
from bzrlib.osutils import quotefn, pumpfile, isdir, isfile
453 by Martin Pool
- Split WorkingTree into its own file
25
from bzrlib.tree import RevisionTree, EmptyTree, Tree
1 by mbp at sourcefrog
import from baz patch-364
26
from bzrlib.revision import Revision
27
from bzrlib import Branch, Inventory, InventoryEntry, ScratchBranch, BZRDIR, \
28
     format_date
493 by Martin Pool
- Merge aaron's merge command
29
from bzrlib import merge
1 by mbp at sourcefrog
import from baz patch-364
30
31
350 by Martin Pool
- refactor command aliases into command classes
32
def _squish_command_name(cmd):
33
    return 'cmd_' + cmd.replace('-', '_')
34
35
36
def _unsquish_command_name(cmd):
37
    assert cmd.startswith("cmd_")
38
    return cmd[4:].replace('_','-')
39
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
40
def _parse_revision_str(revstr):
41
    """This handles a revision string -> revno. 
42
43
    There are several possibilities:
44
45
        '234'       -> 234
46
        '234:345'   -> [234, 345]
47
        ':234'      -> [None, 234]
48
        '234:'      -> [234, None]
49
50
    In the future we will also support:
51
        'uuid:blah-blah-blah'   -> ?
52
        'hash:blahblahblah'     -> ?
53
        potentially:
54
        'tag:mytag'             -> ?
55
    """
56
    if revstr.find(':') != -1:
57
        revs = revstr.split(':')
58
        if len(revs) > 2:
59
            raise ValueError('More than 2 pieces not supported for --revision: %r' % revstr)
60
61
        if not revs[0]:
62
            revs[0] = None
63
        else:
64
            revs[0] = int(revs[0])
65
66
        if not revs[1]:
67
            revs[1] = None
68
        else:
69
            revs[1] = int(revs[1])
70
    else:
71
        revs = int(revstr)
72
    return revs
73
351 by Martin Pool
- Split out help functions into bzrlib.help
74
def get_all_cmds():
350 by Martin Pool
- refactor command aliases into command classes
75
    """Return canonical name and class for all registered commands."""
76
    for k, v in globals().iteritems():
77
        if k.startswith("cmd_"):
78
            yield _unsquish_command_name(k), v
79
351 by Martin Pool
- Split out help functions into bzrlib.help
80
def get_cmd_class(cmd):
350 by Martin Pool
- refactor command aliases into command classes
81
    """Return the canonical name and command class for a command.
82
    """
83
    cmd = str(cmd)                      # not unicode
84
85
    # first look up this command under the specified name
272 by Martin Pool
- Add command aliases
86
    try:
350 by Martin Pool
- refactor command aliases into command classes
87
        return cmd, globals()[_squish_command_name(cmd)]
272 by Martin Pool
- Add command aliases
88
    except KeyError:
350 by Martin Pool
- refactor command aliases into command classes
89
        pass
90
91
    # look for any command which claims this as an alias
351 by Martin Pool
- Split out help functions into bzrlib.help
92
    for cmdname, cmdclass in get_all_cmds():
350 by Martin Pool
- refactor command aliases into command classes
93
        if cmd in cmdclass.aliases:
94
            return cmdname, cmdclass
422 by Martin Pool
- External-command patch from mpe
95
96
    cmdclass = ExternalCommand.find_command(cmd)
97
    if cmdclass:
98
        return cmd, cmdclass
99
100
    raise BzrCommandError("unknown command %r" % cmd)
272 by Martin Pool
- Add command aliases
101
329 by Martin Pool
- refactor command functions into command classes
102
558 by Martin Pool
- All top-level classes inherit from object
103
class Command(object):
329 by Martin Pool
- refactor command functions into command classes
104
    """Base class for commands.
105
106
    The docstring for an actual command should give a single-line
107
    summary, then a complete description of the command.  A grammar
108
    description will be inserted.
109
110
    takes_args
111
        List of argument forms, marked with whether they are optional,
112
        repeated, etc.
113
114
    takes_options
115
        List of options that may be given for this command.
116
117
    hidden
118
        If true, this command isn't advertised.
119
    """
120
    aliases = []
121
    
122
    takes_args = []
123
    takes_options = []
124
125
    hidden = False
126
    
127
    def __init__(self, options, arguments):
128
        """Construct and run the command.
129
130
        Sets self.status to the return value of run()."""
131
        assert isinstance(options, dict)
132
        assert isinstance(arguments, dict)
133
        cmdargs = options.copy()
134
        cmdargs.update(arguments)
135
        assert self.__doc__ != Command.__doc__, \
136
               ("No help message set for %r" % self)
137
        self.status = self.run(**cmdargs)
138
139
    
140
    def run(self):
141
        """Override this in sub-classes.
142
143
        This is invoked with the options and arguments bound to
144
        keyword parameters.
145
337 by Martin Pool
- Clarify return codes from command objects
146
        Return 0 or None if the command was successful, or a shell
147
        error code if not.
329 by Martin Pool
- refactor command functions into command classes
148
        """
337 by Martin Pool
- Clarify return codes from command objects
149
        return 0
329 by Martin Pool
- refactor command functions into command classes
150
151
422 by Martin Pool
- External-command patch from mpe
152
class ExternalCommand(Command):
153
    """Class to wrap external commands.
154
155
    We cheat a little here, when get_cmd_class() calls us we actually give it back
156
    an object we construct that has the appropriate path, help, options etc for the
157
    specified command.
158
159
    When run_bzr() tries to instantiate that 'class' it gets caught by the __call__
160
    method, which we override to call the Command.__init__ method. That then calls
161
    our run method which is pretty straight forward.
162
163
    The only wrinkle is that we have to map bzr's dictionary of options and arguments
164
    back into command line options and arguments for the script.
165
    """
166
167
    def find_command(cls, cmd):
168
        bzrpath = os.environ.get('BZRPATH', '')
169
170
        for dir in bzrpath.split(':'):
171
            path = os.path.join(dir, cmd)
172
            if os.path.isfile(path):
173
                return ExternalCommand(path)
174
175
        return None
176
177
    find_command = classmethod(find_command)
178
179
    def __init__(self, path):
180
        self.path = path
181
424 by Martin Pool
todo
182
        # TODO: If either of these fail, we should detect that and
183
        # assume that path is not really a bzr plugin after all.
184
422 by Martin Pool
- External-command patch from mpe
185
        pipe = os.popen('%s --bzr-usage' % path, 'r')
186
        self.takes_options = pipe.readline().split()
187
        self.takes_args = pipe.readline().split()
188
        pipe.close()
189
190
        pipe = os.popen('%s --bzr-help' % path, 'r')
191
        self.__doc__ = pipe.read()
192
        pipe.close()
193
194
    def __call__(self, options, arguments):
195
        Command.__init__(self, options, arguments)
196
        return self
197
198
    def run(self, **kargs):
199
        opts = []
200
        args = []
201
202
        keys = kargs.keys()
203
        keys.sort()
204
        for name in keys:
205
            value = kargs[name]
206
            if OPTIONS.has_key(name):
207
                # it's an option
208
                opts.append('--%s' % name)
209
                if value is not None and value is not True:
210
                    opts.append(str(value))
211
            else:
212
                # it's an arg, or arg list
213
                if type(value) is not list:
214
                    value = [value]
215
                for v in value:
216
                    if v is not None:
217
                        args.append(str(v))
218
219
        self.status = os.spawnv(os.P_WAIT, self.path, [self.path] + opts + args)
220
        return self.status
221
329 by Martin Pool
- refactor command functions into command classes
222
223
class cmd_status(Command):
1 by mbp at sourcefrog
import from baz patch-364
224
    """Display status summary.
225
466 by Martin Pool
- doc for status command
226
    This reports on versioned and unknown files, reporting them
227
    grouped by state.  Possible states are:
228
229
    added
230
        Versioned in the working copy but not in the previous revision.
231
232
    removed
467 by Martin Pool
- doc for status command
233
        Versioned in the previous revision but removed or deleted
466 by Martin Pool
- doc for status command
234
        in the working copy.
235
236
    renamed
237
        Path of this file changed from the previous revision;
238
        the text may also have changed.  This includes files whose
467 by Martin Pool
- doc for status command
239
        parent directory was renamed.
466 by Martin Pool
- doc for status command
240
241
    modified
242
        Text has changed since the previous revision.
243
244
    unchanged
467 by Martin Pool
- doc for status command
245
        Nothing about this file has changed since the previous revision.
246
        Only shown with --all.
466 by Martin Pool
- doc for status command
247
248
    unknown
249
        Not versioned and not matching an ignore pattern.
250
251
    To see ignored files use 'bzr ignored'.  For details in the
252
    changes to file texts, use 'bzr diff'.
468 by Martin Pool
- Interpret arguments to bzr status
253
254
    If no arguments are specified, the status of the entire working
255
    directory is shown.  Otherwise, only the status of the specified
256
    files or directories is reported.  If a directory is given, status
257
    is reported for everything inside that directory.
1 by mbp at sourcefrog
import from baz patch-364
258
    """
404 by Martin Pool
- bzr status now optionally takes filenames to check
259
    takes_args = ['file*']
465 by Martin Pool
- Move show_status() out of Branch into a new function in
260
    takes_options = ['all', 'show-ids']
350 by Martin Pool
- refactor command aliases into command classes
261
    aliases = ['st', 'stat']
329 by Martin Pool
- refactor command functions into command classes
262
    
465 by Martin Pool
- Move show_status() out of Branch into a new function in
263
    def run(self, all=False, show_ids=False, file_list=None):
468 by Martin Pool
- Interpret arguments to bzr status
264
        if file_list:
265
            b = Branch(file_list[0], lock_mode='r')
266
            file_list = [b.relpath(x) for x in file_list]
267
            # special case: only one path was given and it's the root
268
            # of the branch
269
            if file_list == ['']:
270
                file_list = None
271
        else:
272
            b = Branch('.', lock_mode='r')
465 by Martin Pool
- Move show_status() out of Branch into a new function in
273
        import status
274
        status.show_status(b, show_unchanged=all, show_ids=show_ids,
483 by Martin Pool
- change 'file_list' to more explanatory 'specific_files'
275
                           specific_files=file_list)
329 by Martin Pool
- refactor command functions into command classes
276
277
278
class cmd_cat_revision(Command):
279
    """Write out metadata for a revision."""
280
281
    hidden = True
282
    takes_args = ['revision_id']
283
    
284
    def run(self, revision_id):
285
        Branch('.').get_revision(revision_id).write_xml(sys.stdout)
286
287
288
class cmd_revno(Command):
289
    """Show current revision number.
290
291
    This is equal to the number of revisions on this branch."""
292
    def run(self):
293
        print Branch('.').revno()
294
295
    
296
class cmd_add(Command):
70 by mbp at sourcefrog
Prepare for smart recursive add.
297
    """Add specified files or directories.
298
299
    In non-recursive mode, all the named items are added, regardless
300
    of whether they were previously ignored.  A warning is given if
301
    any of the named files are already versioned.
302
303
    In recursive mode (the default), files are treated the same way
304
    but the behaviour for directories is different.  Directories that
305
    are already versioned do not give a warning.  All directories,
306
    whether already versioned or not, are searched for files or
307
    subdirectories that are neither versioned or ignored, and these
308
    are added.  This search proceeds recursively into versioned
309
    directories.
310
311
    Therefore simply saying 'bzr add .' will version all files that
312
    are currently unknown.
279 by Martin Pool
todo
313
314
    TODO: Perhaps adding a file whose directly is not versioned should
315
    recursively add that parent, rather than giving an error?
70 by mbp at sourcefrog
Prepare for smart recursive add.
316
    """
329 by Martin Pool
- refactor command functions into command classes
317
    takes_args = ['file+']
318
    takes_options = ['verbose']
319
    
320
    def run(self, file_list, verbose=False):
321
        bzrlib.add.smart_add(file_list, verbose)
322
323
386 by Martin Pool
- Typo (reported by uws)
324
class cmd_relpath(Command):
329 by Martin Pool
- refactor command functions into command classes
325
    """Show path of a file relative to root"""
392 by Martin Pool
- fix relpath and add tests
326
    takes_args = ['filename']
329 by Martin Pool
- refactor command functions into command classes
327
    
392 by Martin Pool
- fix relpath and add tests
328
    def run(self, filename):
329
        print Branch(filename).relpath(filename)
329 by Martin Pool
- refactor command functions into command classes
330
331
332
333
class cmd_inventory(Command):
334
    """Show inventory of the current working copy or a revision."""
335
    takes_options = ['revision']
336
    
337
    def run(self, revision=None):
338
        b = Branch('.')
339
        if revision == None:
340
            inv = b.read_working_inventory()
341
        else:
342
            inv = b.get_revision_inventory(b.lookup_revision(revision))
343
556 by Martin Pool
- fix up Inventory.entries()
344
        for path, entry in inv.entries():
329 by Martin Pool
- refactor command functions into command classes
345
            print '%-50s %s' % (entry.file_id, path)
346
347
348
class cmd_move(Command):
349
    """Move files to a different directory.
350
351
    examples:
352
        bzr move *.txt doc
353
354
    The destination must be a versioned directory in the same branch.
355
    """
356
    takes_args = ['source$', 'dest']
357
    def run(self, source_list, dest):
358
        b = Branch('.')
359
360
        b.move([b.relpath(s) for s in source_list], b.relpath(dest))
361
362
363
class cmd_rename(Command):
168 by mbp at sourcefrog
new "rename" command
364
    """Change the name of an entry.
365
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
366
    examples:
367
      bzr rename frob.c frobber.c
368
      bzr rename src/frob.c lib/frob.c
369
370
    It is an error if the destination name exists.
371
372
    See also the 'move' command, which moves files into a different
373
    directory without changing their name.
374
375
    TODO: Some way to rename multiple files without invoking bzr for each
376
    one?"""
329 by Martin Pool
- refactor command functions into command classes
377
    takes_args = ['from_name', 'to_name']
168 by mbp at sourcefrog
new "rename" command
378
    
329 by Martin Pool
- refactor command functions into command classes
379
    def run(self, from_name, to_name):
380
        b = Branch('.')
381
        b.rename_one(b.relpath(from_name), b.relpath(to_name))
382
383
384
385
class cmd_renames(Command):
164 by mbp at sourcefrog
new 'renames' command
386
    """Show list of renamed files.
387
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
388
    TODO: Option to show renames between two historical versions.
389
390
    TODO: Only show renames under dir, rather than in the whole branch.
391
    """
329 by Martin Pool
- refactor command functions into command classes
392
    takes_args = ['dir?']
393
394
    def run(self, dir='.'):
395
        b = Branch(dir)
396
        old_inv = b.basis_tree().inventory
397
        new_inv = b.read_working_inventory()
398
399
        renames = list(bzrlib.tree.find_renames(old_inv, new_inv))
400
        renames.sort()
401
        for old_name, new_name in renames:
402
            print "%s => %s" % (old_name, new_name)        
403
404
405
class cmd_info(Command):
472 by Martin Pool
- Optional branch parameter to info command
406
    """Show statistical information about a branch."""
407
    takes_args = ['branch?']
408
    
409
    def run(self, branch=None):
329 by Martin Pool
- refactor command functions into command classes
410
        import info
472 by Martin Pool
- Optional branch parameter to info command
411
412
        from branch import find_branch
413
        b = find_branch(branch)
414
        info.show_info(b)
329 by Martin Pool
- refactor command functions into command classes
415
416
417
class cmd_remove(Command):
418
    """Make a file unversioned.
419
420
    This makes bzr stop tracking changes to a versioned file.  It does
421
    not delete the working copy.
422
    """
423
    takes_args = ['file+']
424
    takes_options = ['verbose']
425
    
426
    def run(self, file_list, verbose=False):
427
        b = Branch(file_list[0])
428
        b.remove([b.relpath(f) for f in file_list], verbose=verbose)
429
430
431
class cmd_file_id(Command):
178 by mbp at sourcefrog
- Use a non-null file_id for the branch root directory. At the moment
432
    """Print file_id of a particular file or directory.
433
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
434
    The file_id is assigned when the file is first added and remains the
435
    same through all revisions where the file exists, even when it is
436
    moved or renamed.
437
    """
329 by Martin Pool
- refactor command functions into command classes
438
    hidden = True
439
    takes_args = ['filename']
440
    def run(self, filename):
441
        b = Branch(filename)
442
        i = b.inventory.path2id(b.relpath(filename))
443
        if i == None:
444
            bailout("%r is not a versioned file" % filename)
445
        else:
446
            print i
447
448
449
class cmd_file_path(Command):
178 by mbp at sourcefrog
- Use a non-null file_id for the branch root directory. At the moment
450
    """Print path of file_ids to a file or directory.
451
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
452
    This prints one line for each directory down to the target,
453
    starting at the branch root."""
329 by Martin Pool
- refactor command functions into command classes
454
    hidden = True
455
    takes_args = ['filename']
456
    def run(self, filename):
457
        b = Branch(filename)
458
        inv = b.inventory
459
        fid = inv.path2id(b.relpath(filename))
460
        if fid == None:
461
            bailout("%r is not a versioned file" % filename)
462
        for fip in inv.get_idpath(fid):
463
            print fip
464
465
466
class cmd_revision_history(Command):
467
    """Display list of revision ids on this branch."""
468
    def run(self):
469
        for patchid in Branch('.').revision_history():
470
            print patchid
471
472
473
class cmd_directories(Command):
474
    """Display list of versioned directories in this branch."""
475
    def run(self):
476
        for name, ie in Branch('.').read_working_inventory().directories():
477
            if name == '':
478
                print '.'
479
            else:
480
                print name
481
482
483
class cmd_init(Command):
484
    """Make a directory into a versioned branch.
485
486
    Use this to create an empty branch, or before importing an
487
    existing project.
488
489
    Recipe for importing a tree of files:
490
        cd ~/project
491
        bzr init
492
        bzr add -v .
493
        bzr status
494
        bzr commit -m 'imported project'
495
    """
496
    def run(self):
497
        Branch('.', init=True)
498
499
500
class cmd_diff(Command):
501
    """Show differences in working tree.
502
    
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
503
    If files are listed, only the changes in those files are listed.
504
    Otherwise, all changes for the tree are listed.
505
506
    TODO: Given two revision arguments, show the difference between them.
507
508
    TODO: Allow diff across branches.
509
510
    TODO: Option to use external diff command; could be GNU diff, wdiff,
511
          or a graphical diff.
512
276 by Martin Pool
Doc
513
    TODO: Python difflib is not exactly the same as unidiff; should
514
          either fix it up or prefer to use an external diff.
515
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
516
    TODO: If a directory is given, diff everything under that.
517
276 by Martin Pool
Doc
518
    TODO: Selected-file diff is inefficient and doesn't show you
519
          deleted files.
278 by Martin Pool
- Better workaround for trailing newlines in diffs
520
521
    TODO: This probably handles non-Unix newlines poorly.
329 by Martin Pool
- refactor command functions into command classes
522
    """
523
    
524
    takes_args = ['file*']
525
    takes_options = ['revision']
350 by Martin Pool
- refactor command aliases into command classes
526
    aliases = ['di']
329 by Martin Pool
- refactor command functions into command classes
527
528
    def run(self, revision=None, file_list=None):
529
        from bzrlib.diff import show_diff
547 by Martin Pool
- bzr diff finds a branch from the first parameter,
530
        from bzrlib import find_branch
531
532
        if file_list:
533
            b = find_branch(file_list[0], lock_mode='r')
534
            file_list = [b.relpath(f) for f in file_list]
535
            if file_list == ['']:
536
                # just pointing to top-of-tree
537
                file_list = None
538
        else:
539
            b = Branch('.', lock_mode='r')
329 by Martin Pool
- refactor command functions into command classes
540
    
547 by Martin Pool
- bzr diff finds a branch from the first parameter,
541
        show_diff(b, revision, specific_files=file_list)
329 by Martin Pool
- refactor command functions into command classes
542
543
437 by Martin Pool
- new command 'bzr modified' to exercise the statcache
544
        
545
546
329 by Martin Pool
- refactor command functions into command classes
547
class cmd_deleted(Command):
135 by mbp at sourcefrog
Simple new 'deleted' command
548
    """List files deleted in the working tree.
549
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
550
    TODO: Show files deleted since a previous revision, or between two revisions.
135 by mbp at sourcefrog
Simple new 'deleted' command
551
    """
329 by Martin Pool
- refactor command functions into command classes
552
    def run(self, show_ids=False):
553
        b = Branch('.')
554
        old = b.basis_tree()
555
        new = b.working_tree()
556
557
        ## TODO: Much more efficient way to do this: read in new
558
        ## directories with readdir, rather than stating each one.  Same
559
        ## level of effort but possibly much less IO.  (Or possibly not,
560
        ## if the directories are very large...)
561
562
        for path, ie in old.inventory.iter_entries():
563
            if not new.has_id(ie.file_id):
564
                if show_ids:
565
                    print '%-50s %s' % (path, ie.file_id)
566
                else:
567
                    print path
568
437 by Martin Pool
- new command 'bzr modified' to exercise the statcache
569
570
class cmd_modified(Command):
571
    """List files modified in working tree."""
572
    hidden = True
573
    def run(self):
574
        import statcache
575
        b = Branch('.')
438 by Martin Pool
- Avoid calling Inventory.iter_entries() when finding modified
576
        inv = b.read_working_inventory()
577
        sc = statcache.update_cache(b, inv)
437 by Martin Pool
- new command 'bzr modified' to exercise the statcache
578
        basis = b.basis_tree()
579
        basis_inv = basis.inventory
438 by Martin Pool
- Avoid calling Inventory.iter_entries() when finding modified
580
        
581
        # We used to do this through iter_entries(), but that's slow
582
        # when most of the files are unmodified, as is usually the
583
        # case.  So instead we iterate by inventory entry, and only
584
        # calculate paths as necessary.
585
586
        for file_id in basis_inv:
587
            cacheentry = sc.get(file_id)
588
            if not cacheentry:                 # deleted
589
                continue
590
            ie = basis_inv[file_id]
437 by Martin Pool
- new command 'bzr modified' to exercise the statcache
591
            if cacheentry[statcache.SC_SHA1] != ie.text_sha1:
438 by Martin Pool
- Avoid calling Inventory.iter_entries() when finding modified
592
                path = inv.id2path(file_id)
437 by Martin Pool
- new command 'bzr modified' to exercise the statcache
593
                print path
439 by Martin Pool
- new command 'bzr added'
594
595
596
597
class cmd_added(Command):
598
    """List files added in working tree."""
599
    hidden = True
600
    def run(self):
601
        b = Branch('.')
602
        wt = b.working_tree()
603
        basis_inv = b.basis_tree().inventory
604
        inv = wt.inventory
605
        for file_id in inv:
606
            if file_id in basis_inv:
607
                continue
608
            path = inv.id2path(file_id)
609
            if not os.access(b.abspath(path), os.F_OK):
610
                continue
611
            print path
437 by Martin Pool
- new command 'bzr modified' to exercise the statcache
612
                
613
        
614
329 by Martin Pool
- refactor command functions into command classes
615
class cmd_root(Command):
616
    """Show the tree root directory.
617
618
    The root is the nearest enclosing directory with a .bzr control
619
    directory."""
620
    takes_args = ['filename?']
621
    def run(self, filename=None):
622
        """Print the branch root."""
416 by Martin Pool
- bzr log and bzr root now accept an http URL
623
        from branch import find_branch
624
        b = find_branch(filename)
625
        print getattr(b, 'base', None) or getattr(b, 'baseurl')
329 by Martin Pool
- refactor command functions into command classes
626
627
628
class cmd_log(Command):
1 by mbp at sourcefrog
import from baz patch-364
629
    """Show log of this branch.
630
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
631
    To request a range of logs, you can use the command -r begin:end
632
    -r revision requests a specific revision, -r :end or -r begin: are
633
    also valid.
634
635
    TODO: Make --revision support uuid: and hash: [future tag:] notation.
636
  
545 by Martin Pool
- --forward option for log
637
    """
367 by Martin Pool
- New --show-ids option for bzr log
638
378 by Martin Pool
- New usage bzr log FILENAME
639
    takes_args = ['filename?']
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
640
    takes_options = ['forward', 'timezone', 'verbose', 'show-ids', 'revision']
545 by Martin Pool
- --forward option for log
641
    
642
    def run(self, filename=None, timezone='original',
643
            verbose=False,
644
            show_ids=False,
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
645
            forward=False,
646
            revision=None):
527 by Martin Pool
- refactor log command
647
        from bzrlib import show_log, find_branch
562 by Martin Pool
- bug fix for printing logs containing unicode
648
        import codecs
545 by Martin Pool
- --forward option for log
649
650
        direction = (forward and 'forward') or 'reverse'
527 by Martin Pool
- refactor log command
651
        
378 by Martin Pool
- New usage bzr log FILENAME
652
        if filename:
527 by Martin Pool
- refactor log command
653
            b = find_branch(filename, lock_mode='r')
654
            fp = b.relpath(filename)
533 by Martin Pool
- fix up asking for the log for the root of a remote branch
655
            if fp:
656
                file_id = b.read_working_inventory().path2id(fp)
657
            else:
658
                file_id = None  # points to branch root
527 by Martin Pool
- refactor log command
659
        else:
660
            b = find_branch('.', lock_mode='r')
661
            file_id = None
662
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
663
        if revision == None:
664
            revision = [None, None]
665
        elif isinstance(revision, int):
666
            revision = [revision, revision]
667
        else:
668
            # pair of revisions?
669
            pass
670
            
671
        assert len(revision) == 2
672
562 by Martin Pool
- bug fix for printing logs containing unicode
673
        mutter('encoding log as %r' % bzrlib.user_encoding)
674
        outf = codecs.getwriter(bzrlib.user_encoding)(sys.stdout)
675
527 by Martin Pool
- refactor log command
676
        show_log(b, file_id,
677
                 show_timezone=timezone,
678
                 verbose=verbose,
679
                 show_ids=show_ids,
562 by Martin Pool
- bug fix for printing logs containing unicode
680
                 to_file=outf,
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
681
                 direction=direction,
682
                 start_revision=revision[0],
683
                 end_revision=revision[1])
329 by Martin Pool
- refactor command functions into command classes
684
685
375 by Martin Pool
- New command touching-revisions and function to trace
686
687
class cmd_touching_revisions(Command):
523 by Martin Pool
doc
688
    """Return revision-ids which affected a particular file.
689
690
    A more user-friendly interface is "bzr log FILE"."""
375 by Martin Pool
- New command touching-revisions and function to trace
691
    hidden = True
692
    takes_args = ["filename"]
693
    def run(self, filename):
694
        b = Branch(filename, lock_mode='r')
695
        inv = b.read_working_inventory()
696
        file_id = inv.path2id(b.relpath(filename))
697
        for revno, revision_id, what in bzrlib.log.find_touching_revisions(b, file_id):
698
            print "%6d %s" % (revno, what)
699
700
329 by Martin Pool
- refactor command functions into command classes
701
class cmd_ls(Command):
1 by mbp at sourcefrog
import from baz patch-364
702
    """List files in a tree.
703
254 by Martin Pool
- Doc cleanups from Magnus Therning
704
    TODO: Take a revision or remote path and list that tree instead.
1 by mbp at sourcefrog
import from baz patch-364
705
    """
329 by Martin Pool
- refactor command functions into command classes
706
    hidden = True
707
    def run(self, revision=None, verbose=False):
708
        b = Branch('.')
709
        if revision == None:
710
            tree = b.working_tree()
711
        else:
712
            tree = b.revision_tree(b.lookup_revision(revision))
713
714
        for fp, fc, kind, fid in tree.list_files():
715
            if verbose:
716
                if kind == 'directory':
717
                    kindch = '/'
718
                elif kind == 'file':
719
                    kindch = ''
720
                else:
721
                    kindch = '???'
722
723
                print '%-8s %s%s' % (fc, fp, kindch)
1 by mbp at sourcefrog
import from baz patch-364
724
            else:
329 by Martin Pool
- refactor command functions into command classes
725
                print fp
726
727
728
729
class cmd_unknowns(Command):
1 by mbp at sourcefrog
import from baz patch-364
730
    """List unknown files"""
329 by Martin Pool
- refactor command functions into command classes
731
    def run(self):
732
        for f in Branch('.').unknowns():
733
            print quotefn(f)
734
735
736
737
class cmd_ignore(Command):
420 by Martin Pool
Doc
738
    """Ignore a command or pattern
739
740
    To remove patterns from the ignore list, edit the .bzrignore file.
741
742
    If the pattern contains a slash, it is compared to the whole path
743
    from the branch root.  Otherwise, it is comapred to only the last
744
    component of the path.
745
746
    Ignore patterns are case-insensitive on case-insensitive systems.
747
748
    Note: wildcards must be quoted from the shell on Unix.
749
750
    examples:
751
        bzr ignore ./Makefile
752
        bzr ignore '*.class'
753
    """
329 by Martin Pool
- refactor command functions into command classes
754
    takes_args = ['name_pattern']
310 by Martin Pool
- new 'bzr ignored' command!
755
    
329 by Martin Pool
- refactor command functions into command classes
756
    def run(self, name_pattern):
409 by Martin Pool
- New AtomicFile class
757
        from bzrlib.atomicfile import AtomicFile
758
        import codecs
759
329 by Martin Pool
- refactor command functions into command classes
760
        b = Branch('.')
410 by Martin Pool
- Fix ignore command and add tests
761
        ifn = b.abspath('.bzrignore')
329 by Martin Pool
- refactor command functions into command classes
762
410 by Martin Pool
- Fix ignore command and add tests
763
        if os.path.exists(ifn):
498 by Martin Pool
bugfix for bzr ignore reported by ddaa:
764
            f = open(ifn, 'rt')
765
            try:
766
                igns = f.read().decode('utf-8')
767
            finally:
768
                f.close()
409 by Martin Pool
- New AtomicFile class
769
        else:
770
            igns = ''
771
772
        if igns and igns[-1] != '\n':
773
            igns += '\n'
774
        igns += name_pattern + '\n'
775
498 by Martin Pool
bugfix for bzr ignore reported by ddaa:
776
        try:
777
            f = AtomicFile(ifn, 'wt')
778
            f.write(igns.encode('utf-8'))
779
            f.commit()
780
        finally:
781
            f.close()
329 by Martin Pool
- refactor command functions into command classes
782
783
        inv = b.working_tree().inventory
784
        if inv.path2id('.bzrignore'):
785
            mutter('.bzrignore is already versioned')
786
        else:
787
            mutter('need to make new .bzrignore file versioned')
788
            b.add(['.bzrignore'])
789
790
791
792
class cmd_ignored(Command):
421 by Martin Pool
doc
793
    """List ignored files and the patterns that matched them.
794
795
    See also: bzr ignore"""
329 by Martin Pool
- refactor command functions into command classes
796
    def run(self):
797
        tree = Branch('.').working_tree()
798
        for path, file_class, kind, file_id in tree.list_files():
799
            if file_class != 'I':
800
                continue
801
            ## XXX: Slightly inefficient since this was already calculated
802
            pat = tree.is_ignored(path)
803
            print '%-50s %s' % (path, pat)
804
805
806
class cmd_lookup_revision(Command):
807
    """Lookup the revision-id from a revision-number
808
809
    example:
810
        bzr lookup-revision 33
421 by Martin Pool
doc
811
    """
329 by Martin Pool
- refactor command functions into command classes
812
    hidden = True
338 by Martin Pool
- cleanup of some imports
813
    takes_args = ['revno']
814
    
329 by Martin Pool
- refactor command functions into command classes
815
    def run(self, revno):
816
        try:
817
            revno = int(revno)
818
        except ValueError:
338 by Martin Pool
- cleanup of some imports
819
            raise BzrCommandError("not a valid revision-number: %r" % revno)
820
821
        print Branch('.').lookup_revision(revno)
329 by Martin Pool
- refactor command functions into command classes
822
823
824
class cmd_export(Command):
825
    """Export past revision to destination directory.
826
827
    If no revision is specified this exports the last committed revision."""
828
    takes_args = ['dest']
829
    takes_options = ['revision']
394 by Martin Pool
- Fix argument handling in export command
830
    def run(self, dest, revision=None):
329 by Martin Pool
- refactor command functions into command classes
831
        b = Branch('.')
394 by Martin Pool
- Fix argument handling in export command
832
        if revision == None:
833
            rh = b.revision_history()[-1]
329 by Martin Pool
- refactor command functions into command classes
834
        else:
394 by Martin Pool
- Fix argument handling in export command
835
            rh = b.lookup_revision(int(revision))
329 by Martin Pool
- refactor command functions into command classes
836
        t = b.revision_tree(rh)
837
        t.export(dest)
838
839
840
class cmd_cat(Command):
841
    """Write a file's text from a previous revision."""
842
843
    takes_options = ['revision']
844
    takes_args = ['filename']
845
846
    def run(self, filename, revision=None):
847
        if revision == None:
848
            raise BzrCommandError("bzr cat requires a revision number")
849
        b = Branch('.')
850
        b.print_file(b.relpath(filename), int(revision))
851
852
853
class cmd_local_time_offset(Command):
854
    """Show the offset in seconds from GMT to local time."""
855
    hidden = True    
856
    def run(self):
857
        print bzrlib.osutils.local_time_offset()
858
859
860
861
class cmd_commit(Command):
862
    """Commit changes into a new revision.
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
863
491 by Martin Pool
- Selective commit!
864
    If selected files are specified, only changes to those files are
865
    committed.  If a directory is specified then its contents are also
866
    committed.
867
868
    A selected-file commit may fail in some cases where the committed
869
    tree would be invalid, such as trying to commit a file in a
870
    newly-added directory that is not itself committed.
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
871
872
    TODO: Run hooks on tree to-be-committed, and after commit.
873
874
    TODO: Strict commit that fails if there are unknown or deleted files.
875
    """
491 by Martin Pool
- Selective commit!
876
    takes_args = ['selected*']
389 by Martin Pool
- new commit --file option!
877
    takes_options = ['message', 'file', 'verbose']
350 by Martin Pool
- refactor command aliases into command classes
878
    aliases = ['ci', 'checkin']
879
505 by Martin Pool
- commit is verbose by default
880
    def run(self, message=None, file=None, verbose=True, selected_list=None):
485 by Martin Pool
- move commit code into its own module
881
        from bzrlib.commit import commit
882
389 by Martin Pool
- new commit --file option!
883
        ## Warning: shadows builtin file()
884
        if not message and not file:
885
            raise BzrCommandError("please specify a commit message",
886
                                  ["use either --message or --file"])
887
        elif message and file:
888
            raise BzrCommandError("please specify either --message or --file")
889
        
890
        if file:
891
            import codecs
892
            message = codecs.open(file, 'rt', bzrlib.user_encoding).read()
893
485 by Martin Pool
- move commit code into its own module
894
        b = Branch('.')
491 by Martin Pool
- Selective commit!
895
        commit(b, message, verbose=verbose, specific_files=selected_list)
329 by Martin Pool
- refactor command functions into command classes
896
897
898
class cmd_check(Command):
899
    """Validate consistency of branch history.
232 by mbp at sourcefrog
Allow docstrings for help to be in PEP0257 format.
900
901
    This command checks various invariants about the branch storage to
902
    detect data corruption or bzr bugs.
903
    """
329 by Martin Pool
- refactor command functions into command classes
904
    takes_args = ['dir?']
905
    def run(self, dir='.'):
906
        import bzrlib.check
512 by Martin Pool
- bzr check can be run from a branch subdirectory
907
        bzrlib.check.check(Branch(dir))
329 by Martin Pool
- refactor command functions into command classes
908
909
910
911
class cmd_whoami(Command):
912
    """Show bzr user id."""
913
    takes_options = ['email']
286 by Martin Pool
- New bzr whoami --email option
914
    
329 by Martin Pool
- refactor command functions into command classes
915
    def run(self, email=False):
916
        if email:
917
            print bzrlib.osutils.user_email()
918
        else:
919
            print bzrlib.osutils.username()
920
921
922
class cmd_selftest(Command):
55 by mbp at sourcefrog
bzr selftest shows some counts of tests
923
    """Run internal test suite"""
329 by Martin Pool
- refactor command functions into command classes
924
    hidden = True
925
    def run(self):
926
        failures, tests = 0, 0
927
528 by Martin Pool
- fix bzr selftest trying to import non-existant module
928
        import doctest, bzrlib.store
329 by Martin Pool
- refactor command functions into command classes
929
        bzrlib.trace.verbose = False
930
931
        for m in bzrlib.store, bzrlib.inventory, bzrlib.branch, bzrlib.osutils, \
514 by Martin Pool
- remove old internal tests in favour of blackbox tests
932
            bzrlib.tree, bzrlib.commands, bzrlib.add:
329 by Martin Pool
- refactor command functions into command classes
933
            mf, mt = doctest.testmod(m)
934
            failures += mf
935
            tests += mt
936
            print '%-40s %3d tests' % (m.__name__, mt),
937
            if mf:
938
                print '%3d FAILED!' % mf
939
            else:
940
                print
941
942
        print '%-40s %3d tests' % ('total', tests),
943
        if failures:
944
            print '%3d FAILED!' % failures
515 by Martin Pool
- bzr selftest: return shell false (1) if any tests fail
945
            return 1
55 by mbp at sourcefrog
bzr selftest shows some counts of tests
946
        else:
947
            print
515 by Martin Pool
- bzr selftest: return shell false (1) if any tests fail
948
            return 0
55 by mbp at sourcefrog
bzr selftest shows some counts of tests
949
329 by Martin Pool
- refactor command functions into command classes
950
951
952
class cmd_version(Command):
953
    """Show version of bzr"""
954
    def run(self):
955
        show_version()
956
957
def show_version():
958
    print "bzr (bazaar-ng) %s" % bzrlib.__version__
959
    print bzrlib.__copyright__
960
    print "http://bazaar-ng.org/"
961
    print
962
    print "bzr comes with ABSOLUTELY NO WARRANTY.  bzr is free software, and"
963
    print "you may use, modify and redistribute it under the terms of the GNU"
964
    print "General Public License version 2 or later."
965
966
967
class cmd_rocks(Command):
968
    """Statement of optimism."""
969
    hidden = True
970
    def run(self):
971
        print "it sure does!"
972
493 by Martin Pool
- Merge aaron's merge command
973
def parse_spec(spec):
974
    if '/@' in spec:
975
        parsed = spec.split('/@')
976
        assert len(parsed) == 2
977
        if parsed[1] == "":
978
            parsed[1] = -1
979
        else:
980
            parsed[1] = int(parsed[1])
981
            assert parsed[1] >=0
982
    else:
983
        parsed = [spec, None]
984
    return parsed
985
986
class cmd_merge(Command):
987
    """Perform a three-way merge of trees."""
988
    takes_args = ['other_spec', 'base_spec']
989
990
    def run(self, other_spec, base_spec):
991
        merge.merge(parse_spec(other_spec), parse_spec(base_spec))
329 by Martin Pool
- refactor command functions into command classes
992
993
class cmd_assert_fail(Command):
994
    """Test reporting of assertion failures"""
995
    hidden = True
996
    def run(self):
997
        assert False, "always fails"
998
999
1000
class cmd_help(Command):
1001
    """Show help on a command or other topic.
1002
1003
    For a list of all available commands, say 'bzr help commands'."""
1004
    takes_args = ['topic?']
350 by Martin Pool
- refactor command aliases into command classes
1005
    aliases = ['?']
329 by Martin Pool
- refactor command functions into command classes
1006
    
1007
    def run(self, topic=None):
351 by Martin Pool
- Split out help functions into bzrlib.help
1008
        import help
1009
        help.help(topic)
1010
1 by mbp at sourcefrog
import from baz patch-364
1011
429 by Martin Pool
- New command update-stat-cache for testing
1012
class cmd_update_stat_cache(Command):
1013
    """Update stat-cache mapping inodes to SHA-1 hashes.
1014
1015
    For testing only."""
1016
    hidden = True
1017
    def run(self):
1018
        import statcache
1019
        b = Branch('.')
454 by Martin Pool
- fix update-stat-cache command
1020
        statcache.update_cache(b.base, b.read_working_inventory())
429 by Martin Pool
- New command update-stat-cache for testing
1021
1022
1 by mbp at sourcefrog
import from baz patch-364
1023
######################################################################
1024
# main routine
1025
1026
1027
# list of all available options; the rhs can be either None for an
1028
# option that takes no argument, or a constructor function that checks
1029
# the type.
1030
OPTIONS = {
1031
    'all':                    None,
1032
    'help':                   None,
389 by Martin Pool
- new commit --file option!
1033
    'file':                   unicode,
545 by Martin Pool
- --forward option for log
1034
    'forward':                None,
1 by mbp at sourcefrog
import from baz patch-364
1035
    'message':                unicode,
137 by mbp at sourcefrog
new --profile option
1036
    'profile':                None,
567 by Martin Pool
- New form 'bzr log -r FROM:TO'
1037
    'revision':               _parse_revision_str,
1 by mbp at sourcefrog
import from baz patch-364
1038
    'show-ids':               None,
12 by mbp at sourcefrog
new --timezone option for bzr log
1039
    'timezone':               str,
1 by mbp at sourcefrog
import from baz patch-364
1040
    'verbose':                None,
1041
    'version':                None,
286 by Martin Pool
- New bzr whoami --email option
1042
    'email':                  None,
1 by mbp at sourcefrog
import from baz patch-364
1043
    }
1044
1045
SHORT_OPTIONS = {
1046
    'm':                      'message',
389 by Martin Pool
- new commit --file option!
1047
    'F':                      'file', 
1 by mbp at sourcefrog
import from baz patch-364
1048
    'r':                      'revision',
1049
    'v':                      'verbose',
1050
}
1051
1052
1053
def parse_args(argv):
1054
    """Parse command line.
1055
    
1056
    Arguments and options are parsed at this level before being passed
1057
    down to specific command handlers.  This routine knows, from a
1058
    lookup table, something about the available options, what optargs
1059
    they take, and which commands will accept them.
1060
31 by Martin Pool
fix up parse_args doctest
1061
    >>> parse_args('--help'.split())
1 by mbp at sourcefrog
import from baz patch-364
1062
    ([], {'help': True})
31 by Martin Pool
fix up parse_args doctest
1063
    >>> parse_args('--version'.split())
1 by mbp at sourcefrog
import from baz patch-364
1064
    ([], {'version': True})
31 by Martin Pool
fix up parse_args doctest
1065
    >>> parse_args('status --all'.split())
1 by mbp at sourcefrog
import from baz patch-364
1066
    (['status'], {'all': True})
31 by Martin Pool
fix up parse_args doctest
1067
    >>> parse_args('commit --message=biter'.split())
17 by mbp at sourcefrog
allow --option=ARG syntax
1068
    (['commit'], {'message': u'biter'})
1 by mbp at sourcefrog
import from baz patch-364
1069
    """
1070
    args = []
1071
    opts = {}
1072
1073
    # TODO: Maybe handle '--' to end options?
1074
26 by mbp at sourcefrog
fix StopIteration error on python2.3(?)
1075
    while argv:
1076
        a = argv.pop(0)
1 by mbp at sourcefrog
import from baz patch-364
1077
        if a[0] == '-':
264 by Martin Pool
parse_args: option names must be ascii
1078
            # option names must not be unicode
1079
            a = str(a)
17 by mbp at sourcefrog
allow --option=ARG syntax
1080
            optarg = None
1 by mbp at sourcefrog
import from baz patch-364
1081
            if a[1] == '-':
1082
                mutter("  got option %r" % a)
17 by mbp at sourcefrog
allow --option=ARG syntax
1083
                if '=' in a:
1084
                    optname, optarg = a[2:].split('=', 1)
1085
                else:
1086
                    optname = a[2:]
1 by mbp at sourcefrog
import from baz patch-364
1087
                if optname not in OPTIONS:
1088
                    bailout('unknown long option %r' % a)
1089
            else:
1090
                shortopt = a[1:]
1091
                if shortopt not in SHORT_OPTIONS:
1092
                    bailout('unknown short option %r' % a)
1093
                optname = SHORT_OPTIONS[shortopt]
1094
            
1095
            if optname in opts:
1096
                # XXX: Do we ever want to support this, e.g. for -r?
1097
                bailout('repeated option %r' % a)
17 by mbp at sourcefrog
allow --option=ARG syntax
1098
                
1 by mbp at sourcefrog
import from baz patch-364
1099
            optargfn = OPTIONS[optname]
1100
            if optargfn:
17 by mbp at sourcefrog
allow --option=ARG syntax
1101
                if optarg == None:
26 by mbp at sourcefrog
fix StopIteration error on python2.3(?)
1102
                    if not argv:
17 by mbp at sourcefrog
allow --option=ARG syntax
1103
                        bailout('option %r needs an argument' % a)
1104
                    else:
26 by mbp at sourcefrog
fix StopIteration error on python2.3(?)
1105
                        optarg = argv.pop(0)
17 by mbp at sourcefrog
allow --option=ARG syntax
1106
                opts[optname] = optargfn(optarg)
1 by mbp at sourcefrog
import from baz patch-364
1107
            else:
17 by mbp at sourcefrog
allow --option=ARG syntax
1108
                if optarg != None:
1109
                    bailout('option %r takes no argument' % optname)
1 by mbp at sourcefrog
import from baz patch-364
1110
                opts[optname] = True
1111
        else:
1112
            args.append(a)
1113
1114
    return args, opts
1115
1116
1117
1118
329 by Martin Pool
- refactor command functions into command classes
1119
def _match_argform(cmd, takes_args, args):
1 by mbp at sourcefrog
import from baz patch-364
1120
    argdict = {}
26 by mbp at sourcefrog
fix StopIteration error on python2.3(?)
1121
329 by Martin Pool
- refactor command functions into command classes
1122
    # step through args and takes_args, allowing appropriate 0-many matches
1123
    for ap in takes_args:
1 by mbp at sourcefrog
import from baz patch-364
1124
        argname = ap[:-1]
1125
        if ap[-1] == '?':
62 by mbp at sourcefrog
- new find_branch_root function; based on suggestion from aaron
1126
            if args:
1127
                argdict[argname] = args.pop(0)
196 by mbp at sourcefrog
selected-file diff
1128
        elif ap[-1] == '*': # all remaining arguments
1129
            if args:
1130
                argdict[argname + '_list'] = args[:]
1131
                args = []
1132
            else:
1133
                argdict[argname + '_list'] = None
1 by mbp at sourcefrog
import from baz patch-364
1134
        elif ap[-1] == '+':
1135
            if not args:
329 by Martin Pool
- refactor command functions into command classes
1136
                raise BzrCommandError("command %r needs one or more %s"
1 by mbp at sourcefrog
import from baz patch-364
1137
                        % (cmd, argname.upper()))
1138
            else:
1139
                argdict[argname + '_list'] = args[:]
1140
                args = []
160 by mbp at sourcefrog
- basic support for moving files to different directories - have not done support for renaming them yet, but should be straightforward - some tests, but many cases are not handled yet i think
1141
        elif ap[-1] == '$': # all but one
1142
            if len(args) < 2:
329 by Martin Pool
- refactor command functions into command classes
1143
                raise BzrCommandError("command %r needs one or more %s"
160 by mbp at sourcefrog
- basic support for moving files to different directories - have not done support for renaming them yet, but should be straightforward - some tests, but many cases are not handled yet i think
1144
                        % (cmd, argname.upper()))
1145
            argdict[argname + '_list'] = args[:-1]
1146
            args[:-1] = []                
1 by mbp at sourcefrog
import from baz patch-364
1147
        else:
1148
            # just a plain arg
1149
            argname = ap
1150
            if not args:
329 by Martin Pool
- refactor command functions into command classes
1151
                raise BzrCommandError("command %r requires argument %s"
1 by mbp at sourcefrog
import from baz patch-364
1152
                        % (cmd, argname.upper()))
1153
            else:
1154
                argdict[argname] = args.pop(0)
1155
            
1156
    if args:
329 by Martin Pool
- refactor command functions into command classes
1157
        raise BzrCommandError("extra argument to command %s: %s"
1158
                              % (cmd, args[0]))
1 by mbp at sourcefrog
import from baz patch-364
1159
1160
    return argdict
1161
1162
1163
1164
def run_bzr(argv):
1165
    """Execute a command.
1166
1167
    This is similar to main(), but without all the trappings for
245 by mbp at sourcefrog
- control files always in utf-8-unix format
1168
    logging and error handling.  
1 by mbp at sourcefrog
import from baz patch-364
1169
    """
251 by mbp at sourcefrog
- factor out locale.getpreferredencoding()
1170
    argv = [a.decode(bzrlib.user_encoding) for a in argv]
245 by mbp at sourcefrog
- control files always in utf-8-unix format
1171
    
1 by mbp at sourcefrog
import from baz patch-364
1172
    try:
26 by mbp at sourcefrog
fix StopIteration error on python2.3(?)
1173
        args, opts = parse_args(argv[1:])
1 by mbp at sourcefrog
import from baz patch-364
1174
        if 'help' in opts:
351 by Martin Pool
- Split out help functions into bzrlib.help
1175
            import help
159 by mbp at sourcefrog
bzr commit --help now works
1176
            if args:
351 by Martin Pool
- Split out help functions into bzrlib.help
1177
                help.help(args[0])
159 by mbp at sourcefrog
bzr commit --help now works
1178
            else:
351 by Martin Pool
- Split out help functions into bzrlib.help
1179
                help.help()
1 by mbp at sourcefrog
import from baz patch-364
1180
            return 0
1181
        elif 'version' in opts:
336 by Martin Pool
- fix up 'bzr --version'
1182
            show_version()
1 by mbp at sourcefrog
import from baz patch-364
1183
            return 0
265 by Martin Pool
parse_args: command names must also be ascii
1184
        cmd = str(args.pop(0))
1 by mbp at sourcefrog
import from baz patch-364
1185
    except IndexError:
448 by Martin Pool
- bzr with no command now shows help, not just an error
1186
        import help
1187
        help.help()
1 by mbp at sourcefrog
import from baz patch-364
1188
        return 1
448 by Martin Pool
- bzr with no command now shows help, not just an error
1189
          
115 by mbp at sourcefrog
todo
1190
351 by Martin Pool
- Split out help functions into bzrlib.help
1191
    canonical_cmd, cmd_class = get_cmd_class(cmd)
1 by mbp at sourcefrog
import from baz patch-364
1192
137 by mbp at sourcefrog
new --profile option
1193
    # global option
1194
    if 'profile' in opts:
1195
        profile = True
1196
        del opts['profile']
1197
    else:
1198
        profile = False
1 by mbp at sourcefrog
import from baz patch-364
1199
1200
    # check options are reasonable
329 by Martin Pool
- refactor command functions into command classes
1201
    allowed = cmd_class.takes_options
1 by mbp at sourcefrog
import from baz patch-364
1202
    for oname in opts:
1203
        if oname not in allowed:
381 by Martin Pool
- Better message when a wrong argument is given
1204
            raise BzrCommandError("option '--%s' is not allowed for command %r"
329 by Martin Pool
- refactor command functions into command classes
1205
                                  % (oname, cmd))
176 by mbp at sourcefrog
New cat command contributed by janmar.
1206
137 by mbp at sourcefrog
new --profile option
1207
    # mix arguments and options into one dictionary
329 by Martin Pool
- refactor command functions into command classes
1208
    cmdargs = _match_argform(cmd, cmd_class.takes_args, args)
1209
    cmdopts = {}
136 by mbp at sourcefrog
new --show-ids option for 'deleted' command
1210
    for k, v in opts.items():
329 by Martin Pool
- refactor command functions into command classes
1211
        cmdopts[k.replace('-', '_')] = v
1 by mbp at sourcefrog
import from baz patch-364
1212
137 by mbp at sourcefrog
new --profile option
1213
    if profile:
338 by Martin Pool
- cleanup of some imports
1214
        import hotshot, tempfile
239 by mbp at sourcefrog
- remove profiler temporary file when done
1215
        pffileno, pfname = tempfile.mkstemp()
1216
        try:
1217
            prof = hotshot.Profile(pfname)
329 by Martin Pool
- refactor command functions into command classes
1218
            ret = prof.runcall(cmd_class, cmdopts, cmdargs) or 0
239 by mbp at sourcefrog
- remove profiler temporary file when done
1219
            prof.close()
1220
1221
            import hotshot.stats
1222
            stats = hotshot.stats.load(pfname)
1223
            #stats.strip_dirs()
1224
            stats.sort_stats('time')
1225
            ## XXX: Might like to write to stderr or the trace file instead but
1226
            ## print_stats seems hardcoded to stdout
1227
            stats.print_stats(20)
1228
            
337 by Martin Pool
- Clarify return codes from command objects
1229
            return ret.status
239 by mbp at sourcefrog
- remove profiler temporary file when done
1230
1231
        finally:
1232
            os.close(pffileno)
1233
            os.remove(pfname)
137 by mbp at sourcefrog
new --profile option
1234
    else:
500 by Martin Pool
- fix return value from run_bzr
1235
        return cmd_class(cmdopts, cmdargs).status 
1 by mbp at sourcefrog
import from baz patch-364
1236
1237
359 by Martin Pool
- pychecker fixups
1238
def _report_exception(summary, quiet=False):
267 by Martin Pool
- better reporting of errors
1239
    import traceback
1240
    log_error('bzr: ' + summary)
359 by Martin Pool
- pychecker fixups
1241
    bzrlib.trace.log_exception()
317 by Martin Pool
- better error message for broken pipe
1242
1243
    if not quiet:
1244
        tb = sys.exc_info()[2]
1245
        exinfo = traceback.extract_tb(tb)
1246
        if exinfo:
1247
            sys.stderr.write('  at %s:%d in %s()\n' % exinfo[-1][:3])
1248
        sys.stderr.write('  see ~/.bzr.log for debug information\n')
267 by Martin Pool
- better reporting of errors
1249
1250
1251
1 by mbp at sourcefrog
import from baz patch-364
1252
def main(argv):
317 by Martin Pool
- better error message for broken pipe
1253
    import errno
1254
    
344 by Martin Pool
- It's not an error to use the library without
1255
    bzrlib.open_tracefile(argv)
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1256
1 by mbp at sourcefrog
import from baz patch-364
1257
    try:
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1258
        try:
337 by Martin Pool
- Clarify return codes from command objects
1259
            try:
1260
                return run_bzr(argv)
1261
            finally:
1262
                # do this here inside the exception wrappers to catch EPIPE
1263
                sys.stdout.flush()
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1264
        except BzrError, e:
329 by Martin Pool
- refactor command functions into command classes
1265
            quiet = isinstance(e, (BzrCommandError))
359 by Martin Pool
- pychecker fixups
1266
            _report_exception('error: ' + e.args[0], quiet=quiet)
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1267
            if len(e.args) > 1:
1268
                for h in e.args[1]:
267 by Martin Pool
- better reporting of errors
1269
                    # some explanation or hints
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1270
                    log_error('  ' + h)
1271
            return 1
267 by Martin Pool
- better reporting of errors
1272
        except AssertionError, e:
1273
            msg = 'assertion failed'
1274
            if str(e):
1275
                msg += ': ' + str(e)
359 by Martin Pool
- pychecker fixups
1276
            _report_exception(msg)
318 by Martin Pool
- better error message for Ctrl-c
1277
            return 2
1278
        except KeyboardInterrupt, e:
359 by Martin Pool
- pychecker fixups
1279
            _report_exception('interrupted', quiet=True)
318 by Martin Pool
- better error message for Ctrl-c
1280
            return 2
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1281
        except Exception, e:
317 by Martin Pool
- better error message for broken pipe
1282
            quiet = False
419 by Martin Pool
- RemoteBranch.__str__ and repr
1283
            if (isinstance(e, IOError) 
1284
                and hasattr(e, 'errno')
1285
                and e.errno == errno.EPIPE):
317 by Martin Pool
- better error message for broken pipe
1286
                quiet = True
1287
                msg = 'broken pipe'
1288
            else:
1289
                msg = str(e).rstrip('\n')
359 by Martin Pool
- pychecker fixups
1290
            _report_exception(msg, quiet)
318 by Martin Pool
- better error message for Ctrl-c
1291
            return 2
260 by Martin Pool
- remove atexit() dependency for writing out execution times
1292
    finally:
1293
        bzrlib.trace.close_trace()
1 by mbp at sourcefrog
import from baz patch-364
1294
1295
1296
if __name__ == '__main__':
1297
    sys.exit(main(sys.argv))