~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bzrdir.py

  • Committer: Vincent Ladeuil
  • Date: 2010-06-17 16:54:26 UTC
  • mto: This revision was merged to the branch mainline in revision 5306.
  • Revision ID: v.ladeuil+lp@free.fr-20100617165426-741tbmgwi62a9zub
Pass BZR_PLUGINS_AT and BZR_DISABLE_PLINGS to the subprocess fpr test_import_tariff

Show diffs side-by-side

added added

removed removed

Lines of Context:
34
34
from bzrlib.lazy_import import lazy_import
35
35
lazy_import(globals(), """
36
36
from stat import S_ISDIR
 
37
import textwrap
37
38
 
38
39
import bzrlib
39
40
from bzrlib import (
40
41
    branch,
41
42
    config,
42
 
    controldir,
43
43
    errors,
44
44
    graph,
45
45
    lockable_files,
46
46
    lockdir,
47
47
    osutils,
48
 
    pyutils,
49
48
    remote,
50
49
    repository,
51
50
    revision as _mod_revision,
87
86
    registry,
88
87
    symbol_versioning,
89
88
    )
90
 
from bzrlib.symbol_versioning import (
91
 
    deprecated_in,
92
 
    deprecated_method,
93
 
    )
94
 
 
95
 
 
96
 
class BzrDir(controldir.ControlDir):
 
89
    
 
90
    
 
91
class ControlComponent(object):
 
92
    """Abstract base class for control directory components.
 
93
    
 
94
    This provides interfaces that are common across bzrdirs, 
 
95
    repositories, branches, and workingtree control directories.
 
96
    
 
97
    They all expose two urls and transports: the *user* URL is the 
 
98
    one that stops above the control directory (eg .bzr) and that 
 
99
    should normally be used in messages, and the *control* URL is
 
100
    under that in eg .bzr/checkout and is used to read the control
 
101
    files.
 
102
    
 
103
    This can be used as a mixin and is intended to fit with 
 
104
    foreign formats.
 
105
    """
 
106
    
 
107
    @property
 
108
    def control_transport(self):
 
109
        raise NotImplementedError
 
110
   
 
111
    @property
 
112
    def control_url(self):
 
113
        return self.control_transport.base
 
114
    
 
115
    @property
 
116
    def user_transport(self):
 
117
        raise NotImplementedError
 
118
        
 
119
    @property
 
120
    def user_url(self):
 
121
        return self.user_transport.base
 
122
    
 
123
 
 
124
class BzrDir(ControlComponent):
97
125
    """A .bzr control diretory.
98
126
 
99
127
    BzrDir instances let you create or open any of the things that can be
130
158
                    return
131
159
        thing_to_unlock.break_lock()
132
160
 
 
161
    def can_convert_format(self):
 
162
        """Return true if this bzrdir is one whose format we can convert from."""
 
163
        return True
 
164
 
133
165
    def check_conversion_target(self, target_format):
134
166
        """Check that a bzrdir as a whole can be converted to a new format."""
135
167
        # The only current restriction is that the repository content can be 
170
202
                format.get_format_description(),
171
203
                basedir)
172
204
 
 
205
    def clone(self, url, revision_id=None, force_new_repo=False,
 
206
              preserve_stacking=False):
 
207
        """Clone this bzrdir and its contents to url verbatim.
 
208
 
 
209
        :param url: The url create the clone at.  If url's last component does
 
210
            not exist, it will be created.
 
211
        :param revision_id: The tip revision-id to use for any branch or
 
212
            working tree.  If not None, then the clone operation may tune
 
213
            itself to download less data.
 
214
        :param force_new_repo: Do not use a shared repository for the target
 
215
                               even if one is available.
 
216
        :param preserve_stacking: When cloning a stacked branch, stack the
 
217
            new branch on top of the other branch's stacked-on branch.
 
218
        """
 
219
        return self.clone_on_transport(get_transport(url),
 
220
                                       revision_id=revision_id,
 
221
                                       force_new_repo=force_new_repo,
 
222
                                       preserve_stacking=preserve_stacking)
 
223
 
173
224
    def clone_on_transport(self, transport, revision_id=None,
174
225
        force_new_repo=False, preserve_stacking=False, stacked_on=None,
175
 
        create_prefix=False, use_existing_dir=True, no_tree=False):
 
226
        create_prefix=False, use_existing_dir=True):
176
227
        """Clone this bzrdir and its contents to transport verbatim.
177
228
 
178
229
        :param transport: The transport for the location to produce the clone
190
241
        """
191
242
        # Overview: put together a broad description of what we want to end up
192
243
        # with; then make as few api calls as possible to do it.
193
 
 
 
244
        
194
245
        # We may want to create a repo/branch/tree, if we do so what format
195
246
        # would we want for each:
196
247
        require_stacking = (stacked_on is not None)
197
248
        format = self.cloning_metadir(require_stacking)
198
 
 
 
249
        
199
250
        # Figure out what objects we want:
200
251
        try:
201
252
            local_repo = self.find_repository()
220
271
        # we should look up the policy needs first, or just use it as a hint,
221
272
        # or something.
222
273
        if local_repo:
223
 
            make_working_trees = local_repo.make_working_trees() and not no_tree
 
274
            make_working_trees = local_repo.make_working_trees()
224
275
            want_shared = local_repo.is_shared()
225
276
            repo_format_name = format.repository_format.network_name()
226
277
        else:
278
329
        t = get_transport(url)
279
330
        t.ensure_base()
280
331
 
 
332
    @classmethod
 
333
    def create(cls, base, format=None, possible_transports=None):
 
334
        """Create a new BzrDir at the url 'base'.
 
335
 
 
336
        :param format: If supplied, the format of branch to create.  If not
 
337
            supplied, the default is used.
 
338
        :param possible_transports: If supplied, a list of transports that
 
339
            can be reused to share a remote connection.
 
340
        """
 
341
        if cls is not BzrDir:
 
342
            raise AssertionError("BzrDir.create always creates the default"
 
343
                " format, not one of %r" % cls)
 
344
        t = get_transport(base, possible_transports)
 
345
        t.ensure_base()
 
346
        if format is None:
 
347
            format = BzrDirFormat.get_default_format()
 
348
        return format.initialize_on_transport(t)
 
349
 
281
350
    @staticmethod
282
351
    def find_bzrdirs(transport, evaluate=None, list_current=None):
283
352
        """Find bzrdirs recursively from current location.
306
375
            recurse = True
307
376
            try:
308
377
                bzrdir = BzrDir.open_from_transport(current_transport)
309
 
            except (errors.NotBranchError, errors.PermissionDenied):
 
378
            except errors.NotBranchError:
310
379
                pass
311
380
            else:
312
381
                recurse, value = evaluate(bzrdir)
313
382
                yield value
314
383
            try:
315
384
                subdirs = list_current(current_transport)
316
 
            except (errors.NoSuchFile, errors.PermissionDenied):
 
385
            except errors.NoSuchFile:
317
386
                continue
318
387
            if recurse:
319
388
                for subdir in sorted(subdirs, reverse=True):
320
389
                    pending.append(current_transport.clone(subdir))
321
390
 
 
391
    def list_branches(self):
 
392
        """Return a sequence of all branches local to this control directory.
 
393
 
 
394
        """
 
395
        try:
 
396
            return [self.open_branch()]
 
397
        except errors.NotBranchError:
 
398
            return []
 
399
 
322
400
    @staticmethod
323
401
    def find_branches(transport):
324
402
        """Find all branches under a transport.
347
425
                ret.extend(branches)
348
426
        return ret
349
427
 
 
428
    def destroy_repository(self):
 
429
        """Destroy the repository in this BzrDir"""
 
430
        raise NotImplementedError(self.destroy_repository)
 
431
 
 
432
    def create_branch(self, name=None):
 
433
        """Create a branch in this BzrDir.
 
434
 
 
435
        :param name: Name of the colocated branch to create, None for
 
436
            the default branch.
 
437
 
 
438
        The bzrdir's format will control what branch format is created.
 
439
        For more control see BranchFormatXX.create(a_bzrdir).
 
440
        """
 
441
        raise NotImplementedError(self.create_branch)
 
442
 
 
443
    def destroy_branch(self, name=None):
 
444
        """Destroy a branch in this BzrDir.
 
445
        
 
446
        :param name: Name of the branch to destroy, None for the default 
 
447
            branch.
 
448
        """
 
449
        raise NotImplementedError(self.destroy_branch)
 
450
 
350
451
    @staticmethod
351
452
    def create_branch_and_repo(base, force_new_repo=False, format=None):
352
453
        """Create a new BzrDir, Branch and Repository at the url 'base'.
501
602
                                               format=format).bzrdir
502
603
        return bzrdir.create_workingtree()
503
604
 
504
 
    @deprecated_method(deprecated_in((2, 3, 0)))
505
 
    def generate_backup_name(self, base):
506
 
        return self._available_backup_name(base)
507
 
 
508
 
    def _available_backup_name(self, base):
509
 
        """Find a non-existing backup file name based on base.
