~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Aaron Bentley
  • Date: 2005-11-04 23:18:25 UTC
  • Revision ID: aaron.bentley@utoronto.ca-20051104231825-e1c079c6f517a0ee
Cherry-picked Robert's diff and push fixes

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 Aaron Bentley <aaron@aaronbentley.com>
2
 
# Copyright (C) 2005, 2006 Canonical Limited.
3
 
# Copyright (C) 2006 Michael Ellerman.
4
 
#
5
 
#    This program is free software; you can redistribute it and/or modify
6
 
#    it under the terms of the GNU General Public License as published by
7
 
#    the Free Software Foundation; either version 2 of the License, or
8
 
#    (at your option) any later version.
9
 
#
10
 
#    This program is distributed in the hope that it will be useful,
11
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
#    GNU General Public License for more details.
14
 
#
15
 
#    You should have received a copy of the GNU General Public License
16
 
#    along with this program; if not, write to the Free Software
17
 
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
 
 
 
1
#!/usr/bin/python
19
2
"""\
20
3
Various useful plugins for working with bzr.
21
4
"""
22
 
 
23
 
import bzrlib
24
 
 
25
 
from bzrlib.lazy_import import lazy_import
26
 
lazy_import(globals(), """
27
 
from bzrlib import help, urlutils
28
 
import shelf
29
 
""")
30
 
 
31
 
from version import version_info, __version__
32
 
from command import BzrToolsCommand
33
 
from errors import CommandError, NoPyBaz
34
 
from patchsource import BzrPatchSource
 
5
import bzrlib.commands
 
6
import push
 
7
import annotate
 
8
from shelf import Shelf
35
9
import sys
36
10
import os.path
37
 
 
38
 
import bzrlib.builtins
39
 
import bzrlib.commands
40
 
from bzrlib.branch import Branch
41
 
from bzrlib.bzrdir import BzrDir
42
 
from bzrlib.commands import get_cmd_object
 
11
from bzrlib.option import Option
 
12
import bzrlib.branch
43
13
from bzrlib.errors import BzrCommandError
44
 
import bzrlib.ignores
45
 
from bzrlib.trace import note
46
 
from bzrlib.option import Option
47
 
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
48
 
                                                 "external")))
49
 
 
50
 
import show_paths
51
 
from command import BzrToolsCommand
52
 
 
53
 
bzrlib.ignores.add_runtime_ignores(['./.shelf'])
54
 
 
55
 
 
56
 
class cmd_clean_tree(BzrToolsCommand):
 
14
sys.path.append(os.path.dirname(__file__))
 
15
 
 
16
Option.OPTIONS['ignored'] = Option('ignored',
 
17
        help='delete all ignored files.')
 
18
Option.OPTIONS['detritus'] = Option('detritus',
 
19
        help='delete conflict files merge backups, and failed selftest dirs.' +
 
20
              '(*.THIS, *.BASE, *.OTHER, *~, *.tmp')
 
21
Option.OPTIONS['dry-run'] = Option('dry-run',
 
22
        help='show files to delete instead of deleting them.')
 
23
 
 
24
class cmd_clean_tree(bzrlib.commands.Command):
57
25
    """Remove unwanted files from working tree.
58
 
 
59
 
    By default, only unknown files, not ignored files, are deleted.  Versioned
60
 
    files are never deleted.
61
 
 
62
 
    Another class is 'detritus', which includes files emitted by bzr during
63
 
    normal operations and selftests.  (The value of these files decreases with
64
 
    time.)
65
 
 
66
 
    If no options are specified, unknown files are deleted.  Otherwise, option
67
 
    flags are respected, and may be combined.
68
 
 
69
 
    To check what clean-tree will do, use --dry-run.
 
26
    Normally, ignored files are left alone.
70
27
    """
71
 
    takes_options = [Option('ignored', help='Delete all ignored files.'),
72
 
                     Option('detritus', help='Delete conflict files, merge'
73
 
                            ' backups, and failed selftest dirs.'),
74
 
                     Option('unknown',
75
 
                            help='Delete files unknown to bzr (default).'),
76
 
                     Option('dry-run', help='Show files to delete instead of'
77
 
                            ' deleting them.')]
78
 
    def run(self, unknown=False, ignored=False, detritus=False, dry_run=False):
 
28
    takes_options = ['ignored', 'detritus', 'dry-run']
 
29
    def run(self, ignored=False, detritus=False, dry_run=False):
79
30
        from clean_tree import clean_tree
80
 
        if not (unknown or ignored or detritus):
