~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Charlie Shepherd
  • Date: 2007-04-04 18:12:00 UTC
  • mto: This revision was merged to the branch mainline in revision 538.
  • Revision ID: masterdriverz@gentoo.org-20070404181200-wqiwytdor9srux2v
Remove all trailing whitespace

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.15.0'
 
26
__version__ = '0.16.0'
27
27
 
28
28
 
29
29
version_info = tuple(int(n) for n in __version__.split('.'))
47
47
        # get the message out any way we can
48
48
        from warnings import warn as warning
49
49
    if bzrlib_version < desired:
50
 
        warning('Installed bzr version %s is too old to be used with bzrtools'
51
 
                ' %s.' % (bzrlib.__version__, __version__))
 
50
        warning('Installed Bazaar version %s is too old to be used with'
 
51
                ' plugin \n'
 
52
                '"Bzrtools" %s.' % (bzrlib.__version__, __version__))
52
53
        # Not using BzrNewError, because it may not exist.
53
54
        raise Exception, ('Version mismatch', version_info)
54
55
    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.' 
 
56
        warning('Plugin "Bzrtools" is not up to date with installed Bazaar'
 
57
                ' version %s.\n'
 
58
                ' There should be a newer version of Bzrtools available, e.g.'
 
59
                ' %i.%i.'
57
60
                % (bzrlib.__version__, bzrlib_version[0], bzrlib_version[1]))
58
61
        if bzrlib_version != desired_plus:
59
62
            raise Exception, 'Version mismatch'
61
64
 
62
65
check_bzrlib_version(version_info[:2])
63
66
 
 
67
from bzrlib.lazy_import import lazy_import
 
68
lazy_import(globals(), """
 
69
from bzrlib import help
 
70
import shelf
 
71
""")
64
72
 
65
73
from errors import CommandError, NoPyBaz
66
74
from patchsource import BzrPatchSource
67
 
from shelf import Shelf
68
75
import sys
69
76
import os.path
70
77
 
72
79
import bzrlib.commands
73
80
from bzrlib.commands import get_cmd_object
74
81
from bzrlib.errors import BzrCommandError
75
 
from bzrlib.help import command_usage
76
82
import bzrlib.ignores
77
83
from bzrlib.option import Option
78
 
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__), 
 
84
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__),
79
85
                                                 "external")))
80
86
 
81
87
import show_paths
98
104
 
99
105
    To check what clean-tree will do, use --dry-run.
100
106
    """
101
 
    takes_options = [Option('ignored', help='delete all ignored files.'), 
 
107
    takes_options = [Option('ignored', help='delete all ignored files.'),
102
108
                     Option('detritus', help='delete conflict files, merge'
103
 
                            ' backups, and failed selftest dirs.'), 
104
 
                     Option('unknown', 
 
109
                            ' backups, and failed selftest dirs.'),
 
110
                     Option('unknown',
105
111
                            help='delete files unknown to bzr.  (default)'),
106
112
                     Option('dry-run', help='show files to delete instead of'
107
113
                            ' deleting them.')]
109
115
        from clean_tree import clean_tree
110
116
        if not (unknown or ignored or detritus):
111
117
            unknown = True
112
 
        clean_tree('.', unknown=unknown, ignored=ignored, detritus=detritus, 
113
 
                   dry_run=dry_run)
 
118
        clean_tree('.', out=self.outf, unknown=unknown, ignored=ignored,
 
119
                   detritus=detritus, dry_run=dry_run)
114
120
 
115
121
 
116
122
class cmd_graph_ancestry(bzrlib.commands.Command):
117
123
    """Produce ancestry graphs using dot.
118
 
    
 
124
 
119
125
    Output format is detected according to file extension.  Some of the more
120
126
    common output formats are html, png, gif, svg, ps.  An extension of '.dot'
121
127
    will cause a dot graph file to be produced.  HTML output has mouseovers
125
131
    with the last 5 characters of their revision identifier are used instead.
126
132
 
127
133
    The value starting with d is "(maximum) distance from the null revision".
128
 
    
 
134
 
129
135
    If --merge-branch is specified, the two branches are compared and a merge
130
136
    base is selected.
131
 
    
 
137
 
132
138
    Legend:
133
139
    white    normal revision
134
140
    yellow   THIS  history
149
155
    be disabled with --no-antialias.