510
 
 
511
 
        See bzrlib.osutils.available_backup_name about race conditions.
 
605
    def create_workingtree(self, revision_id=None, from_branch=None,
 
606
        accelerator_tree=None, hardlink=False):
 
607
        """Create a working tree at this BzrDir.
 
608
 
 
609
        :param revision_id: create it as of this revision id.
 
610
        :param from_branch: override bzrdir branch (for lightweight checkouts)
 
611
        :param accelerator_tree: A tree which can be used for retrieving file
 
612
            contents more quickly than the revision tree, i.e. a workingtree.
 
613
            The revision tree will be used for cases where accelerator_tree's
 
614
            content is different.
512
615
        """
513
 
        return osutils.available_backup_name(base, self.root_transport.has)
 
616
        raise NotImplementedError(self.create_workingtree)
514
617
 
515
618
    def backup_bzrdir(self):
516
619
        """Backup this bzr control directory.
517
620
 
518
621
        :return: Tuple with old path name and new path name
519
622
        """
 
623
        def name_gen(base='backup.bzr'):
 
624
            counter = 1
 
625
            name = "%s.~%d~" % (base, counter)
 
626
            while self.root_transport.has(name):
 
627
                counter += 1
 
628
                name = "%s.~%d~" % (base, counter)
 
629
            return name
520
630
 
 
631
        backup_dir=name_gen()
521
632
        pb = ui.ui_factory.nested_progress_bar()
522
633
        try:
 
634
            # FIXME: bug 300001 -- the backup fails if the backup directory
 
635
            # already exists, but it should instead either remove it or make
 
636
            # a new backup directory.
 
637
            #
523
638
            old_path = self.root_transport.abspath('.bzr')
524
 
            backup_dir = self._available_backup_name('backup.bzr')
525
639
            new_path = self.root_transport.abspath(backup_dir)
526
 
            ui.ui_factory.note('making backup of %s\n  to %s'
527
 
                               % (old_path, new_path,))
 
640
            ui.ui_factory.note('making backup of %s\n  to %s' % (old_path, new_path,))
528
641
            self.root_transport.copy_tree('.bzr', backup_dir)
529
642
            return (old_path, new_path)
530
643
        finally:
555
668
                else:
556
669
                    pass
557
670
 
 
671
    def destroy_workingtree(self):
 
672
        """Destroy the working tree at this BzrDir.
 
673
 
 
674
        Formats that do not support this may raise UnsupportedOperation.
 
675
        """
 
676
        raise NotImplementedError(self.destroy_workingtree)
 
677
 
 
678
    def destroy_workingtree_metadata(self):
 
679
        """Destroy the control files for the working tree at this BzrDir.
 
680
 
 
681
        The contents of working tree files are not affected.
 
682
        Formats that do not support this may raise UnsupportedOperation.
 
683
        """
 
684
        raise NotImplementedError(self.destroy_workingtree_metadata)
 
685
 
558
686
    def _find_containing(self, evaluate):
559
687
        """Find something in a containing control directory.
560
688
 
609
737
            raise errors.NoRepositoryPresent(self)
610
738
        return found_repo
611
739
 
 
740
    def get_branch_reference(self, name=None):
 
741
        """Return the referenced URL for the branch in this bzrdir.
 
742
 
 
743
        :param name: Optional colocated branch name
 
744
        :raises NotBranchError: If there is no Branch.
 
745
        :raises NoColocatedBranchSupport: If a branch name was specified
 
746
            but colocated branches are not supported.
 
747
        :return: The URL the branch in this bzrdir references if it is a
 
748
            reference branch, or None for regular branches.
 
749
        """
 
750
        if name is not None:
 
751
            raise errors.NoColocatedBranchSupport(self)
 
752
        return None
 
753
 
 
754
    def get_branch_transport(self, branch_format, name=None):
 
755
        """Get the transport for use by branch format in this BzrDir.
 
756
 
 
757
        Note that bzr dirs that do not support format strings will raise
 
758
        IncompatibleFormat if the branch format they are given has
 
759
        a format string, and vice versa.
 
760
 
 
761
        If branch_format is None, the transport is returned with no
 
762
        checking. If it is not None, then the returned transport is
 
763
        guaranteed to point to an existing directory ready for use.
 
764
        """
 
765
        raise NotImplementedError(self.get_branch_transport)
 
766
 
612
767
    def _find_creation_modes(self):
613
768
        """Determine the appropriate modes for files and directories.
614
769
 
653
808
            self._find_creation_modes()
654
809
        return self._dir_mode
655
810
 
 
811
    def get_repository_transport(self, repository_format):
 
812
        """Get the transport for use by repository format in this BzrDir.
 
813
 
 
814
        Note that bzr dirs that do not support format strings will raise
 
815
        IncompatibleFormat if the repository format they are given has
 
816
        a format string, and vice versa.
 
817
 
 
818
        If repository_format is None, the transport is returned with no
 
819
        checking. If it is not None, then the returned transport is
 
820
        guaranteed to point to an existing directory ready for use.
 
821
        """
 
822
        raise NotImplementedError(self.get_repository_transport)
 
823
 
 
824
    def get_workingtree_transport(self, tree_format):
 
825
        """Get the transport for use by workingtree format in this BzrDir.
 
826
 
 
827
        Note that bzr dirs that do not support format strings will raise
 
828
        IncompatibleFormat if the workingtree format they are given has a
 
829
        format string, and vice versa.
 
830
 
 
831
        If workingtree_format is None, the transport is returned with no
 
832
        checking. If it is not None, then the returned transport is
 
833
        guaranteed to point to an existing directory ready for use.
 
834
        """
 
835
        raise NotImplementedError(self.get_workingtree_transport)
 
836
 
656
837
    def get_config(self):
657
838
        """Get configuration for this BzrDir."""
658
839
        return config.BzrDirConfig(self)
676
857
        self.transport = _transport.clone('.bzr')
677
858
        self.root_transport = _transport
678
859
        self._mode_check_done = False
679
 
 
 
860
        
680
861
    @property 
681
862
    def user_transport(self):
682
863
        return self.root_transport
683
 
 
 
864
        
684
865
    @property
685
866
    def control_transport(self):
686
867
        return self.transport
692
873
 
693
874
        This is true IF and ONLY IF the filename is part of the namespace reserved
694
875
        for bzr control dirs. Currently this is the '.bzr' directory in the root
695
 
        of the root_transport. 
 
876
        of the root_transport. it is expected that plugins will need to extend
 
877
        this in the future - for instance to make bzr talk with svn working
 
878
        trees.
696
879
        """
697
880
        # this might be better on the BzrDirFormat class because it refers to
698
881
        # all the possible bzrdir disk formats.
702
885
        # add new tests for it to the appropriate place.
703
886
        return filename == '.bzr' or filename.startswith('.bzr/')
704
887
 
 
888
    def needs_format_conversion(self, format=None):
 
889
        """Return true if this bzrdir needs convert_format run on it.
 
890
 
 
891
        For instance, if the repository format is out of date but the
 
892
        branch and working tree are not, this should return True.
 
893
 
 
894
        :param format: Optional parameter indicating a specific desired
 
895
                       format we plan to arrive at.
 
896
        """
 
897
        raise NotImplementedError(self.needs_format_conversion)
 
898
 
705
899
    @staticmethod
706
900
    def open_unsupported(base):
707
901
        """Open a branch which is not supported."""
730
924
        # the redirections.
731
925
        base = transport.base
732
926
        def find_format(transport):
733
 
            return transport, controldir.ControlDirFormat.find_format(
 
927
            return transport, BzrDirFormat.find_format(
734
928
                transport, _server_formats=_server_formats)
735
929
 
736
930
        def redirected(transport, e, redirection_notice):
751
945
        BzrDir._check_supported(format, _unsupported)
752
946
        return format.open(transport, _found=True)
753
947
 
 
948
    def open_branch(self, name=None, unsupported=False,
 
949
                    ignore_fallbacks=False):
 
950
        """Open the branch object at this BzrDir if one is present.
 
951
 
 
952
        If unsupported is True, then no longer supported branch formats can
 
953
        still be opened.
 
954
 
 
955
        TODO: static convenience version of this?
 
956
        """
 
957
        raise NotImplementedError(self.open_branch)
 
958
 
754
959
    @staticmethod
755
960
    def open_containing(url, possible_transports=None):
756
961
        """Open an existing branch which contains url.
794
999
                raise errors.NotBranchError(path=url)
795
1000
            a_transport = new_t
796
1001
 
 
1002
    def _get_tree_branch(self, name=None):
 