81
 
            unknown = True
82
 
        clean_tree('.', unknown=unknown, ignored=ignored, detritus=detritus, 
83
 
                   dry_run=dry_run)
84
 
 
85
 
 
86
 
class cmd_graph_ancestry(BzrToolsCommand):
 
31
        clean_tree('.', ignored=ignored, detritus=detritus, dry_run=dry_run)
 
32
 
 
33
Option.OPTIONS['no-collapse'] = Option('no-collapse')
 
34
Option.OPTIONS['no-antialias'] = Option('no-antialias')
 
35
Option.OPTIONS['cluster'] = Option('cluster')
 
36
Option.OPTIONS['merge-branch'] = Option('merge-branch',type=str)
 
37
 
 
38
class cmd_graph_ancestry(bzrlib.commands.Command):
87
39
    """Produce ancestry graphs using dot.
88
40
    
89
41
    Output format is detected according to file extension.  Some of the more
90
 
    common output formats are html, png, gif, svg, ps.  An extension of '.dot'
91
 
    will cause a dot graph file to be produced.  HTML output has mouseovers
92
 
    that show the commit message.
 
42
    common output formats are png, gif, svg, ps.  An extension of '.dot' will
 
43
    cause a dot graph file to be produced.
93
44
 
94
45
    Branches are labeled r?, where ? is the revno.  If they have no revno,
95
46
    with the last 5 characters of their revision identifier are used instead.
96
 
 
97
 
    The value starting with d is "(maximum) distance from the null revision".
98
 
 
 
47
    
99
48
    If --merge-branch is specified, the two branches are compared and a merge
100
49
    base is selected.
101
 
 
 
50
    
102
51
    Legend:
103
52
    white    normal revision
104
53
    yellow   THIS  history
105
54
    red      OTHER history
106
55
    orange   COMMON history
107
56
    blue     COMMON non-history ancestor
108
 
    green    Merge base (COMMON ancestor farthest from the null revision)
109
 
    dotted   Ghost revision (missing from branch storage)
 
57
    dotted   Missing from branch storage
110
58
 
111
 
    Ancestry is usually collapsed by skipping revisions with a single parent
 
59
    Ancestry is usually collapsed by removing revisions with a single parent
112
60
    and descendant.  The number of skipped revisions is shown on the arrow.
113
61
    This feature can be disabled with --no-collapse.
114
62
 
118
66
    If available, rsvg is used to antialias PNG and JPEG output, but this can
119
67
    be disabled with --no-antialias.
120
68
    """
121
 
    takes_args = ['file', 'merge_branch?']
122
 
    takes_options = [Option('no-collapse', help="Do not skip simple nodes."),
123
 
                     Option('no-antialias',
124
 
                     help="Do not use rsvg to produce antialiased output."),
125
 
                     Option('merge-branch', type=str,
126
 
                     help="Use this branch to calcuate a merge base."),
127
 
                     Option('cluster', help="Use clustered output."),
128
 
                     Option('max-distance',
129
 
                            help="Show no nodes farther than this.", type=int),
130
 
                     Option('directory',
131
 
                            help='Source branch to use (default is current'
132
 
                            ' directory).',
133
 
                            short_name='d',
134
 
                            type=unicode),
135
 
                    ]
136
 
    def run(self, file, merge_branch=None, no_collapse=False,
137
 
            no_antialias=False, cluster=False, max_distance=100,
138
 
            directory='.'):
139
 
        if max_distance == -1:
140
 
            max_distance = None
 
69
    takes_args = ['branch', 'file']
 
70
    takes_options = ['no-collapse', 'no-antialias', 'merge-branch', 'cluster']
 
71
    def run(self, branch, file, no_collapse=False, no_antialias=False,
 
72
        merge_branch=None, cluster=False):
141
73
        import graph
142
74
        if cluster:
143
75
            ranking = "cluster"
144
76
        else:
145
77
            ranking = "forced"
146
 
        graph.write_ancestry_file(directory, file, not no_collapse,
147
 
                                  not no_antialias, merge_branch, ranking,
148
 
                                  max_distance=max_distance)
149
 
 
150
 
 
151
 
class cmd_fetch_ghosts(BzrToolsCommand):
 
78
        graph.write_ancestry_file(branch, file, not no_collapse, 
 
79
                                  not no_antialias, merge_branch, ranking)
 
80
 
 
81
class cmd_fetch_ghosts(bzrlib.commands.Command):
152
82
    """Attempt to retrieve ghosts from another branch.
153
83
    If the other branch is not supplied, the last-pulled branch is used.
154
84
    """
