~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Aaron Bentley
  • Date: 2007-06-12 22:09:44 UTC
  • mfrom: (540.1.2 bzrtools-0.17)
  • Revision ID: aaron.bentley@utoronto.ca-20070612220944-5zw4hlzp1ctq6mkl
Merge fixes from 0.17

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Aaron Bentley <aaron.bentley@utoronto.ca>
 
1
# Copyright (C) 2005, 2006, 2007 Aaron Bentley <aaron.bentley@utoronto.ca>
2
2
# Copyright (C) 2005, 2006 Canonical Limited.
3
3
# Copyright (C) 2006 Michael Ellerman.
4
4
#
23
23
import bzrlib
24
24
 
25
25
 
26
 
__version__ = '0.14.0'
 
26
__version__ = '0.18.0'
27
27
 
28
28
 
29
29
version_info = tuple(int(n) for n in __version__.split('.'))
39
39
    """
40
40
    desired_plus = (desired[0], desired[1]+1)
41
41
    bzrlib_version = bzrlib.version_info[:2]
42
 
    if bzrlib_version == desired:
 
42
    if bzrlib_version == desired or (bzrlib_version == desired_plus and
 
43
                                     bzrlib.version_info[3] == 'dev'):
43
44
        return
44
45
    try:
45
46
        from bzrlib.trace import warning
47
48
        # get the message out any way we can
48
49
        from warnings import warn as warning
49
50
    if bzrlib_version < desired:
50
 
        warning('Installed bzr version %s is too old to be used with bzrtools'
51
 
                ' %s.' % (bzrlib.__version__, __version__))
 
51
        warning('Installed Bazaar version %s is too old to be used with'
 
52
                ' plugin \n'
 
53
                '"Bzrtools" %s.' % (bzrlib.__version__, __version__))
52
54
        # Not using BzrNewError, because it may not exist.
53
55
        raise Exception, ('Version mismatch', version_info)
54
56
    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
        warning('Plugin "Bzrtools" is not up to date with installed Bazaar'
 
58
                ' version %s.\n'
 
59
                ' There should be a newer version of Bzrtools available, e.g.'
 
60
                ' %i.%i.'
57
61
                % (bzrlib.__version__, bzrlib_version[0], bzrlib_version[1]))
58
62
        if bzrlib_version != desired_plus:
59
63
            raise Exception, 'Version mismatch'
61
65
 
62
66
check_bzrlib_version(version_info[:2])
63
67
 
 
68
from bzrlib.lazy_import import lazy_import
 
69
lazy_import(globals(), """
 
70
from bzrlib import help
 
71
import shelf
 
72
""")
64
73
 
65
74
from errors import CommandError, NoPyBaz
66
75
from patchsource import BzrPatchSource
67
 
from shelf import Shelf
68
76
import sys
69
77
import os.path
70
78
 
72
80
import bzrlib.commands
73
81
from bzrlib.commands import get_cmd_object
74
82
from bzrlib.errors import BzrCommandError
75
 
from bzrlib.help import command_usage
76
83
import bzrlib.ignores
77
84
from bzrlib.option import Option
78
 
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__), 
 
85
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
79
86
                                                 "external")))
80
87
 
81
88
import show_paths
98
105
 
99
106
    To check what clean-tree will do, use --dry-run.
100
107
    """