150
156
    """
151
157
    takes_args = ['branch', 'file']
152
 
    takes_options = [Option('no-collapse', help="Do not skip simple nodes"), 
 
158
    takes_options = [Option('no-collapse', help="Do not skip simple nodes"),
153
159
                     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"), 
 
160
                     help="Do not use rsvg to produce antialiased output"),
 
161
                     Option('merge-branch', type=str,
 
162
                     help="Use this branch to calcuate a merge base"),
157
163
                     Option('cluster', help="Use clustered output."),
158
164
                     Option('max-distance', help="Show no nodes farther than this",
159
165
                            type=int)]
164
170
            ranking = "cluster"
165
171
        else:
166
172
            ranking = "forced"
167
 
        graph.write_ancestry_file(branch, file, not no_collapse, 
168
 
                                  not no_antialias, merge_branch, ranking, 
 
173
        graph.write_ancestry_file(branch, file, not no_collapse,
 
174
                                  not no_antialias, merge_branch, ranking,
169
175
                                  max_distance=max_distance)
170
176
 
171
177
 
231
237
 
232
238
    takes_args = ['file*']
233
239
    takes_options = ['message', 'revision',
234
 
            Option('all', help='Shelve all changes without prompting'), 
 
240
            Option('all', help='Shelve all changes without prompting'),
235
241
            Option('no-color', help='Never display changes in color')]
236
242
 
237
243
    def run(self, all=False, file_list=None, message=None, revision=None,
244
250
                                  "parameter.")
245
251
 
246
252
        source = BzrPatchSource(revision, file_list)
247
 
        s = Shelf(source.base)
 
253
        s = shelf.Shelf(source.base)
248
254
        s.shelve(source, all, message, no_color)
249
255
        return 0
250
256
 
272
278
    aliases = ['switch']
273
279
    takes_args = ['othershelf']
274
280
    def run(self, othershelf):
275
 
        s = Shelf(self.shelf.base, othershelf)
 
281
        s = shelf.Shelf(self.shelf.base, othershelf)
276
282
        s.make_default()
277
283
 
278
284
 
305
311
            args_list = []
306
312
        cmd = self._get_cmd_object(subcommand)
307
313
        source = BzrPatchSource()
308
 
        s = Shelf(source.base)
 
314
        s = shelf.Shelf(source.base)
309
315
        cmd.shelf = s
 
316
 
 
317
        if args_list is None:
 
318
            args_list = []
310
319
        return cmd.run_argv_aliases(args_list)
311
320
 
312
321
    def _get_cmd_object(self, cmd_name):
329
338
        cmd_obj = cmd_class()
330
339
        indent = 2 * ' '
331
340
 
332
 
        usage = command_usage(cmd_obj)
 
341
        usage = help.command_usage(cmd_obj)
333
342
        usage = usage.replace('bzr shelf-', '')
334
343
        text.append('%s%s\n' % (indent, usage))
335
344
 
370
379
    takes_args = ['patch?']
371
380
    def run(self, patch=None, all=False, force=False, no_color=False):
372
381
        source = BzrPatchSource()
373
 
        s = Shelf(source.base)
 
382
        s = shelf.Shelf(source.base)
374
383
        s.unshelve(source, patch, all, force, no_color)
375
384
        return 0
376
385
 
408
417
    """
409
418
    takes_args = ["branch?"]
410
419
    def run(self, branch=None):
411
 
        from branchhistory import branch_history 
 
420
        from branchhistory import branch_history
412
421
        return branch_history(branch)
413
422
 
414
423
 
434
443
class cmd_cbranch(bzrlib.commands.Command):
435
444
    """
436
445
    Create a new checkout, associated with a new repository branch.
437
 
    
 
446
 
438
447
    When you cbranch, bzr looks up a target location in locations.conf, and
439
448
    creates the branch there.
440
449
 
444
453
    cbranch_target:policy = appendpath
445
454
 
446
455
    This will mean that if you run "bzr cbranch foo/bar foo/baz" in the
447
 
    working directory root, the branch will be created in 
 
456
    working directory root, the branch will be created in
448
457
    "/branch_root/foo/baz"
449
458
 
450
459
    NOTE: cbranch also supports "cbranch_root", but that behaviour is
451
460
    deprecated.
452
461
    """