155
85
    aliases = ['fetch-missing']
156
86
    takes_args = ['branch?']
157
 
    takes_options = [Option('no-fix', help="Skip additional synchonization.")]
158
 
    def run(self, branch=None, no_fix=False):
 
87
    def run(self, branch=None):
159
88
        from fetch_ghosts import fetch_ghosts
160
 
        fetch_ghosts(branch, no_fix)
 
89
        fetch_ghosts(branch)
161
90
 
162
91
strip_help="""Strip the smallest prefix containing num leading slashes  from \
163
92
each file name found in the patch file."""
164
 
 
165
 
 
166
 
class cmd_patch(BzrToolsCommand):
 
93
Option.OPTIONS['strip'] = Option('strip', type=int, help=strip_help)
 
94
class cmd_patch(bzrlib.commands.Command):
167
95
    """Apply a named patch to the current tree.
168
96
    """
169
97
    takes_args = ['filename?']
170
 
    takes_options = [Option('strip', type=int, help=strip_help),
171
 
                     Option('silent', help='Suppress chatter.')]
172
 
    def run(self, filename=None, strip=None, silent=False):
 
98
    takes_options = ['strip']
 
99
    def run(self, filename=None, strip=0):
173
100
        from patch import patch
174
 
        from bzrlib.workingtree import WorkingTree
175
 
        wt = WorkingTree.open_containing('.')[0]
176
 
        if strip is None:
177
 
            strip = 0
178
 
        return patch(wt, filename, strip, silent)
179
 
 
180
 
 
181
 
class cmd_shelve(BzrToolsCommand):
182
 
    """Temporarily set aside some changes from the current tree.
183
 
 
184
 
    Shelve allows you to temporarily put changes you've made "on the shelf",
185
 
    ie. out of the way, until a later time when you can bring them back from
186
 
    the shelf with the 'unshelve' command.
187
 
 
188
 
    Shelve is intended to help separate several sets of text changes that have
189
 
    been inappropriately mingled.  If you just want to get rid of all changes
190
 
    (text and otherwise) and you don't need to restore them later, use revert.
191
 
    If you want to shelve all text changes at once, use shelve --all.
192
 
 
193
 
    By default shelve asks you what you want to shelve, press '?' at the
194
 
    prompt to get help. To shelve everything run shelve --all.
195
 
 
196
 
    If filenames are specified, only the changes to those files will be
197
 
    shelved, other files will be left untouched.
198
 
 
199
 
    If a revision is specified, changes since that revision will be shelved.
200
 
 
201
 
    You can put multiple items on the shelf. Normally each time you run
202
 
    unshelve the most recently shelved changes will be reinstated. However,
203
 
    you can also unshelve changes in a different order by explicitly
204
 
    specifiying which changes to unshelve. This works best when the changes
205
 
    don't depend on each other.
206
 
 
207
 
    While you have patches on the shelf you can view and manipulate them with
208
 
    the 'shelf' command. Run 'bzr shelf -h' for more info.
 
101
        from bzrlib.branch import Branch
 
102
        b = Branch.open_containing('.')[0]
 
103
        return patch(b, filename, strip)
 
104
 
 
105
 
 
106
class cmd_shelve(bzrlib.commands.Command):
 
107
    """Temporarily remove some changes from the current tree.
 
108
    Use 'unshelve' to restore these changes.
 
109
 
 
110
    If filenames are specified, only changes to those files will be unshelved.
 
111
    If a revision is specified, all changes since that revision will may be
 
112
    unshelved.