1003
        """Return the branch and tree, if any, for this bzrdir.
 
1004
 
 
1005
        :param name: Name of colocated branch to open.
 
1006
 
 
1007
        Return None for tree if not present or inaccessible.
 
1008
        Raise NotBranchError if no branch is present.
 
1009
        :return: (tree, branch)
 
1010
        """
 
1011
        try:
 
1012
            tree = self.open_workingtree()
 
1013
        except (errors.NoWorkingTree, errors.NotLocalUrl):
 
1014
            tree = None
 
1015
            branch = self.open_branch(name=name)
 
1016
        else:
 
1017
            if name is not None:
 
1018
                branch = self.open_branch(name=name)
 
1019
            else:
 
1020
                branch = tree.branch
 
1021
        return tree, branch
 
1022
 
797
1023
    @classmethod
798
1024
    def open_tree_or_branch(klass, location):
799
1025
        """Return the branch and working tree at a location.
845
1071
                raise errors.NotBranchError(location)
846
1072
        return tree, branch, branch.repository, relpath
847
1073
 
 
1074
    def open_repository(self, _unsupported=False):
 
1075
        """Open the repository object at this BzrDir if one is present.
 
1076
 
 
1077
        This will not follow the Branch object pointer - it's strictly a direct
 
1078
        open facility. Most client code should use open_branch().repository to
 
1079
        get at a repository.
 
1080
 
 
1081
        :param _unsupported: a private parameter, not part of the api.
 
1082
        TODO: static convenience version of this?
 
1083
        """
 
1084
        raise NotImplementedError(self.open_repository)
 
1085
 
 
1086
    def open_workingtree(self, _unsupported=False,
 
1087
                         recommend_upgrade=True, from_branch=None):
 
1088
        """Open the workingtree object at this BzrDir if one is present.
 
1089
 
 
1090
        :param recommend_upgrade: Optional keyword parameter, when True (the
 
1091
            default), emit through the ui module a recommendation that the user
 
1092
            upgrade the working tree when the workingtree being opened is old
 
1093
            (but still fully supported).
 
1094
        :param from_branch: override bzrdir branch (for lightweight checkouts)
 
1095
        """
 
1096
        raise NotImplementedError(self.open_workingtree)
 
1097
 
 
1098
    def has_branch(self, name=None):
 
1099
        """Tell if this bzrdir contains a branch.
 
1100
 
 
1101
        Note: if you're going to open the branch, you should just go ahead
 
