~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Aaron Bentley
  • Date: 2006-03-24 13:13:01 UTC
  • Revision ID: aaron.bentley@utoronto.ca-20060324131301-6a6d76314c9e9b98
Removed docs for resolve and conflicts, now that they're in the mainline

Show diffs side-by-side

added added

removed removed

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