209
113
    """
210
 
 
211
114
    takes_args = ['file*']
212
 
    takes_options = [Option('message',
213
 
            help='A message to associate with the shelved changes.',
214
 
            short_name='m', type=unicode),
215
 
            'revision',
216
 
            Option('all', help='Shelve all changes without prompting.'),
217
 
            Option('no-color', help='Never display changes in color.')]
 
115
    takes_options = ['all', 'message', 'revision']
 
116
    def run(self, all=False, file_list=None, message=None, revision=None):
 
117
        if file_list is not None and len(file_list) > 0:
 
118
            branchdir = file_list[0]
 
119
        else:
 
120
            branchdir = '.'
218
121
 
219
 
    def run(self, all=False, file_list=None, message=None, revision=None,
220
 
            no_color=False):
221
122
        if revision is not None and revision:
222
123
            if len(revision) == 1:
223
124
                revision = revision[0]
224
125
            else:
225
 
                raise CommandError("shelve only accepts a single revision "
 
126
                raise BzrCommandError("shelve only accepts a single revision "
226
127
                                  "parameter.")
227
128
 
228
 
        source = BzrPatchSource(revision, file_list)
229
 
        s = shelf.Shelf(source.base)
230
 
        s.shelve(source, all, message, no_color)
231
 
        return 0
232
 
 
233
 
 
234
 
# The following classes are only used as subcommands for 'shelf', they're
235
 
# not to be registered directly with bzr.
236
 
 
237
 
class cmd_shelf_list(bzrlib.commands.Command):
238
 
    """List the patches on the current shelf."""
239
 
    aliases = ['list', 'ls']
240
 
    def run(self):
241
 
        self.shelf.list()
242
 
 
243
 
 
244
 
class cmd_shelf_delete(bzrlib.commands.Command):
245
 
    """Delete the patch from the current shelf."""
246
 
    aliases = ['delete', 'del']
247
 
    takes_args = ['patch']
248
 
    def run(self, patch):
249
 
        self.shelf.delete(patch)
250
 
 
251
 
 
252
 
class cmd_shelf_switch(bzrlib.commands.Command):
253
 
    """Switch to the other shelf, create it if necessary."""
254
 
    aliases = ['switch']
255
 
    takes_args = ['othershelf']
256
 
    def run(self, othershelf):
257
 
        s = shelf.Shelf(self.shelf.base, othershelf)
258
 
        s.make_default()
259
 
 
260
 
 
261
 
class cmd_shelf_show(bzrlib.commands.Command):
262
 
    """Show the contents of the specified or topmost patch."""
263
 
    aliases = ['show', 'cat', 'display']
264
 
    takes_args = ['patch?']
265
 
    def run(self, patch=None):
266
 
        self.shelf.display(patch)
267
 
 
268
 
 
269
 
class cmd_shelf_upgrade(bzrlib.commands.Command):
270
 
    """Upgrade old format shelves."""
271
 
    aliases = ['upgrade']
272
 
    def run(self):
273
 
        self.shelf.upgrade()
274
 
 
275
 
 
276
 
class cmd_shelf(BzrToolsCommand):
277
 
    """Perform various operations on your shelved patches. See also shelve."""
278
 
    takes_args = ['subcommand', 'args*']
279
 
 
280
 
    subcommands = [cmd_shelf_list, cmd_shelf_delete, cmd_shelf_switch,
281
 
        cmd_shelf_show, cmd_shelf_upgrade]
282
 
 
283
 
    def run(self, subcommand, args_list):
284
 
        import sys
285
 
 
286
 
        if args_list is None:
287
 
            args_list = []
288
 
        cmd = self._get_cmd_object(subcommand)
289
 
        source = BzrPatchSource()
290
 
        s = shelf.Shelf(source.base)
291
 
        cmd.shelf = s
292
 
 
293
 
        if args_list is None:
294
 
            args_list = []
295
 
        return cmd.run_argv_aliases(args_list)
296
 
 
297
 
    def _get_cmd_object(self, cmd_name):
298
 
        for cmd_class in self.subcommands:
299
 
            for alias in cmd_class.aliases:
300
 
                if alias == cmd_name:
301
 
                    return cmd_class()
302
 
        raise CommandError("Unknown shelf subcommand '%s'" % cmd_name)
303
 
 
304
 
    def help(self):
305
 
        text = ["%s\n\nSubcommands:\n" % self.__doc__]
306
 
 
307
 
        for cmd_class in self.subcommands:
308
 
            text.extend(self.sub_help(cmd_class) + ['\n'])
309
 
 
310
 
        return ''.join(text)
311
 
 
312
 
    def sub_help(self, cmd_class):
313
 
        text = []
314
 
        cmd_obj = cmd_class()
315
 
        indent = 2 * ' '
316
 
 
317
 
        usage = cmd_obj._usage()
318
 
        usage = usage.replace('bzr shelf-', '')
319
 
        text.append('%s%s\n' % (indent, usage))
320
 
 
321
 
        text.append('%s%s\n' % (2 * indent, cmd_class.__doc__))
322
 
 
323
 
        # Somewhat copied from bzrlib.help.help_on_command_options
324
 
        option_help = []
325
 
        for option_name, option in sorted(cmd_obj.options().items()):
326
 
            if option_name == 'help':
327
 
                continue
328
 
            option_help.append('%s--%s' % (3 * indent, option_name))
329
 
            if option.type is not None:
330
 
                option_help.append(' %s' % option.argname.upper())
331
 
            if option.short_name():
332
 
                option_help.append(', -%s' % option.short_name())
333
 
            option_help.append('%s%s\n' % (2 * indent, option.help))
334
 
 
335
 
        if len(option_help) > 0:
336
 
            text.append('%soptions:\n' % (2 * indent))
337
 
            text.extend(option_help)
338
 
 
339
 
        return text
340
 
 
341
 
 
342
 
class cmd_unshelve(BzrToolsCommand):
343
 
    """Restore shelved changes.