1102
        and try, and not ask permission first.  (This method just opens the
 
1103
        branch and discards it, and that's somewhat expensive.)
 
1104
        """
 
1105
        try:
 
1106
            self.open_branch(name)
 
1107
            return True
 
1108
        except errors.NotBranchError:
 
1109
            return False
 
1110
 
 
1111
    def has_workingtree(self):
 
1112
        """Tell if this bzrdir contains a working tree.
 
1113
 
 
1114
        This will still raise an exception if the bzrdir has a workingtree that
 
1115
        is remote & inaccessible.
 
1116
 
 
1117
        Note: if you're going to open the working tree, you should just go ahead
 
1118
        and try, and not ask permission first.  (This method just opens the
 
1119
        workingtree and discards it, and that's somewhat expensive.)
 
1120
        """
 
1121
        try:
 
1122
            self.open_workingtree(recommend_upgrade=False)
 
1123
            return True
 
1124
        except errors.NoWorkingTree:
 
1125
            return False
 
1126
 
848
1127
    def _cloning_metadir(self):
849
1128
        """Produce a metadir suitable for cloning with.
850
1129
 
908
1187
            format.require_stacking()
909
1188
        return format
910
1189
 
911
 
    @classmethod
912
 
    def create(cls, base, format=None, possible_transports=None):
913
 
        """Create a new BzrDir at the url 'base'.
914
 
 
915
 
        :param format: If supplied, the format of branch to create.  If not
916
 
            supplied, the default is used.
917
 
        :param possible_transports: If supplied, a list of transports that
918
 
            can be reused to share a remote connection.
 
1190
    def checkout_metadir(self):
 
1191
        return self.cloning_metadir()
 
1192
 
 
1193
    def sprout(self, url, revision_id=None, force_new_repo=False,
 
1194
               recurse='down', possible_transports=None,
 
1195
               accelerator_tree=None, hardlink=False, stacked=False,
 
1196
               source_branch=None, create_tree_if_local=True):
 
1197
        """Create a copy of this bzrdir prepared for use as a new line of
 
1198
        development.
 
1199
 
 
1200
        If url's last component does not exist, it will be created.
 
1201
 
 
1202
        Attributes related to the identity of the source branch like
 
1203
        branch nickname will be cleaned, a working tree is created
 
1204
        whether one existed before or not; and a local branch is always
 
1205
        created.
 
1206
 
 
1207
        if revision_id is not None, then the clone operation may tune
 
1208
            itself to download less data.
 
1209
        :param accelerator_tree: A tree which can be used for retrieving file
 
1210
            contents more quickly than the revision tree, i.e. a workingtree.
 
1211
            The revision tree will be used for cases where accelerator_tree's
 
1212
            content is different.
 
1213
        :param hardlink: If true, hard-link files from accelerator_tree,
 
1214
            where possible.
 
1215
        :param stacked: If true, create a stacked branch referring to the
 
1216
            location of this control directory.
 
1217
        :param create_tree_if_local: If true, a working-tree will be created
 
1218
            when working locally.
919
1219
        """
920
 
        if cls is not BzrDir:
921
 
            raise AssertionError("BzrDir.create always creates the"
922
 
                "default format, not one of %r" % cls)
923
 
        t = get_transport(base, possible_transports)
924
 
        t.ensure_base()
925
 
        if format is None:
926
 
            format = controldir.ControlDirFormat.get_default_format()
927
 
        return format.initialize_on_transport(t)
928
 
 
 
1220
        target_transport = get_transport(url, possible_transports)
 
1221
        target_transport.ensure_base()
 
1222
        cloning_format = self.cloning_metadir(stacked)
 
1223
        # Create/update the result branch
 
1224
        result = cloning_format.initialize_on_transport(target_transport)
 
1225
        # if a stacked branch wasn't requested, we don't create one
 
1226
        # even if the origin was stacked
 
1227
        stacked_branch_url = None
 
1228
        if source_branch is not None:
 
1229
            if stacked:
 
1230
                stacked_branch_url = self.root_transport.base
 
1231
            source_repository = source_branch.repository
 
1232
        else:
 
1233
            try:
 
1234
                source_branch = self.open_branch()
 
1235
                source_repository = source_branch.repository
 
1236
                if stacked:
 
1237
                    stacked_branch_url = self.root_transport.base
 
1238
            except errors.NotBranchError:
 
1239
                source_branch = None
 
1240
                try:
 
1241
                    source_repository = self.open_repository()
 
1242
                except errors.NoRepositoryPresent:
 
1243
                    source_repository = None
 
1244
        repository_policy = result.determine_repository_policy(
 
1245
            force_new_repo, stacked_branch_url, require_stacking=stacked)
 
1246
        result_repo, is_new_repo = repository_policy.acquire_repository()
 
1247
        if is_new_repo and revision_id is not None and not stacked:
 
1248
            fetch_spec = graph.PendingAncestryResult(
 
1249
                [revision_id], source_repository)
 
1250
        else:
 
1251
            fetch_spec = None
 
1252
        if source_repository is not None:
 
1253
            # Fetch while stacked to prevent unstacked fetch from
 
1254
            # Branch.sprout.
 
1255
            if fetch_spec is None:
 
1256
                result_repo.fetch(source_repository, revision_id=revision_id)
 
1257
            else:
 
1258
                result_repo.fetch(source_repository, fetch_spec=fetch_spec)
 
1259
 
 
1260
        if source_branch is None:
 
1261
            # this is for sprouting a bzrdir without a branch; is that
 
1262
            # actually useful?
 
1263
            # Not especially, but it's part of the contract.
 
1264
            result_branch = result.create_branch()
 
1265
        else:
 
1266
            result_branch = source_branch.sprout(result,
 
1267
                revision_id=revision_id, repository_policy=repository_policy)
 
1268
        mutter("created new branch %r" % (result_branch,))
 
1269
 
 
1270
        # Create/update the result working tree
 
1271
        if (create_tree_if_local and
 
1272
            isinstance(target_transport, local.LocalTransport) and
 
1273
            (result_repo is None or result_repo.make_working_trees())):
 
1274
            wt = result.create_workingtree(accelerator_tree=accelerator_tree,
 
1275
                hardlink=hardlink)
 
1276
            wt.lock_write()
 
1277
            try:
 
1278
                if wt.path2id('') is None:
 
1279
                    try:
 
1280
                        wt.set_root_id(self.open_workingtree.get_root_id())
 
1281
                    except errors.NoWorkingTree:
 
1282
                        pass
 
1283
            finally:
 
1284
                wt.unlock()
 
1285
        else:
 
1286
            wt = None
 
1287
        if recurse == 'down':
 
1288
            if wt is not None:
 
1289
                basis = wt.basis_tree()
 
1290
                basis.lock_read()
 
1291
                subtrees = basis.iter_references()
 
1292
            elif result_branch is not None:
 
1293
                basis = result_branch.basis_tree()
 
1294
                basis.lock_read()
 
1295
                subtrees = basis.iter_references()
 
1296
            elif source_branch is not None:
 
1297
                basis = source_branch.basis_tree()
 
1298
                basis.lock_read()
 
1299
                subtrees = basis.iter_references()
 
1300
            else:
 
1301
                subtrees = []
 
1302
                basis = None
 
1303
            try:
 
1304
                for path, file_id in subtrees:
 
1305
                    target = urlutils.join(url, urlutils.escape(path))
 
1306
                    sublocation = source_branch.reference_parent(file_id, path)
 
1307
                    sublocation.bzrdir.sprout(target,
 
1308
                        basis.get_reference_revision(file_id, path),
 
1309
                        force_new_repo=force_new_repo, recurse=recurse,
 
1310
                        stacked=stacked)
 
1311
            finally:
 
1312
                if basis is not None:
 
1313
                    basis.unlock()
 
1314
        return result
 
1315
 
 
1316
    def push_branch(self, source, revision_id=None, overwrite=False, 
 
1317
        remember=False, create_prefix=False):
 
1318
        """Push the source branch into this BzrDir."""
 
1319
        br_to = None
 
1320
        # If we can open a branch, use its direct repository, otherwise see
 
1321
        # if there is a repository without a branch.
 
1322
        try:
 
1323
            br_to = self.open_branch()
 
1324
        except errors.NotBranchError:
 
1325
            # Didn't find a branch, can we find a repository?
 
1326
            repository_to = self.find_repository()
 
1327
        else:
 
1328
            # Found a branch, so we must have found a repository
 
1329
            repository_to = br_to.repository
 
1330
 
 
1331
        push_result = PushResult()
 
1332
        push_result.source_branch = source
 
1333
        if br_to is None:
 
1334
            # We have a repository but no branch, copy the revisions, and then
 
1335
            # create a branch.
 
1336
            repository_to.fetch(source.repository, revision_id=revision_id)
 
1337
            br_to = source.clone(self, revision_id=revision_id)
 
1338
            if source.get_push_location() is None or remember:
 
1339
                source.set_push_location(br_to.base)
 
1340
            push_result.stacked_on = None
 
1341
            push_result.branch_push_result = None
 
1342
            push_result.old_revno = None
 
1343
            push_result.old_revid = _mod_revision.NULL_REVISION
 
1344
            push_result.target_branch = br_to
 
1345
            push_result.master_branch = None
 
1346
            push_result.workingtree_updated = False
 
1347
        else:
 
1348
            # We have successfully opened the branch, remember if necessary:
 
1349
            if source.get_push_location() is None or remember:
 
1350
                source.set_push_location(br_to.base)
 
1351
            try:
 
1352
                tree_to = self.open_workingtree()
 
1353
            except errors.NotLocalUrl:
 
1354
                push_result.branch_push_result = source.push(br_to, 
 
1355
                    overwrite, stop_revision=revision_id)
 
1356
                push_result.workingtree_updated = False
 
1357
            except errors.NoWorkingTree:
 
1358
                push_result.branch_push_result = source.push(br_to,
 
1359
                    overwrite, stop_revision=revision_id)
 
1360
                push_result.workingtree_updated = None # Not applicable
 
1361
            else:
 
1362
                tree_to.lock_write()
 
1363
                try:
 
1364
                    push_result.branch_push_result = source.push(
 
1365
                        tree_to.branch, overwrite, stop_revision=revision_id)
 
1366
                    tree_to.update()
 
1367
                finally:
 
1368
                    tree_to.unlock()
 
1369
                push_result.workingtree_updated = True
 
1370
            push_result.old_revno = push_result.branch_push_result.old_revno
 
1371
            push_result.old_revid = push_result.branch_push_result.old_revid
 
1372
            push_result.target_branch = \
 
1373
                push_result.branch_push_result.target_branch
 
1374
        return push_result
929
1375
 
930
1376
 
931
1377
class BzrDirHooks(hooks.Hooks):
1001
1447
    def cloning_metadir(self, require_stacking=False):
1002
1448
        """Produce a metadir suitable for cloning with."""
1003
1449
        if require_stacking:
1004
 
            return controldir.format_registry.make_bzrdir('1.6')
 
1450
            return format_registry.make_bzrdir('1.6')
1005
1451
        return self._format.__class__()
1006
1452
 
1007
1453
    def clone(self, url, revision_id=None, force_new_repo=False,
1294
1740
        wt = self.open_workingtree(recommend_upgrade=False)
1295
1741
        repository = wt.branch.repository
1296
1742
        empty = repository.revision_tree(_mod_revision.NULL_REVISION)
1297
 
        # We ignore the conflicts returned by wt.revert since we're about to
1298
 
        # delete the wt metadata anyway, all that should be left here are
1299
 
        # detritus. But see bug #634470 about subtree .bzr dirs.
1300
 
        conflicts = wt.revert(old_tree=empty)
 
1743
        wt.revert(old_tree=empty)
1301
1744
        self.destroy_workingtree_metadata()
1302
1745
 
1303
1746
    def destroy_workingtree_metadata(self):
1447
1890
        return config.TransportConfig(self.transport, 'control.conf')
1448
1891
 
1449
1892
 
1450
 
class BzrProber(controldir.Prober):
1451
 
    """Prober for formats that use a .bzr/ control directory."""
1452
 
 
1453
 
    _formats = {}
1454
 
    """The known .bzr formats."""
1455
 
 
1456
 
    @classmethod
1457
 
    def register_bzrdir_format(klass, format):
1458
 
        klass._formats[format.get_format_string()] = format
1459
 
 
1460
 
    @classmethod
1461
 
    def unregister_bzrdir_format(klass, format):
1462
 
        del klass._formats[format.get_format_string()]
1463
 
 
1464
 
    @classmethod
1465
 
    def probe_transport(klass, transport):
1466
 
        """Return the .bzrdir style format present in a directory."""
1467
 
        try:
1468
 
            format_string = transport.get_bytes(".bzr/branch-format")
1469
 
        except errors.NoSuchFile:
1470
 
            raise errors.NotBranchError(path=transport.base)
1471
 
        try:
1472
 
            return klass._formats[format_string]
1473
 
        except KeyError:
1474
 
            raise errors.UnknownFormatError(format=format_string, kind='bzrdir')
1475
 
 
1476
 
 
1477
 
controldir.ControlDirFormat.register_prober(BzrProber)
1478
 
 
1479
 
 
1480
 
class RemoteBzrProber(controldir.Prober):
1481
 
    """Prober for remote servers that provide a Bazaar smart server."""
1482
 
 
1483
 
    @classmethod
1484
 
    def probe_transport(klass, transport):
1485
 
        """Return a RemoteBzrDirFormat object if it looks possible."""
1486
 
        try:
1487
 
            medium = transport.get_smart_medium()
1488
 
        except (NotImplementedError, AttributeError,
1489
 
                errors.TransportNotPossible, errors.NoSmartMedium,
1490
 
                errors.SmartProtocolError):
1491
 
            # no smart server, so not a branch for this format type.
1492
 
            raise errors.NotBranchError(path=transport.base)
1493
 
        else:
1494
 
            # Decline to open it if the server doesn't support our required
1495
 
            # version (3) so that the VFS-based transport will do it.
1496
 
            if medium.should_probe():
1497
 
                try:
1498
 
                    server_version = medium.protocol_version()
1499
 
                except errors.SmartProtocolError:
1500
 
                    # Apparently there's no usable smart server there, even though
1501
 
                    # the medium supports the smart protocol.
1502
 
                    raise errors.NotBranchError(path=transport.base)
1503
 
                if server_version != '2':
1504
 
                    raise errors.NotBranchError(path=transport.base)
1505
 
            return RemoteBzrDirFormat()
1506
 
 
1507
 
 
1508
 
class BzrDirFormat(controldir.ControlDirFormat):
1509
 
    """ControlDirFormat base class for .bzr/ directories.
 
1893
class BzrDirFormat(object):
 
1894
    """An encapsulation of the initialization and open routines for a format.
 
1895
 
 
1896
    Formats provide three things:
 
1897
     * An initialization routine,
 
1898
     * a format string,
 
1899
     * an open routine.
1510
1900
 
1511
1901
    Formats are placed in a dict by their format string for reference
1512
1902
    during bzrdir opening. These should be subclasses of BzrDirFormat
1515
1905
    Once a format is deprecated, just deprecate the initialize and open
1516
1906
    methods on the format class. Do not deprecate the object, as the
1517
1907
    object will be created every system load.
 
1908
 
 
1909
    :cvar colocated_branches: Whether this formats supports colocated branches.
 
1910
    """
 
1911
 
 
1912
    _default_format = None
 
1913
    """The default format used for new .bzr dirs."""
 
1914
 
 
1915
    _formats = {}
 
1916
    """The known formats."""
 
1917
 
 
1918
    _control_formats = []
 
1919
    """The registered control formats - .bzr, ....
 
1920
 
 
1921
    This is a list of BzrDirFormat objects.
 
1922
    """
 
1923
 
 
1924
    _control_server_formats = []
 
1925
    """The registered control server formats, e.g. RemoteBzrDirs.
 
1926
 
 
1927
    This is a list of BzrDirFormat objects.
1518
1928
    """
1519
1929
 
1520
1930
    _lock_file_name = 'branch-lock'
1521
1931
 
 
1932
    colocated_branches = False
 
1933
    """Whether co-located branches are supported for this control dir format.
 
1934
    """
 
1935
 
1522
1936
    # _lock_class must be set in subclasses to the lock type, typ.
1523
1937
    # TransportLock or LockDir
1524
1938
 
 
1939
    @classmethod
 
1940
    def find_format(klass, transport, _server_formats=True):
 
1941
        """Return the format present at transport."""
 
1942
        if _server_formats:
 
1943
            formats = klass._control_server_formats + klass._control_formats
 
1944
        else:
 
1945
            formats = klass._control_formats
 
1946
        for format in formats:
 
1947
            try:
 
1948
                return format.probe_transport(transport)
 
1949
            except errors.NotBranchError:
 
1950
                # this format does not find a control dir here.
 
1951
                pass
 
1952
        raise errors.NotBranchError(path=transport.base)
 
1953
 
 
1954
    @classmethod
 
1955
    def probe_transport(klass, transport):
 
1956
        """Return the .bzrdir style format present in a directory."""
 
1957
        try:
 
1958
            format_string = transport.get_bytes(".bzr/branch-format")
 
1959
        except errors.NoSuchFile:
 
1960
            raise errors.NotBranchError(path=transport.base)
 
1961
 
 
1962
        try:
 
1963
            return klass._formats[format_string]
 
1964
        except KeyError:
 
1965
            raise errors.UnknownFormatError(format=format_string, kind='bzrdir')
 
1966
 
 
1967
    @classmethod
 
1968
    def get_default_format(klass):
 
1969
        """Return the current default format."""
 
1970
        return klass._default_format
 
1971
 
1525
1972
    def get_format_string(self):
1526
1973
        """Return the ASCII format string that identifies this format."""
1527
1974
        raise NotImplementedError(self.get_format_string)
1528
1975
 
 
1976
    def get_format_description(self):
 
1977
        """Return the short description for this format."""
 
1978
        raise NotImplementedError(self.get_format_description)
 
1979
 
 
1980
    def get_converter(self, format=None):
 
1981
        """Return the converter to use to convert bzrdirs needing converts.
 
1982
 
 
1983
        This returns a bzrlib.bzrdir.Converter object.
 
1984
 
 
1985
        This should return the best upgrader to step this format towards the
 
1986
        current default format. In the case of plugins we can/should provide
 
1987
        some means for them to extend the range of returnable converters.
 
1988
 
 
1989
        :param format: Optional format to override the default format of the
 
1990
                       library.
 
1991
        """
 
1992
        raise NotImplementedError(self.get_converter)
 
1993
 
 
1994
    def initialize(self, url, possible_transports=None):
 
1995
        """Create a bzr control dir at this url and return an opened copy.
 
1996
 
 
1997
        While not deprecated, this method is very specific and its use will
 
1998
        lead to many round trips to setup a working environment. See
 
1999
        initialize_on_transport_ex for a [nearly] all-in-one method.
 
2000
 
 
2001
        Subclasses should typically override initialize_on_transport
 
2002
        instead of this method.
 
2003
        """
 
2004
        return self.initialize_on_transport(get_transport(url,
 
2005
                                                          possible_transports))
 
2006
 
1529
2007
    def initialize_on_transport(self, transport):
1530
2008
        """Initialize a new bzrdir in the base directory of a Transport."""
1531
2009
        try:
1679
2157
            control_files.unlock()
1680
2158
        return self.open(transport, _found=True)
1681
2159
 
 
2160
    def is_supported(self):
 
2161
        """Is this format supported?
 
2162
 
 
2163
        Supported formats must be initializable and openable.
 
2164
        Unsupported formats may not support initialization or committing or
 
2165
        some other features depending on the reason for not being supported.
 
2166
        """
 
2167
        return True
 
2168
 
 
2169
    def network_name(self):
 
2170
        """A simple byte string uniquely identifying this format for RPC calls.
 
2171
 
 
2172
        Bzr control formats use thir disk format string to identify the format
 
2173
        over the wire. Its possible that other control formats have more
 
2174
        complex detection requirements, so we permit them to use any unique and
 
2175
        immutable string they desire.
 
2176
        """
 
2177
        raise NotImplementedError(self.network_name)
 
2178
 
 
2179
    def same_model(self, target_format):
 
2180
        return (self.repository_format.rich_root_data ==
 
2181
            target_format.rich_root_data)
 
2182
 
 
2183
    @classmethod
 
2184
    def known_formats(klass):
 
2185
        """Return all the known formats.
 
2186
 
 
2187
        Concrete formats should override _known_formats.
 
2188
        """
 
2189
        # There is double indirection here to make sure that control
 
2190
        # formats used by more than one dir format will only be probed
 
2191
        # once. This can otherwise be quite expensive for remote connections.
 
2192
        result = set()
 
2193
        for format in klass._control_formats:
 
2194
            result.update(format._known_formats())
 
2195
        return result
 
2196
 
 
2197
    @classmethod
 
2198
    def _known_formats(klass):
 
2199
        """Return the known format instances for this control format."""
 
2200
        return set(klass._formats.values())
 
2201
 
1682
2202
    def open(self, transport, _found=False):
1683
2203
        """Return an instance of this format for the dir transport points at.
1684
2204
 
1685
2205
        _found is a private parameter, do not use it.
1686
2206
        """
1687
2207
        if not _found:
1688
 
            found_format = controldir.ControlDirFormat.find_format(transport)
 
2208
            found_format = BzrDirFormat.find_format(transport)
1689
2209
            if not isinstance(found_format, self.__class__):
1690
2210
                raise AssertionError("%s was asked to open %s, but it seems to need "
1691
2211
                        "format %s"
1705
2225
 
1706
2226
    @classmethod
1707
2227
    def register_format(klass, format):
1708
 
        BzrProber.register_bzrdir_format(format)
 
2228
        klass._formats[format.get_format_string()] = format
1709
2229
        # bzr native formats have a network name of their format string.
1710
 
        controldir.network_format_registry.register(format.get_format_string(), format.__class__)
1711
 
        controldir.ControlDirFormat.register_format(format)
 
2230
        network_format_registry.register(format.get_format_string(), format.__class__)
 
2231
 
 
2232
    @classmethod
 
2233
    def register_control_format(klass, format):
 
2234
        """Register a format that does not use '.bzr' for its control dir.
 
2235
 
 
2236
        TODO: This should be pulled up into a 'ControlDirFormat' base class
 
2237
        which BzrDirFormat can inherit from, and renamed to register_format
 
2238
        there. It has been done without that for now for simplicity of
 
2239
        implementation.
 
2240
        """
 
2241
        klass._control_formats.append(format)
 
2242
 
 
2243
    @classmethod
 
2244
    def register_control_server_format(klass, format):
 
2245
        """Register a control format for client-server environments.
 
2246
 
 
2247
        These formats will be tried before ones registered with
 
2248
        register_control_format.  This gives implementations that decide to the
 
2249
        chance to grab it before anything looks at the contents of the format
 
2250
        file.
 
2251
        """
 
2252
        klass._control_server_formats.append(format)
 
2253
 
 
2254
    @classmethod
 
2255
    def _set_default_format(klass, format):
 
2256
        """Set default format (for testing behavior of defaults only)"""
 
2257
        klass._default_format = format
 
2258
 
 
2259
    def __str__(self):
 
2260
        # Trim the newline
 
2261
        return self.get_format_description().rstrip()
1712
2262
 
1713
2263
    def _supply_sub_formats_to(self, other_format):
1714
2264
        """Give other_format the same values for sub formats as this has.
1724
2274
 
1725
2275
    @classmethod
1726
2276
    def unregister_format(klass, format):
1727
 
        BzrProber.unregister_bzrdir_format(format)
1728
 
        controldir.ControlDirFormat.unregister_format(format)
1729
 
        controldir.network_format_registry.remove(format.get_format_string())
 
2277
        del klass._formats[format.get_format_string()]
 
2278
 
 
2279
    @classmethod
 
2280
    def unregister_control_format(klass, format):
 
2281
        klass._control_formats.remove(format)
1730
2282
 
1731
2283
 
1732
2284
class BzrDirFormat4(BzrDirFormat):
2135
2687
                                  __set_workingtree_format)
2136
2688
 
2137
2689
 
 
2690
network_format_registry = registry.FormatRegistry()
 
2691
"""Registry of formats indexed by their network name.
 
2692
 
 
2693
The network name for a BzrDirFormat is an identifier that can be used when
 
2694
referring to formats with smart server operations. See
 
2695
BzrDirFormat.network_name() for more detail.
 
2696
"""
 
2697
 
 
2698
 
 
2699
# Register bzr control format
 
2700
BzrDirFormat.register_control_format(BzrDirFormat)
 
2701
 
2138
2702
# Register bzr formats
2139
2703
BzrDirFormat.register_format(BzrDirFormat4())
2140
2704
BzrDirFormat.register_format(BzrDirFormat5())
2141
2705
BzrDirFormat.register_format(BzrDirFormat6())
2142
2706
__default_format = BzrDirMetaFormat1()
2143
2707
BzrDirFormat.register_format(__default_format)
2144
 
controldir.ControlDirFormat._default_format = __default_format
 
2708
BzrDirFormat._default_format = __default_format
2145
2709
 
2146
2710
 
2147
2711
class Converter(object):
2375
2939
        previous_entries = dict((head, parent_candiate_entries[head]) for head
2376
2940
            in heads)
2377
2941
        self.snapshot_ie(previous_entries, ie, w, rev_id)
 
2942
        del ie.text_id
2378
2943
 
2379
2944
    def get_parent_map(self, revision_ids):
2380
2945
        """See graph.StackedParentsProvider.get_parent_map"""
2682
3247
class RemoteBzrDirFormat(BzrDirMetaFormat1):
2683
3248
    """Format representing bzrdirs accessed via a smart server"""
2684
3249
 
2685
 
    supports_workingtrees = False
2686
 
 
2687
3250
    def __init__(self):
2688
3251
        BzrDirMetaFormat1.__init__(self)
2689
3252
        # XXX: It's a bit ugly that the network name is here, because we'd
2698
3261
 
2699
3262
    def get_format_description(self):
2700
3263
        if self._network_name:
2701
 
            real_format = controldir.network_format_registry.get(self._network_name)
 
3264
            real_format = network_format_registry.get(self._network_name)
2702
3265
            return 'Remote: ' + real_format.get_format_description()
2703
3266
        return 'bzr remote bzrdir'
2704
3267
 
2711
3274
        else:
2712
3275
            raise AssertionError("No network name set.")
2713
3276
 
 
3277
    @classmethod
 
3278
    def probe_transport(klass, transport):
 
3279
        """Return a RemoteBzrDirFormat object if it looks possible."""
 
3280
        try:
 
3281
            medium = transport.get_smart_medium()
 
3282
        except (NotImplementedError, AttributeError,
 
3283
                errors.TransportNotPossible, errors.NoSmartMedium,
 
3284
                errors.SmartProtocolError):
 
3285
            # no smart server, so not a branch for this format type.
 
3286
            raise errors.NotBranchError(path=transport.base)
 
3287
        else:
 
3288
            # Decline to open it if the server doesn't support our required
 
3289
            # version (3) so that the VFS-based transport will do it.
 
3290
            if medium.should_probe():
 
3291
                try:
 
3292
                    server_version = medium.protocol_version()
 
3293
                except errors.SmartProtocolError:
 
3294
                    # Apparently there's no usable smart server there, even though
 
3295
                    # the medium supports the smart protocol.
 
3296
                    raise errors.NotBranchError(path=transport.base)
 
3297
                if server_version != '2':
 
3298
                    raise errors.NotBranchError(path=transport.base)
 
3299
            return klass()
 
3300
 
2714
3301
    def initialize_on_transport(self, transport):
2715
3302
        try:
2716
3303
            # hand off the request to the smart server
2915
3502
        BzrDirMetaFormat1._set_repository_format) #.im_func)
2916
3503
 
2917
3504
 
2918
 
controldir.ControlDirFormat.register_server_prober(RemoteBzrProber)
 
3505
BzrDirFormat.register_control_server_format(RemoteBzrDirFormat)
 
3506
 
 
3507
 
 
3508
class BzrDirFormatInfo(object):
 
3509
 
 
3510
    def __init__(self, native, deprecated, hidden, experimental):
 
3511
        self.deprecated = deprecated
 
3512
        self.native = native
 
3513
        self.hidden = hidden
 
3514
        self.experimental = experimental
 
3515
 
 
3516
 
 
3517
class BzrDirFormatRegistry(registry.Registry):
 
3518
    """Registry of user-selectable BzrDir subformats.
 
3519
 
 
3520
    Differs from BzrDirFormat._control_formats in that it provides sub-formats,
 
3521
    e.g. BzrDirMeta1 with weave repository.  Also, it's more user-oriented.
 
3522
    """
 
3523
 
 
3524
    def __init__(self):
 
3525
        """Create a BzrDirFormatRegistry."""
 
3526
        self._aliases = set()
 
3527
        self._registration_order = list()
 
3528
        super(BzrDirFormatRegistry, self).__init__()
 
3529
 
 
3530
    def aliases(self):
 
3531
        """Return a set of the format names which are aliases."""
 
3532
        return frozenset(self._aliases)
 
3533
 
 
3534
    def register_metadir(self, key,
 
3535
             repository_format, help, native=True, deprecated=False,
 
3536
             branch_format=None,
 
3537
             tree_format=None,
 
3538
             hidden=False,
 
3539
             experimental=False,
 
3540
             alias=False):
 
3541
        """Register a metadir subformat.
 
3542
 
 
3543
        These all use a BzrDirMetaFormat1 bzrdir, but can be parameterized
 
3544
        by the Repository/Branch/WorkingTreeformats.
 
3545
 
 
3546
        :param repository_format: The fully-qualified repository format class
 
3547
            name as a string.
 
3548
        :param branch_format: Fully-qualified branch format class name as
 
3549
            a string.
 
3550
        :param tree_format: Fully-qualified tree format class name as
 
3551
            a string.
 
3552
        """
 
3553
        # This should be expanded to support setting WorkingTree and Branch
 
3554
        # formats, once BzrDirMetaFormat1 supports that.
 
3555
        def _load(full_name):
 
3556
            mod_name, factory_name = full_name.rsplit('.', 1)
 
3557
            try:
 
3558
                mod = __import__(mod_name, globals(), locals(),
 
3559
                        [factory_name])
 
3560
            except ImportError, e:
 
3561
                raise ImportError('failed to load %s: %s' % (full_name, e))
 
3562
            try:
 
3563
                factory = getattr(mod, factory_name)
 
3564
            except AttributeError:
 
3565
                raise AttributeError('no factory %s in module %r'
 
3566
                    % (full_name, mod))
 
3567
            return factory()
 
3568
 
 
3569
        def helper():
 
3570
            bd = BzrDirMetaFormat1()
 
3571
            if branch_format is not None:
 
3572
                bd.set_branch_format(_load(branch_format))
 
3573
            if tree_format is not None:
 
3574
                bd.workingtree_format = _load(tree_format)
 
3575
            if repository_format is not None:
 
3576
                bd.repository_format = _load(repository_format)
 
3577
            return bd
 
3578
        self.register(key, helper, help, native, deprecated, hidden,
 
3579
            experimental, alias)
 
3580
 
 
3581
    def register(self, key, factory, help, native=True, deprecated=False,
 
3582
                 hidden=False, experimental=False, alias=False):
 
3583
        """Register a BzrDirFormat factory.
 
3584
 
 
3585
        The factory must be a callable that takes one parameter: the key.
 
3586
        It must produce an instance of the BzrDirFormat when called.
 
3587
 
 
3588
        This function mainly exists to prevent the info object from being
 
3589
        supplied directly.
 
3590
        """
 
3591
        registry.Registry.register(self, key, factory, help,
 
3592
            BzrDirFormatInfo(native, deprecated, hidden, experimental))
 
3593
        if alias:
 
3594
            self._aliases.add(key)
 
3595
        self._registration_order.append(key)
 
3596
 
 
3597
    def register_lazy(self, key, module_name, member_name, help, native=True,
 
3598
        deprecated=False, hidden=False, experimental=False, alias=False):
 
3599
        registry.Registry.register_lazy(self, key, module_name, member_name,
 
3600
            help, BzrDirFormatInfo(native, deprecated, hidden, experimental))
 
3601
        if alias:
 
3602
            self._aliases.add(key)
 
3603
        self._registration_order.append(key)
 
3604
 
 
3605
    def set_default(self, key):
 
3606
        """Set the 'default' key to be a clone of the supplied key.
 
3607
 
 
3608
        This method must be called once and only once.
 
3609
        """
 
3610
        registry.Registry.register(self, 'default', self.get(key),
 
3611
            self.get_help(key), info=self.get_info(key))
 
3612
        self._aliases.add('default')
 
3613
 
 
3614
    def set_default_repository(self, key):
 
3615
        """Set the FormatRegistry default and Repository default.
 
3616
 
 
3617
        This is a transitional method while Repository.set_default_format
 
3618
        is deprecated.
 
3619
        """
 
3620
        if 'default' in self:
 
3621
            self.remove('default')
 
3622
        self.set_default(key)
 
3623
        format = self.get('default')()
 
3624
 
 
3625
    def make_bzrdir(self, key):
 
3626
        return self.get(key)()
 
3627
 
 
3628
    def help_topic(self, topic):
 
3629
        output = ""
 
3630
        default_realkey = None
 
3631
        default_help = self.get_help('default')
 
3632
        help_pairs = []
 
3633
        for key in self._registration_order:
 
3634
            if key == 'default':
 
3635
                continue
 
3636
            help = self.get_help(key)
 
3637
            if help == default_help:
 
3638
                default_realkey = key
 
3639
            else:
 
3640
                help_pairs.append((key, help))
 
3641
 
 
3642
        def wrapped(key, help, info):
 
3643
            if info.native:
 
3644
                help = '(native) ' + help
 
3645
            return ':%s:\n%s\n\n' % (key,
 
3646
                textwrap.fill(help, initial_indent='    ',
 
3647
                    subsequent_indent='    ',
 
3648
                    break_long_words=False))
 
3649
        if default_realkey is not None:
 
3650
            output += wrapped(default_realkey, '(default) %s' % default_help,
 
3651
                              self.get_info('default'))
 
3652
        deprecated_pairs = []
 
3653
        experimental_pairs = []
 
3654
        for key, help in help_pairs:
 
3655
            info = self.get_info(key)
 
3656
            if info.hidden:
 
3657
                continue
 
3658
            elif info.deprecated:
 
3659
                deprecated_pairs.append((key, help))
 
3660
            elif info.experimental:
 
3661
                experimental_pairs.append((key, help))
 
3662
            else:
 
3663
                output += wrapped(key, help, info)
 
3664
        output += "\nSee :doc:`formats-help` for more about storage formats."
 
3665
        other_output = ""
 
3666
        if len(experimental_pairs) > 0:
 
3667
            other_output += "Experimental formats are shown below.\n\n"
 
3668
            for key, help in experimental_pairs:
 
3669
                info = self.get_info(key)
 
3670
                other_output += wrapped(key, help, info)
 
3671
        else:
 
3672
            other_output += \
 
3673
                "No experimental formats are available.\n\n"
 
3674
        if len(deprecated_pairs) > 0:
 
3675
            other_output += "\nDeprecated formats are shown below.\n\n"
 
3676
            for key, help in deprecated_pairs:
 
3677
                info = self.get_info(key)
 
3678
                other_output += wrapped(key, help, info)
 
3679
        else:
 
3680
            other_output += \
 
3681
                "\nNo deprecated formats are available.\n\n"
 
3682
        other_output += \
 
3683
                "\nSee :doc:`formats-help` for more about storage formats."
 
3684
 
 
3685
        if topic == 'other-formats':
 
3686
            return other_output
 
3687
        else:
 
3688
            return output
2919
3689
 
2920
3690
 
2921
3691
class RepositoryAcquisitionPolicy(object):
3075
3845
        return self._repository, False
3076
3846
 
3077
3847
 
3078
 
def register_metadir(registry, key,
3079
 
         repository_format, help, native=True, deprecated=False,
3080
 
         branch_format=None,
3081
 
         tree_format=None,
3082
 
         hidden=False,
3083
 
         experimental=False,
3084
 
         alias=False):
3085
 
    """Register a metadir subformat.
3086
 
 
3087
 
    These all use a BzrDirMetaFormat1 bzrdir, but can be parameterized
3088
 
    by the Repository/Branch/WorkingTreeformats.
3089
 
 
3090
 
    :param repository_format: The fully-qualified repository format class
3091
 
        name as a string.
3092
 
    :param branch_format: Fully-qualified branch format class name as
3093
 
        a string.
3094
 
    :param tree_format: Fully-qualified tree format class name as
3095
 
        a string.
3096
 
    """
3097
 
    # This should be expanded to support setting WorkingTree and Branch
3098
 
    # formats, once BzrDirMetaFormat1 supports that.
3099
 
    def _load(full_name):
3100
 
        mod_name, factory_name = full_name.rsplit('.', 1)
3101
 
        try:
3102
 
            factory = pyutils.get_named_object(mod_name, factory_name)
3103
 
        except ImportError, e:
3104
 
            raise ImportError('failed to load %s: %s' % (full_name, e))
3105
 
        except AttributeError:
3106
 
            raise AttributeError('no factory %s in module %r'
3107
 
                % (full_name, sys.modules[mod_name]))
3108
 
        return factory()
3109
 
 
3110
 
    def helper():
3111
 
        bd = BzrDirMetaFormat1()
3112
 
        if branch_format is not None:
3113
 
            bd.set_branch_format(_load(branch_format))
3114
 
        if tree_format is not None:
3115
 
            bd.workingtree_format = _load(tree_format)
3116
 
        if repository_format is not None:
3117
 
            bd.repository_format = _load(repository_format)
3118
 
        return bd
3119
 
    registry.register(key, helper, help, native, deprecated, hidden,
3120
 
        experimental, alias)
3121
 
 
 
3848
# Please register new formats after old formats so that formats
 
3849
# appear in chronological order and format descriptions can build
 
3850
# on previous ones.
 
3851
format_registry = BzrDirFormatRegistry()
3122
3852
# The pre-0.8 formats have their repository format network name registered in
3123
3853
# repository.py. MetaDir formats have their repository format network name
3124
3854
# inferred from their disk format string.
3125
 
controldir.format_registry.register('weave', BzrDirFormat6,
 
3855
format_registry.register('weave', BzrDirFormat6,
3126
3856
    'Pre-0.8 format.  Slower than knit and does not'
3127
3857
    ' support checkouts or shared repositories.',
3128
3858
    hidden=True,
3129
3859
    deprecated=True)
3130
 
register_metadir(controldir.format_registry, 'metaweave',
 
3860
format_registry.register_metadir('metaweave',
3131
3861
    'bzrlib.repofmt.weaverepo.RepositoryFormat7',
3132
3862
    'Transitional format in 0.8.  Slower than knit.',
3133
3863
    branch_format='bzrlib.branch.BzrBranchFormat5',
3134
3864
    tree_format='bzrlib.workingtree.WorkingTreeFormat3',
3135
3865
    hidden=True,
3136
3866
    deprecated=True)
3137
 
register_metadir(controldir.format_registry, 'knit',
 
3867
format_registry.register_metadir('knit',
3138
3868
    'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
3139
3869
    'Format using knits.  Recommended for interoperation with bzr <= 0.14.',
3140
3870
    branch_format='bzrlib.branch.BzrBranchFormat5',
3141
3871
    tree_format='bzrlib.workingtree.WorkingTreeFormat3',
3142
3872
    hidden=True,
3143
3873
    deprecated=True)
3144
 
register_metadir(controldir.format_registry, 'dirstate',
 
3874
format_registry.register_metadir('dirstate',
3145
3875
    'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
3146
3876
    help='New in 0.15: Fast local operations. Compatible with bzr 0.8 and '
3147
3877
        'above when accessed over the network.',
3151
3881
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3152
3882
    hidden=True,
3153
3883
    deprecated=True)
3154
 
register_metadir(controldir.format_registry, 'dirstate-tags',
 
3884
format_registry.register_metadir('dirstate-tags',
3155
3885
    'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
3156
3886
    help='New in 0.15: Fast local operations and improved scaling for '
3157
3887
        'network operations. Additionally adds support for tags.'
3160
3890
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3161
3891
    hidden=True,
3162
3892
    deprecated=True)
3163
 
register_metadir(controldir.format_registry, 'rich-root',
 
3893
format_registry.register_metadir('rich-root',
3164
3894
    'bzrlib.repofmt.knitrepo.RepositoryFormatKnit4',
3165
3895
    help='New in 1.0.  Better handling of tree roots.  Incompatible with'
3166
3896
        ' bzr < 1.0.',
3168
3898
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3169
3899
    hidden=True,
3170
3900
    deprecated=True)
3171
 
register_metadir(controldir.format_registry, 'dirstate-with-subtree',
 
3901
format_registry.register_metadir('dirstate-with-subtree',
3172
3902
    'bzrlib.repofmt.knitrepo.RepositoryFormatKnit3',
3173
3903
    help='New in 0.15: Fast local operations and improved scaling for '
3174
3904
        'network operations. Additionally adds support for versioning nested '
3178
3908
    experimental=True,
3179
3909
    hidden=True,
3180
3910
    )
3181
 
register_metadir(controldir.format_registry, 'pack-0.92',
 
3911
format_registry.register_metadir('pack-0.92',
3182
3912
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack1',
3183
3913
    help='New in 0.92: Pack-based format with data compatible with '
3184
3914
        'dirstate-tags format repositories. Interoperates with '
3187
3917
    branch_format='bzrlib.branch.BzrBranchFormat6',
3188
3918
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3189
3919
    )
3190
 
register_metadir(controldir.format_registry, 'pack-0.92-subtree',
 
3920
format_registry.register_metadir('pack-0.92-subtree',
3191
3921
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack3',
3192
3922
    help='New in 0.92: Pack-based format with data compatible with '
3193
3923
        'dirstate-with-subtree format repositories. Interoperates with '
3198
3928
    hidden=True,
3199
3929
    experimental=True,
3200
3930
    )
3201
 
register_metadir(controldir.format_registry, 'rich-root-pack',
 
3931
format_registry.register_metadir('rich-root-pack',
3202
3932
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack4',
3203
3933
    help='New in 1.0: A variant of pack-0.92 that supports rich-root data '
3204
3934
         '(needed for bzr-svn and bzr-git).',
3206
3936
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3207
3937
    hidden=True,
3208
3938
    )
3209
 
register_metadir(controldir.format_registry, '1.6',
 
3939
format_registry.register_metadir('1.6',
3210
3940
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack5',
3211
3941
    help='A format that allows a branch to indicate that there is another '
3212
3942
         '(stacked) repository that should be used to access data that is '
3215
3945
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3216
3946
    hidden=True,
3217
3947
    )
3218
 
register_metadir(controldir.format_registry, '1.6.1-rich-root',
 
3948
format_registry.register_metadir('1.6.1-rich-root',
3219
3949
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack5RichRoot',
3220
3950
    help='A variant of 1.6 that supports rich-root data '
3221
3951
         '(needed for bzr-svn and bzr-git).',
3223
3953
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3224
3954
    hidden=True,
3225
3955
    )
3226
 
register_metadir(controldir.format_registry, '1.9',
 
3956
format_registry.register_metadir('1.9',
3227
3957
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack6',
3228
3958
    help='A repository format using B+tree indexes. These indexes '
3229
3959
         'are smaller in size, have smarter caching and provide faster '
3232
3962
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3233
3963
    hidden=True,
3234
3964
    )
3235
 
register_metadir(controldir.format_registry, '1.9-rich-root',
 
3965
format_registry.register_metadir('1.9-rich-root',
3236
3966
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack6RichRoot',
3237
3967
    help='A variant of 1.9 that supports rich-root data '
3238
3968
         '(needed for bzr-svn and bzr-git).',
3240
3970
    tree_format='bzrlib.workingtree.WorkingTreeFormat4',
3241
3971
    hidden=True,
3242
3972
    )
3243
 
register_metadir(controldir.format_registry, '1.14',
 
3973
format_registry.register_metadir('1.14',
3244
3974
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack6',
3245
3975
    help='A working-tree format that supports content filtering.',
3246
3976
    branch_format='bzrlib.branch.BzrBranchFormat7',
3247
3977
    tree_format='bzrlib.workingtree.WorkingTreeFormat5',
3248
3978
    )
3249
 
register_metadir(controldir.format_registry, '1.14-rich-root',
 
3979
format_registry.register_metadir('1.14-rich-root',
3250
3980
    'bzrlib.repofmt.pack_repo.RepositoryFormatKnitPack6RichRoot',
3251
3981
    help='A variant of 1.14 that supports rich-root data '
3252
3982
         '(needed for bzr-svn and bzr-git).',
3254
3984
    tree_format='bzrlib.workingtree.WorkingTreeFormat5',
3255
3985
    )
3256
3986
# The following un-numbered 'development' formats should always just be aliases.
3257
 
register_metadir(controldir.format_registry, 'development-rich-root',
 
3987
format_registry.register_metadir('development-rich-root',
3258
3988
    'bzrlib.repofmt.groupcompress_repo.RepositoryFormatCHK1',
3259
3989
    help='Current development format. Supports rich roots. Can convert data '
3260
3990
        'to and from rich-root-pack (and anything compatible with '
3268
3998
    alias=True,
3269
3999
    hidden=True,
3270
4000
    )
3271
 
register_metadir(controldir.format_registry, 'development5-subtree',
 
4001
format_registry.register_metadir('development-subtree',
3272
4002
    'bzrlib.repofmt.pack_repo.RepositoryFormatPackDevelopment2Subtree',
3273
 
    help='Development format, subtree variant. Can convert data to and '
3274
 
        'from pack-0.92-subtree (and anything compatible with '
3275
 
        'pack-0.92-subtree) format repositories. Repositories and branches in '
3276
 
        'this format can only be read by bzr.dev. Please read '
3277
 
        'http://doc.bazaar.canonical.com/latest/developers/development-repo.html '
3278
 
        'before use.',
3279
 
    branch_format='bzrlib.branch.BzrBranchFormat7',
3280
 
    tree_format='bzrlib.workingtree.WorkingTreeFormat6',
3281
 
    experimental=True,
3282
 
    hidden=True,
3283
 
    alias=False,
3284
 
    )
3285
 
 
3286
 
 
3287
 
register_metadir(controldir.format_registry, 'development-subtree',
3288
 
    'bzrlib.repofmt.groupcompress_repo.RepositoryFormat2aSubtree',
3289
4003
    help='Current development format, subtree variant. Can convert data to and '
3290
4004
        'from pack-0.92-subtree (and anything compatible with '
3291
4005
        'pack-0.92-subtree) format repositories. Repositories and branches in '
3302
4016
    )
3303
4017
 
3304
4018
# And the development formats above will have aliased one of the following:
3305
 
register_metadir(controldir.format_registry, 'development6-rich-root',
 
4019
format_registry.register_metadir('development6-rich-root',
3306
4020
    'bzrlib.repofmt.groupcompress_repo.RepositoryFormatCHK1',
3307
4021
    help='pack-1.9 with 255-way hashed CHK inv, group compress, rich roots '
3308
4022
        'Please read '
3314
4028
    experimental=True,
3315
4029
    )
3316
4030
 
3317
 
register_metadir(controldir.format_registry, 'development7-rich-root',
 
4031
format_registry.register_metadir('development7-rich-root',
3318
4032
    'bzrlib.repofmt.groupcompress_repo.RepositoryFormatCHK2',
3319
4033
    help='pack-1.9 with 255-way hashed CHK inv, bencode revision, group compress, '
3320
4034
        'rich roots. Please read '
3326
4040
    experimental=True,
3327
4041
    )
3328
4042
 
3329
 
register_metadir(controldir.format_registry, '2a',
 
4043
format_registry.register_metadir('2a',
3330
4044
    'bzrlib.repofmt.groupcompress_repo.RepositoryFormat2a',
3331
4045
    help='First format for bzr 2.0 series.\n'
3332
4046
        'Uses group-compress storage.\n'
3335
4049
        # 'rich roots. Supported by bzr 1.16 and later.',
3336
4050
    branch_format='bzrlib.branch.BzrBranchFormat7',
3337
4051
    tree_format='bzrlib.workingtree.WorkingTreeFormat6',
3338
 
    experimental=False,
 
4052
    experimental=True,
3339
4053
    )
3340
4054
 
3341
4055
# The following format should be an alias for the rich root equivalent 
3342
4056
# of the default format
3343
 
register_metadir(controldir.format_registry, 'default-rich-root',
 
4057
format_registry.register_metadir('default-rich-root',
3344
4058
    'bzrlib.repofmt.groupcompress_repo.RepositoryFormat2a',
3345
4059
    branch_format='bzrlib.branch.BzrBranchFormat7',
3346
4060
    tree_format='bzrlib.workingtree.WorkingTreeFormat6',
3349
4063
    help='Same as 2a.')
3350
4064
 
3351
4065
# The current format that is made on 'bzr init'.
3352
 
format_name = config.GlobalConfig().get_user_option('default_format')
3353
 
if format_name is None:
3354
 
    controldir.format_registry.set_default('2a')
3355
 
else:
3356
 
    controldir.format_registry.set_default(format_name)
3357
 
 
3358
 
# XXX 2010-08-20 JRV: There is still a lot of code relying on
3359
 
# bzrlib.bzrdir.format_registry existing. When BzrDir.create/BzrDir.open/etc
3360
 
# get changed to ControlDir.create/ControlDir.open/etc this should be removed.
3361
 
format_registry = controldir.format_registry
 
4066
format_registry.set_default('2a')