453
 
    takes_options = [Option("lightweight", 
 
462
    takes_options = [Option("lightweight",
454
463
                            help="Create a lightweight checkout"), 'revision']
455
464
    takes_args = ["source", "target?"]
456
465
    def run(self, source, target=None, lightweight=False, revision=None):
457
466
        from cbranch import cbranch
458
 
        return cbranch(source, target, lightweight=lightweight, 
 
467
        return cbranch(source, target, lightweight=lightweight,
459
468
                       revision=revision)
460
469
 
461
470
 
469
478
 
470
479
class cmd_multi_pull(bzrlib.commands.Command):
471
480
    """Pull all the branches under a location, e.g. a repository.
472
 
    
 
481
 
473
482
    Both branches present in the directory and the branches of checkouts are
474
483
    pulled.
475
484
    """
527
536
 
528
537
class cmd_import(bzrlib.commands.Command):
529
538
    """Import sources from a directory, tarball or zip file
530
 
    
 
539
 
531
540
    This command will import a directory, tarball or zip file into a bzr
532
541
    tree, replacing any versioned files already present.  If a directory is
533
542
    specified, it is used as the target.  If the directory does not exist, or
538
547
    If the tarball or zip has a single root directory, that directory is
539
548
    stripped when extracting the tarball.  This is not done for directories.
540
549
    """
541
 
    
 
550
 
542
551
    takes_args = ['source', 'tree?']
543
552
    def run(self, source, tree=None):
544
553
        from upstream_import import do_import
568
577
 
569
578
    This command should be used on local archives (or mirrors) only.  It is
570
579
    quite slow on remote archives.
571
 
    
572
 
    reuse_history allows you to specify any previous imports you 
 
580
 
 
581
    reuse_history allows you to specify any previous imports you
573
582
    have done of different archives, which this archive has branches
574
 
    tagged from. This will dramatically reduce the time to convert 
 
583
    tagged from. This will dramatically reduce the time to convert
575
584
    the archive as it will not have to convert the history already
576
585
    converted in that other branch.
577
586
 
591
600
    takes_args = ['to_root_dir', 'from_archive', 'reuse_history*']
592
601
    takes_options = ['verbose', Option('prefixes', type=str,
593
602
                     help="Prefixes of branches to import, colon-separated"),
594
 
                     Option('encoding', type=str, 
 
603
                     Option('encoding', type=str,
595
604
                     help='Force encoding to specified value.  See WARNING.')]
596
605
 
597
606
    def run(self, to_root_dir, from_archive, encoding=None, verbose=False,
617
626
    """
618
627
    takes_args = ['to_location', 'from_branch?', 'reuse_history*']
619
628
    takes_options = ['verbose', Option('max-count', type=int),
620
 
                     Option('encoding', type=str, 
 
629
                     Option('encoding', type=str,
621
630
                     help='Force encoding to specified value.  See WARNING.')]
622
631
 
623
632
    def run(self, to_location, from_branch=None, fast=False, max_count=None,
626
635
        from errors import NoPyBaz
627
636
        try:
628
637
            import baz_import
629
 
            baz_import.baz_import_branch(to_location, from_branch, fast, 
 
638
            baz_import.baz_import_branch(to_location, from_branch, fast,
630
639
                                         max_count, verbose, encoding, dry_run,
631
640
                                         reuse_history_list)
632
641
        except NoPyBaz:
636
645
class cmd_rspush(bzrlib.commands.Command):
637
646
    """Upload this branch to another location using rsync.
638
647
 
639
 
    If no location is specified, the last-used location will be used.  To 
640
 
    prevent dirty trees from being uploaded, rspush will error out if there are 
641
 
    unknown files or local changes.  It will also error out if the upstream 
642
 
    directory is non-empty and not an earlier version of the branch. 
 
648
    If no location is specified, the last-used location will be used.  To
 
649
    prevent dirty trees from being uploaded, rspush will error out if there are
 
650
    unknown files or local changes.  It will also error out if the upstream
 
651
    directory is non-empty and not an earlier version of the branch.
643
652
    """
644
653
    takes_args = ['location?']
645
654
    takes_options = [Option('overwrite', help='Ignore differences between'
651
660
        from bzrlib import workingtree
652
661
        import bzrtools
653
662
        cur_branch = workingtree.WorkingTree.open_containing(".")[0]
654
 
        bzrtools.rspush(cur_branch, location, overwrite=overwrite, 
 
663
        bzrtools.rspush(cur_branch, location, overwrite=overwrite,
655
664
                      working_tree=not no_tree)
656
665
 
657
666
 
671
680
            cmd_branches,
672
681
            cmd_branch_history,
673
682
            cmd_branch_mark,
674
 
            cmd_cbranch,  
 
683
            cmd_cbranch,
675
684
            cmd_cdiff,
676
685
            cmd_clean_tree,
677
686
            cmd_fetch_ghosts,
680
689
            cmd_multi_pull,
681
690
            cmd_patch,
682
691
            cmd_rspush,
683
 
            cmd_shelf, 
 
692
            cmd_shelf,
684
693
            cmd_shell,
685
 
            cmd_shelve, 
 
694
            cmd_shelve,
686
695
            cmd_switch,
687
 
            cmd_unshelve, 
688
 
            cmd_zap,            
 
696
            cmd_unshelve,
 
697
            cmd_zap,
689
698
            ]
690
699
 
691
700
 
701
710
    from unittest import TestSuite
702
711
    import bzrtools
703
712
    import tests.clean_tree
 
713
    import tests.is_clean
704
714
    import tests.upstream_import
705
715
    import zap
706
716
    import tests.blackbox
718
728
    result.addTest(tests.blackbox.test_suite())
719
729
    result.addTest(tests.upstream_import.test_suite())
720
730
    result.addTest(zap.test_suite())
 
731
    result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))
721
732
    return result