344
 
 
345
 
    By default the most recently shelved changes are restored. However if you
346
 
    specify a patch by name those changes will be restored instead.
347
 
 
348
 
    See 'shelve' for more information.
349
 
    """
350
 
    takes_options = [
351
 
            Option('all', help='Unshelve all changes without prompting.'),
352
 
            Option('force', help='Force unshelving even if errors occur.'),
353
 
            Option('no-color', help='Never display changes in color.')
354
 
        ]
355
 
    takes_args = ['patch?']
356
 
    def run(self, patch=None, all=False, force=False, no_color=False):
357
 
        source = BzrPatchSource()
358
 
        s = shelf.Shelf(source.base)
359
 
        s.unshelve(source, patch, all, force, no_color)
360
 
        return 0
361
 
 
362
 
 
363
 
class cmd_shell(BzrToolsCommand):
364
 
    """Begin an interactive shell tailored for bzr.
365
 
    Bzr commands can be used without typing bzr first, and will be run natively
366
 
    when possible.  Tab completion is tailored for bzr.  The shell prompt shows
367
 
    the branch nick, revno, and path.
368
 
 
369
 
    If it encounters any moderately complicated shell command, it will punt to
370
 
    the system shell.
371
 
 
372
 
    Example:
373
 
    $ bzr shell
374
 
    bzr bzrtools:287/> status
375
 
    modified:
376
 
      __init__.py
377
 
    bzr bzrtools:287/> status --[TAB][TAB]
378
 
    --all        --help       --revision   --show-ids
379
 
    bzr bzrtools:287/> status --
380
 
    """
 
129
        s = Shelf(branchdir)
 
130
        return s.shelve(all, message, revision, file_list)
 
131
 
 
132
 
 
133
class cmd_unshelve(bzrlib.commands.Command):
 
134
    """Restore previously-shelved changes to the current tree.
 
135
    See also 'shelve'.
 
136
    """
 
137
    def run(self):
 
138
        s = Shelf('.')
 
139
        return s.unshelve()
 
140
 
 
141
class cmd_shell(bzrlib.commands.Command):
381
142
    def run(self):
382
143
        import shell
383
 
        return shell.run_shell()
384
 
 
385
 
 
386
 
class cmd_branch_history(BzrToolsCommand):
387
 
    """\
388
 
    Display the development history of a branch.
389
 
 
390
 
    Each different committer or branch nick is considered a different line of
391
 
    development.  Committers are treated as the same if they have the same
392
 
    name, or if they have the same email address.
393
 
    """
394
 
    takes_args = ["branch?"]
395
 
    def run(self, branch=None):
396
 
        from branchhistory import branch_history
397
 
        return branch_history(branch)
398
 
 
399
 
 
400
 
class cmd_zap(BzrToolsCommand):
401
 
    """\
402
 
    Remove a lightweight checkout, if it can be done safely.
403
 
 
404
 
    This command will remove a lightweight checkout without losing data.  That
405
 
    means it only removes lightweight checkouts, and only if they have no
406
 
    uncommitted changes.
407
 
 
408
 
    If --branch is specified, the branch will be deleted too, but only if the
409
 
    the branch has no new commits (relative to its parent).
410
 
    """
411
 
    takes_options = [Option("branch", help="Remove associated branch from"
412
 
                                           " repository."),
413
 
                     Option('force', help='Delete tree even if contents are'
414
 
                     ' modified.')]
415
 
    takes_args = ["checkout"]
416
 
    def run(self, checkout, branch=False, force=False):
417
 
        from zap import zap
418
 
        return zap(checkout, remove_branch=branch, allow_modified=force)
419
 
 
420
 
 
421
 
class cmd_cbranch(BzrToolsCommand):
422
 
    """
423
 
    Create a new checkout, associated with a new repository branch.
424
 
 
425
 
    When you cbranch, bzr looks up a target location in locations.conf, and
426
 
    creates the branch there.
427
 
 
428
 
    In your locations.conf, add the following lines:
429
 
    [/working_directory_root]
430
 
    cbranch_target = /branch_root