101
 
    takes_options = [Option('ignored', help='delete all ignored files.'), 
 
108
    takes_options = [Option('ignored', help='delete all ignored files.'),
102
109
                     Option('detritus', help='delete conflict files, merge'
103
 
                            ' backups, and failed selftest dirs.'), 
104
 
                     Option('unknown', 
 
110
                            ' backups, and failed selftest dirs.'),
 
111
                     Option('unknown',
105
112
                            help='delete files unknown to bzr.  (default)'),
106
113
                     Option('dry-run', help='show files to delete instead of'
107
114
                            ' deleting them.')]
125
132
    with the last 5 characters of their revision identifier are used instead.
126
133
 
127
134
    The value starting with d is "(maximum) distance from the null revision".
128
 
    
 
135
 
129
136
    If --merge-branch is specified, the two branches are compared and a merge
130
137
    base is selected.
131
 
    
 
138
 
132
139
    Legend:
133
140
    white    normal revision
134
141
    yellow   THIS  history
149
156
    be disabled with --no-antialias.
150
157
    """
151
158
    takes_args = ['branch', 'file']
152
 
    takes_options = [Option('no-collapse', help="Do not skip simple nodes"), 
 
159
    takes_options = [Option('no-collapse', help="Do not skip simple nodes"),
153
160
                     Option('no-antialias',
154
 
                     help="Do not use rsvg to produce antialiased output"), 
155
 
                     Option('merge-branch', type=str, 
156
 
                     help="Use this branch to calcuate a merge base"), 
157
 
                     Option('cluster', help="Use clustered output.")]
 
161
                     help="Do not use rsvg to produce antialiased output"),
 
162
                     Option('merge-branch', type=str,
 
163
                     help="Use this branch to calcuate a merge base"),
 
164
                     Option('cluster', help="Use clustered output."),
 
165
                     Option('max-distance', help="Show no nodes farther than this",
 
166
                            type=int)]
158
167
    def run(self, branch, file, no_collapse=False, no_antialias=False,
159
 
        merge_branch=None, cluster=False):
 
168
        merge_branch=None, cluster=False, max_distance=None):
160
169
        import graph
161
170
        if cluster:
162
171
            ranking = "cluster"
163
172
        else:
164
173
            ranking = "forced"
165
 
        graph.write_ancestry_file(branch, file, not no_collapse, 
166
 
                                  not no_antialias, merge_branch, ranking)
 
174
        graph.write_ancestry_file(branch, file, not no_collapse,
 
175
                                  not no_antialias, merge_branch, ranking,
 
176
                                  max_distance=max_distance)
167
177
 
168
178
 
169
179
class cmd_fetch_ghosts(bzrlib.commands.Command):
185
195
    """Apply a named patch to the current tree.
186
196
    """
187
197
    takes_args = ['filename?']
188
 
    takes_options = [Option('strip', type=int, help=strip_help)]
189
 
    def run(self, filename=None, strip=None):
 
198
    takes_options = [Option('strip', type=int, help=strip_help),
 
199
                     Option('silent', help='Suppress chatter')]
 
200
    def run(self, filename=None, strip=None, silent=False):
190
201
        from patch import patch
191
202
        from bzrlib.workingtree import WorkingTree
192
203
        wt = WorkingTree.open_containing('.')[0]
193
204
        if strip is None:
194
205
            strip = 0
195
 
        return patch(wt, filename, strip)
 
206
        return patch(wt, filename, strip, silent)
196
207
 
197
208
 
198
209
class cmd_shelve(bzrlib.commands.Command):
227
238
 
228
239
    takes_args = ['file*']
229
240
    takes_options = ['message', 'revision',
230
 
            Option('all', help='Shelve all changes without prompting'), 
 
241
            Option('all', help='Shelve all changes without prompting'),
231
242
            Option('no-color', help='Never display changes in color')]
232
243
 
233
244
    def run(self, all=False, file_list=None, message=None, revision=None,
240
251
                                  "parameter.")
241
252
 
242
253
        source = BzrPatchSource(revision, file_list)
243
 
        s = Shelf(source.base)
 
254
        s = shelf.Shelf(source.base)
244
255
        s.shelve(source, all, message, no_color)
245
256
        return 0
246
257
 
268
279
    aliases = ['switch']
269
280
    takes_args = ['othershelf']
270
281
    def run(self, othershelf):
271
 
        s = Shelf(self.shelf.base, othershelf)
 
282
        s = shelf.Shelf(self.shelf.base, othershelf)
272
283
        s.make_default()
273
284
 
274
285
 
301
312
            args_list = []
302
313
        cmd = self._get_cmd_object(subcommand)
303
314
        source = BzrPatchSource()
304
 
        s = Shelf(source.base)
 
315
        s = shelf.Shelf(source.base)
305
316
        cmd.shelf = s
 
317
 
 
318
        if args_list is None:
 
319
            args_list = []
306
320
        return cmd.run_argv_aliases(args_list)
307
321
 
308
322
    def _get_cmd_object(self, cmd_name):
325
339
        cmd_obj = cmd_class()
326
340
        indent = 2 * ' '
327
341
 
328
 
        usage = command_usage(cmd_obj)
 
342
        usage = cmd_obj._usage()
329
343
        usage = usage.replace('bzr shelf-', '')
330
344
        text.append('%s%s\n' % (indent, usage))
331
345
 
366
380
    takes_args = ['patch?']
367
381
    def run(self, patch=None, all=False, force=False, no_color=False):
368
382
        source = BzrPatchSource()
369
 
        s = Shelf(source.base)
 
383
        s = shelf.Shelf(source.base)
370
384
        s.unshelve(source, patch, all, force, no_color)
371
385
        return 0
372
386
 
404
418
    """
405
419
    takes_args = ["branch?"]
406
420
    def run(self, branch=None):
407
 
        from branchhistory import branch_history 
 
421
        from branchhistory import branch_history
408
422
        return branch_history(branch)
409
423
 
410
424
 
430
444
class cmd_cbranch(bzrlib.commands.Command):
431
445
    """
432
446
    Create a new checkout, associated with a new repository branch.
433
 
    
434
 
    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
436
 
    with the same name and relative path as the checkout you request.
437
 
 
438
 
    The locations.conf parameter is "cbranch_root".  So if you want 
439
 
    cbranch operations in /home/jrandom/bigproject to produce branches in 
440
 
    /home/jrandom/bigproject/repository, you'd add this:
441
 
 
442
 
    [/home/jrandom/bigproject]
443
 
    cbranch_root = /home/jrandom/bigproject/repository
444
 
 
445
 
    Note that if "/home/jrandom/bigproject/repository" isn't a repository,
446
 
    standalone branches will be produced.  Standalone branches will also
447
 
    be produced if the source branch is in 0.7 format (or earlier).
 
447
 
 
448
    When you cbranch, bzr looks up a target location in locations.conf, and
 
449
    creates the branch there.
 
450
 
 
451
    In your locations.conf, add the following lines:
 
452
    [/working_directory_root]
 
453
    cbranch_target = /branch_root
 
454
    cbranch_target:policy = appendpath
 
455
 
 
456
    This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
 
457
    working directory root, the branch will be created in
 
458
    "/branch_root/foo/baz"
 
459
 
 
460
    NOTE: cbranch also supports "cbranch_root", but that behaviour is
 
461
    deprecated.
448
462
    """
449
 
    takes_options = [Option("lightweight", 
 
463
    takes_options = [Option("lightweight",
450
464
                            help="Create a lightweight checkout"), 'revision']
451
465
    takes_args = ["source", "target?"]
452
466
    def run(self, source, target=None, lightweight=False, revision=None):
453
467
        from cbranch import cbranch
454
 
        return cbranch(source, target, lightweight=lightweight, 
 
468
        return cbranch(source, target, lightweight=lightweight,
455
469
                       revision=revision)
456
470
 
457
471
 
465
479
 
466
480
class cmd_multi_pull(bzrlib.commands.Command):
467
481
    """Pull all the branches under a location, e.g. a repository.
468
 
    
 
482
 
469
483
    Both branches present in the directory and the branches of checkouts are
470
484
    pulled.
471
485
    """
522
536
 
523
537
 
524
538
class cmd_import(bzrlib.commands.Command):
525
 
    """Import sources from a tarball
526
 
    
527
 
    This command will import a tarball into a bzr tree, replacing any versioned
528
 
    files already present.  If a directory is specified, it is used as the
529
 
    target.  If the directory does not exist, or is not versioned, it is
530
 
    created.
 
539
    """Import sources from a directory, tarball or zip file
 
540
 
 
541
    This command will import a directory, tarball or zip file into a bzr
 
542
    tree, replacing any versioned files already present.  If a directory is
 
543
    specified, it is used as the target.  If the directory does not exist, or
 
544
    is not versioned, it is created.
531
545
 
532
546
    Tarballs may be gzip or bzip2 compressed.  This is autodetected.
533
547
 
534
 
    If the tarball has a single root directory, that directory is stripped
535
 
    when extracting the tarball.
 
548
    If the tarball or zip has a single root directory, that directory is
 
549
    stripped when extracting the tarball.  This is not done for directories.
536
550
    """
537
 
    
 
551
 
538
552
    takes_args = ['source', 'tree?']
539
553
    def run(self, source, tree=None):
540
554
        from upstream_import import do_import
544
558
class cmd_cdiff(bzrlib.commands.Command):
545
559
    """A color version of bzr's diff"""
546
560
    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):
 
561
 
 
562
    def _takes_options(self):
 
563
        options = list(get_cmd_object('diff').takes_options)
 
564
        options.append(Option('check-style',
 
565
            help='Warn if trailing whitespace or spurious changes have been'
 
566
                 ' added.'))
 
567
        return options
 
568
 
 
569
    takes_options = property(_takes_options)
 
570
 
 
571
    def run(self, check_style=False, *args, **kwargs):
549
572
        from colordiff import colordiff
550
 
        colordiff(*args, **kwargs)
 
573
        colordiff(check_style, *args, **kwargs)
551
574
 
552
575
 
553
576
class cmd_baz_import(bzrlib.commands.Command):
555
578
 
556
579
    This command should be used on local archives (or mirrors) only.  It is
557
580
    quite slow on remote archives.
558
 
    
559
 
    reuse_history allows you to specify any previous imports you 
 
581
 
 
582
    reuse_history allows you to specify any previous imports you
560
583
    have done of different archives, which this archive has branches
561
 
    tagged from. This will dramatically reduce the time to convert 
 
584
    tagged from. This will dramatically reduce the time to convert
562
585
    the archive as it will not have to convert the history already
563
586
    converted in that other branch.
564
587
 
578
601
    takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
579
602
    takes_options = ['verbose', Option('prefixes', type=str,
580
603
                     help="Prefixes of branches to import, colon-separated"),
581
 
                     Option('encoding', type=str, 
 
604
                     Option('encoding', type=str,
582
605
                     help='Force encoding to specified value.  See WARNING.')]
583
606
 
584
607
    def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
604
627
    """
605
628
    takes_args = ['to_location', 'from_branch?', 'reuse_history*']
606
629
    takes_options = ['verbose', Option('max-count', type=int),
607
 
                     Option('encoding', type=str, 
 
630
                     Option('encoding', type=str,
608
631
                     help='Force encoding to specified value.  See WARNING.')]
609
632
 
610
633
    def run(self, to_location, from_branch=None, fast=False, max_count=None,
613
636
        from errors import NoPyBaz
614
637
        try:
615
638
            import baz_import
616
 
            baz_import.baz_import_branch(to_location, from_branch, fast, 
 
639
            baz_import.baz_import_branch(to_location, from_branch, fast,
617
640
                                         max_count, verbose, encoding, dry_run,
618
641
                                         reuse_history_list)
619
642
        except NoPyBaz:
623
646
class cmd_rspush(bzrlib.commands.Command):
624
647
    """Upload this branch to another location using rsync.
625
648
 
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. 
 
649
    If no location is specified, the last-used location will be used.  To
 
650
    prevent dirty trees from being uploaded, rspush will error out if there are
 
651
    unknown files or local changes.  It will also error out if the upstream
 
652
    directory is non-empty and not an earlier version of the branch.
630
653
    """
631
654
    takes_args = ['location?']
632
655
    takes_options = [Option('overwrite', help='Ignore differences between'
638
661
        from bzrlib import workingtree
639
662
        import bzrtools
640
663
        cur_branch = workingtree.WorkingTree.open_containing(".")[0]
641
 
        bzrtools.rspush(cur_branch, location, overwrite=overwrite, 
 
664
        bzrtools.rspush(cur_branch, location, overwrite=overwrite,
642
665
                      working_tree=not no_tree)
643
666
 
644
667
 
658
681
            cmd_branches,
659
682
            cmd_branch_history,
660
683
            cmd_branch_mark,
661
 
            cmd_cbranch,  
 
684
            cmd_cbranch,
662
685
            cmd_cdiff,
663
686
            cmd_clean_tree,
664
687
            cmd_fetch_ghosts,
667
690
            cmd_multi_pull,
668
691
            cmd_patch,
669
692
            cmd_rspush,
670
 
            cmd_shelf, 
 
693
            cmd_shelf,
671
694
            cmd_shell,
672
 
            cmd_shelve, 
 
695
            cmd_shelve,
673
696
            cmd_switch,
674
 
            cmd_unshelve, 
675
 
            cmd_zap,            
 
697
            cmd_unshelve,
 
698
            cmd_zap,
676
699
            ]
677
700
 
678
701
 
688
711
    from unittest import TestSuite
689
712
    import bzrtools
690
713
    import tests.clean_tree
691
 
    import upstream_import
 
714
    import tests.is_clean
 
715
    import tests.upstream_import
692
716
    import zap
693
717
    import tests.blackbox
694
718
    import tests.shelf_tests
703
727
    result.addTest(tests.test_suite())
704
728
    result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))
705
729
    result.addTest(tests.blackbox.test_suite())
706
 
    result.addTest(upstream_import.test_suite())
 
730
    result.addTest(tests.upstream_import.test_suite())
707
731
    result.addTest(zap.test_suite())
 
732
    result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))
708
733
    return result