~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Aaron Bentley
  • Date: 2006-04-17 13:40:34 UTC
  • mfrom: (360.1.4 bzrtools)
  • Revision ID: abentley@panoramicfeedback.com-20060417134034-6e23a2000508f1e6
Merge branch marking

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