431
 
    cbranch_target:policy = appendpath
432
 
 
433
 
    This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
434
 
    working directory root, the branch will be created in
435
 
    "/branch_root/foo/baz"
436
 
 
437
 
    NOTE: cbranch also supports "cbranch_root", but that behaviour is
438
 
    deprecated.
439
 
    """
440
 
    takes_options = [Option("lightweight",
441
 
                            help="Create a lightweight checkout."), 'revision',
442
 
                     Option('files-from', type=unicode,
443
 
                            help='Accelerate checkout using files from this'
444
 
                                 ' tree.')]
445
 
    takes_args = ["source", "target?"]
446
 
    def run(self, source, target=None, lightweight=False, revision=None,
447
 
            files_from=None):
448
 
        from cbranch import cbranch
449
 
        return cbranch(source, target, lightweight=lightweight,
450
 
                       revision=revision, files_from=files_from)
451
 
 
452
 
 
453
 
class cmd_branches(BzrToolsCommand):
454
 
    """Scan a location for branches"""
455
 
    takes_args = ["location?"]
456
 
    def run(self, location=None):
457
 
        from branches import branches
458
 
        return branches(location)
459
 
 
460
 
class cmd_trees(BzrToolsCommand):
461
 
    """Scan a location for trees"""
462
 
    takes_args = ['location?']
463
 
    def run(self, location='.'):
464
 
        from bzrlib.workingtree import WorkingTree
465
 
        from bzrlib.transport import get_transport
466
 
        t = get_transport(location)
467
 
        for tree in WorkingTree.find_trees(location):
468
 
            self.outf.write('%s\n' % t.relpath(
469
 
                tree.bzrdir.root_transport.base))
470
 
 
471
 
class cmd_multi_pull(BzrToolsCommand):
472
 
    """Pull all the branches under a location, e.g. a repository.
473
 
 
474
 
    Both branches present in the directory and the branches of checkouts are
475
 
    pulled.
476
 
    """
477
 
    takes_args = ["location?"]
478
 
    def run(self, location=None):
479
 
        from bzrlib.transport import get_transport
480
 
        from bzrtools import iter_branch_tree
481
 
        if location is None:
482
 
            location = '.'
483
 
        t = get_transport(location)
484
 
        possible_transports = []
485
 
        if not t.listable():
486
 
            print "Can't list this type of location."
487
 
            return 3
488
 
        for branch, wt in iter_branch_tree(t):
489
 
            if wt is None:
490
 
                pullable = branch
491
 
            else:
492
 
                pullable = wt
493
 
            parent = branch.get_parent()
494
 
            if parent is None:
495
 
                continue
496
 
            if wt is not None:
497
 
                base = wt.basedir
498
 
            else:
499
 
                base = branch.base
500
 
            if base.startswith(t.base):
501
 
                relpath = base[len(t.base):].rstrip('/')
502
 
            else:
503
 
                relpath = base
504
 
            print "Pulling %s from %s" % (relpath, parent)
505
 
            try:
506
 
                branch_t = get_transport(parent, possible_transports)
507
 
                pullable.pull(Branch.open_from_transport(branch_t))
508
 
            except Exception, e:
509
 
                print e
510
 
 
511
 
 
512
 
 
513
 
class cmd_import(BzrToolsCommand):
514
 
    """Import sources from a directory, tarball or zip file
515
 
 
516
 
    This command will import a directory, tarball or zip file into a bzr
517
 
    tree, replacing any versioned files already present.  If a directory is
518
 
    specified, it is used as the target.  If the directory does not exist, or
519
 
    is not versioned, it is created.
520
 
 
521
 
    Tarballs may be gzip or bzip2 compressed.  This is autodetected.
522
 
 
523
 
    If the tarball or zip has a single root directory, that directory is
524
 
    stripped when extracting the tarball.  This is not done for directories.
525
 
    """
526
 
 
527
 
    takes_args = ['source', 'tree?']
528
 
    def run(self, source, tree=None):
529
 
        from upstream_import import do_import
530
 
        do_import(source, tree)
531
 
 
532
 
 
533
 
class cmd_cdiff(BzrToolsCommand):
534
 
    """A color version of bzr's diff"""
535
 
    takes_args = property(lambda x: get_cmd_object('diff').takes_args)
536
 
    takes_options = list(get_cmd_object('diff').takes_options) + [
537
 
        Option('check-style',
538
 
            help='Warn if trailing whitespace or spurious changes have been'
539
 
                 ' added.')]
540
 
 
541
 
    def run(self, check_style=False, *args, **kwargs):
542
 
        from colordiff import colordiff
543
 
        colordiff(check_style, *args, **kwargs)
544
 
 
545
 
 
546
 
class cmd_baz_import(BzrToolsCommand):
547
 
    """Import an Arch or Baz archive into a bzr repository.
548
 
 
549
 
    This command should be used on local archives (or mirrors) only.  It is
550
 
    quite slow on remote archives.
551
 
 
552
 
    reuse_history allows you to specify any previous imports you
553
 
    have done of different archives, which this archive has branches
554
 
    tagged from. This will dramatically reduce the time to convert
555
 
    the archive as it will not have to convert the history already
556
 
    converted in that other branch.
557
 
 
558
 
    If you specify prefixes, only branches whose names start with that prefix
559
 
    will be imported.  Skipped branches will be listed, so you can import any
560
 
    branches you missed by accident.  Here's an example of doing a partial
561
 
    import from thelove@canonical.com:
562
 
    bzr baz-import thelove thelove@canonical.com --prefixes dists:talloc-except
563
 
 
564
 
    WARNING: Encoding should not be specified unless necessary, because if you
565
 
    specify an encoding, your converted branch will not interoperate with
566
 
    independently-converted branches, unless the other branches were converted
567
 
    with exactly the same encoding.  Any encoding recognized by Python may
568
 
    be specified.  Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
569
 
    are incompatible.
570
 
    """
571
 
    takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
572
 
    takes_options = ['verbose', Option('prefixes', type=str,
573
 
                     help="Prefixes of branches to import, colon-separated."),
574
 
                     Option('encoding', type=str,
575
 
                     help='Force encoding to specified value.  See WARNING.')]
576
 
 
577
 
    def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
578
 
            reuse_history_list=[], prefixes=None):
579
 
        from errors import NoPyBaz
580
 
        try:
581
 
            import baz_import
582
 
            baz_import.baz_import(to_root_dir, from_archive, encoding,
583
 
                                  verbose, reuse_history_list, prefixes)
584
 
        except NoPyBaz:
585
 
            print "This command is disabled.  Please install PyBaz."
586
 
 
587
 
 
588
 
class cmd_baz_import_branch(BzrToolsCommand):
589
 
    """Import an Arch or Baz branch into a bzr branch.
590
 
 
591
 
    WARNING: Encoding should not be specified unless necessary, because if you
592
 
    specify an encoding, your converted branch will not interoperate with
593
 
    independently-converted branches, unless the other branches were converted
594
 
    with exactly the same encoding.  Any encoding recognized by Python may
595
 
    be specified.  Aliases are not detected, so 'utf_8', 'U8', 'UTF' and 'utf8'
596
 
    are incompatible.
597
 
    """
598
 
    takes_args = ['to_location', 'from_branch?', 'reuse_history*']
599
 
    takes_options = ['verbose', 
600
 
                     Option('max-count', type=int, 
601
 
                     help='Maximim revisions to import at once.'),
602
 
                     Option('encoding', type=str,
603
 
                     help='Force encoding to specified value.  See WARNING.')]
604
 
 
605
 
    def run(self, to_location, from_branch=None, fast=False, max_count=None,
606
 
            encoding=None, verbose=False, dry_run=False,
607
 
            reuse_history_list=[]):
608
 
        from errors import NoPyBaz
609
 
        try:
610
 
            import baz_import
611
 
            baz_import.baz_import_branch(to_location, from_branch, fast,
612
 
                                         max_count, verbose, encoding, dry_run,
613
 
                                         reuse_history_list)
614
 
        except NoPyBaz:
615
 
            print "This command is disabled.  Please install PyBaz."
616
 
 
617
 
 
618
 
class cmd_rspush(BzrToolsCommand):
619
 
    """Upload this branch to another location using rsync.
620
 
 
621
 
    If no location is specified, the last-used location will be used.  To
622
 
    prevent dirty trees from being uploaded, rspush will error out if there are
623
 
    unknown files or local changes.  It will also error out if the upstream
624
 
    directory is non-empty and not an earlier version of the branch.
625
 
    """
626
 
    takes_args = ['location?']
627
 
    takes_options = [Option('overwrite', help='Ignore differences between'
628
 
                            ' branches and overwrite unconditionally.'),
629
 
                     Option('no-tree', help='Do not push the working tree,'
630
 
                            ' just the .bzr.')]
631
 
 
632
 
    def run(self, location=None, overwrite=False, no_tree=False):
633
 
        from bzrlib import workingtree
634
 
        import bzrtools
635
 
        cur_branch = workingtree.WorkingTree.open_containing(".")[0]
636
 
        bzrtools.rspush(cur_branch, location, overwrite=overwrite,
637
 
                      working_tree=not no_tree)
638
 
 
639
 
 
640
 
commands = [
641
 
            cmd_baz_import,
642
 
            cmd_baz_import_branch,
643
 
            cmd_branches,
644
 
            cmd_trees,
645
 
            cmd_branch_history,
646
 
            cmd_cbranch,
647
 
            cmd_cdiff,
648
 
            cmd_clean_tree,
649
 
            cmd_fetch_ghosts,
650
 
            cmd_graph_ancestry,
651
 
            cmd_import,
652
 
            cmd_multi_pull,
653
 
            cmd_patch,
654
 
            cmd_rspush,
655
 
            cmd_shelf,
656
 
            cmd_shell,
657
 
            cmd_shelve,
658
 
            cmd_unshelve,
659
 
            cmd_zap,
660
 
            ]
 
144
        shell.run_shell()
 
145
 
 
146
commands = [cmd_shelve, cmd_unshelve, cmd_clean_tree, cmd_graph_ancestry,
 
147
            cmd_fetch_ghosts, cmd_patch, cmd_shell]
 
148
 
 
149
command_decorators = []
 
150
 
 
151
import bzrlib.builtins
 
152
if not hasattr(bzrlib.builtins, "cmd_annotate"):
 
153
    commands.append(annotate.cmd_annotate)
 
154
if not hasattr(bzrlib.builtins, "cmd_push"):
 
155
    commands.append(push.cmd_push)
 
156
else:
 
157
    command_decorators.append(push.cmd_push)
 
158
 
 
159
from errors import NoPyBaz
 
160
try:
 
161
    import baz_import
 
162
    commands.append(baz_import.cmd_baz_import)
 
163
 
 
164
except NoPyBaz:
 
165
    class cmd_baz_import(bzrlib.commands.Command):
 
166
        """Disabled. (Requires PyBaz)"""
 
167
        takes_args = ['to_root_dir?', 'from_archive?']
 
168
        takes_options = ['verbose']
 
169
        def run(self, to_root_dir=None, from_archive=None, verbose=False):
 
170
            print "This command is disabled.  Please install PyBaz."
 
171
    commands.append(cmd_baz_import)
661
172
 
662
173
 
663
174
if hasattr(bzrlib.commands, 'register_command'):
664
175
    for command in commands:
665
176
        bzrlib.commands.register_command(command)
 
177
    for command in command_decorators:
 
178
        command._original_command = bzrlib.commands.register_command(
 
179
            command, True)
666
180
 
667
181
 
668
182
def test_suite():
669
 
    from bzrlib.tests.TestUtil import TestLoader
670
 
    import tests
671
 
    from doctest import DocTestSuite, ELLIPSIS
672
 
    from unittest import TestSuite
673
 
    import bzrtools
674
 
    import tests.clean_tree
675
 
    import tests.is_clean
676
 
    import tests.test_patch
677
 
    import tests.test_rspush
678
 
    import tests.upstream_import
679
 
    import zap
680
 
    import tests.blackbox
681
 
    import tests.shelf_tests
 
183
    from doctest import DocTestSuite
 
184
    from unittest import TestSuite, TestLoader
 
185
    import clean_tree
 
186
    import blackbox
 
187
    import shelf_tests
682
188
    result = TestSuite()
683
 
    result.addTest(DocTestSuite(bzrtools, optionflags=ELLIPSIS))
684
 
    result.addTest(tests.clean_tree.test_suite())
685
 
    try:
686
 
        import baz_import
687
 
        result.addTest(DocTestSuite(baz_import))
688
 
    except NoPyBaz:
689
 
        pass
690
 
    result.addTest(tests.test_suite())
691
 
    result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))
692
 
    result.addTest(tests.blackbox.test_suite())
693
 
    result.addTest(tests.upstream_import.test_suite())
694
 
    result.addTest(zap.test_suite())
695
 
    result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))
696
 
    result.addTest(TestLoader().loadTestsFromModule(tests.test_patch))
697
 
    result.addTest(TestLoader().loadTestsFromModule(tests.test_rspush))
 
189
    result.addTest(DocTestSuite(bzrtools))
 
190
    result.addTest(clean_tree.test_suite())
 
191
    result.addTest(TestLoader().loadTestsFromModule(shelf_tests))
 
192
    result.addTest(blackbox.test_suite())
698
193
    return result