~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bzrdir.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-02-11 04:02:41 UTC
  • mfrom: (5017.2.2 tariff)
  • Revision ID: pqm@pqm.ubuntu.com-20100211040241-w6n021dz0uus341n
(mbp) add import-tariff tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
86
86
    registry,
87
87
    symbol_versioning,
88
88
    )
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):
 
89
 
 
90
 
 
91
class BzrDir(object):
125
92
    """A .bzr control diretory.
126
93
 
127
94
    BzrDir instances let you create or open any of the things that can be
294
261
                # copied, and finally if we are copying up to a specific
295
262
                # revision_id then we can use the pending-ancestry-result which
296
263
                # does not require traversing all of history to describe it.
297
 
                if (result_repo.user_url == result.user_url
298
 
                    and not require_stacking and
 
264
                if (result_repo.bzrdir.root_transport.base ==
 
265
                    result.root_transport.base and not require_stacking and
299
266
                    revision_id is not None):
300
267
                    fetch_spec = graph.PendingAncestryResult(
301
268
                        [revision_id], local_repo)
388
355
                for subdir in sorted(subdirs, reverse=True):
389
356
                    pending.append(current_transport.clone(subdir))
390
357
 
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
 
 
400
358
    @staticmethod
401
359
    def find_branches(transport):
402
360
        """Find all branches under a transport.
414
372
            except errors.NoRepositoryPresent:
415
373
                pass
416
374
            else:
417
 
                return False, ([], repository)
418
 
            return True, (bzrdir.list_branches(), None)
419
 
        ret = []
420
 
        for branches, repo in BzrDir.find_bzrdirs(transport,
421
 
                                                  evaluate=evaluate):
 
375
                return False, (None, repository)
 
376
            try:
 
377
                branch = bzrdir.open_branch()
 
378
            except errors.NotBranchError:
 
379
                return True, (None, None)
 
380
            else:
 
381
                return True, (branch, None)
 
382
        branches = []
 
383
        for branch, repo in BzrDir.find_bzrdirs(transport, evaluate=evaluate):
422
384
            if repo is not None:
423
 
                ret.extend(repo.find_branches())
424
 
            if branches is not None:
425
 
                ret.extend(branches)
426
 
        return ret
 
385
                branches.extend(repo.find_branches())
 
386
            if branch is not None:
 
387
                branches.append(branch)
 
388
        return branches
427
389
 
428
390
    def destroy_repository(self):
429
391
        """Destroy the repository in this BzrDir"""
430
392
        raise NotImplementedError(self.destroy_repository)
431
393
 
432
 
    def create_branch(self, name=None):
 
394
    def create_branch(self):
433
395
        """Create a branch in this BzrDir.
434
396
 
435
 
        :param name: Name of the colocated branch to create, None for
436
 
            the default branch.
437
 
 
438
397
        The bzrdir's format will control what branch format is created.
439
398
        For more control see BranchFormatXX.create(a_bzrdir).
440
399
        """
441
400
        raise NotImplementedError(self.create_branch)
442
401
 
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
 
        """
 
402
    def destroy_branch(self):
 
403
        """Destroy the branch in this BzrDir"""
449
404
        raise NotImplementedError(self.destroy_branch)
450
405
 
451
406
    @staticmethod
491
446
            stop = False
492
447
            stack_on = config.get_default_stack_on()
493
448
            if stack_on is not None:
494
 
                stack_on_pwd = found_bzrdir.user_url
 
449
                stack_on_pwd = found_bzrdir.root_transport.base
495
450
                stop = True
496
451
            # does it have a repository ?
497
452
            try:
499
454
            except errors.NoRepositoryPresent:
500
455
                repository = None
501
456
            else:
502
 
                if (found_bzrdir.user_url != self.user_url 
503
 
                    and not repository.is_shared()):
 
457
                if ((found_bzrdir.root_transport.base !=
 
458
                     self.root_transport.base) and not repository.is_shared()):
504
459
                    # Don't look higher, can't use a higher shared repo.
505
460
                    repository = None
506
461
                    stop = True
620
575
 
621
576
        :return: Tuple with old path name and new path name
622
577
        """
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
630
 
 
631
 
        backup_dir=name_gen()
632
578
        pb = ui.ui_factory.nested_progress_bar()
633
579
        try:
634
580
            # FIXME: bug 300001 -- the backup fails if the backup directory
635
581
            # already exists, but it should instead either remove it or make
636
582
            # a new backup directory.
637
583
            #
 
584
            # FIXME: bug 262450 -- the backup directory should have the same
 
585
            # permissions as the .bzr directory (probably a bug in copy_tree)
638
586
            old_path = self.root_transport.abspath('.bzr')
639
 
            new_path = self.root_transport.abspath(backup_dir)
 
587
            new_path = self.root_transport.abspath('backup.bzr')
640
588
            ui.ui_factory.note('making backup of %s\n  to %s' % (old_path, new_path,))
641
 
            self.root_transport.copy_tree('.bzr', backup_dir)
 
589
            self.root_transport.copy_tree('.bzr', 'backup.bzr')
642
590
            return (old_path, new_path)
643
591
        finally:
644
592
            pb.finished()
702
650
            if stop:
703
651
                return result
704
652
            next_transport = found_bzrdir.root_transport.clone('..')
705
 
            if (found_bzrdir.user_url == next_transport.base):
 
653
            if (found_bzrdir.root_transport.base == next_transport.base):
706
654
                # top of the file system
707
655
                return None
708
656
            # find the next containing bzrdir
725
673
                repository = found_bzrdir.open_repository()
726
674
            except errors.NoRepositoryPresent:
727
675
                return None, False
728
 
            if found_bzrdir.user_url == self.user_url:
 
676
            if found_bzrdir.root_transport.base == self.root_transport.base:
729
677
                return repository, True
730
678
            elif repository.is_shared():
731
679
                return repository, True
746
694
        """
747
695
        return None
748
696
 
749
 
    def get_branch_transport(self, branch_format, name=None):
 
697
    def get_branch_transport(self, branch_format):
750
698
        """Get the transport for use by branch format in this BzrDir.
751
699
 
752
700
        Note that bzr dirs that do not support format strings will raise
847
795
        :param _transport: the transport this dir is based at.
848
796
        """
849
797
        self._format = _format
850
 
        # these are also under the more standard names of 
851
 
        # control_transport and user_transport
852
798
        self.transport = _transport.clone('.bzr')
853
799
        self.root_transport = _transport
854
800
        self._mode_check_done = False
855
 
        
856
 
    @property 
857
 
    def user_transport(self):
858
 
        return self.root_transport
859
 
        
860
 
    @property
861
 
    def control_transport(self):
862
 
        return self.transport
863
801
 
864
802
    def is_control_filename(self, filename):
865
803
        """True if filename is the name of a path which is reserved for bzrdir's.
940
878
        BzrDir._check_supported(format, _unsupported)
941
879
        return format.open(transport, _found=True)
942
880
 
943
 
    def open_branch(self, name=None, unsupported=False,
944
 
                    ignore_fallbacks=False):
 
881
    def open_branch(self, unsupported=False, ignore_fallbacks=False):
945
882
        """Open the branch object at this BzrDir if one is present.
946
883
 
947
884
        If unsupported is True, then no longer supported branch formats can
1085
1022
        """
1086
1023
        raise NotImplementedError(self.open_workingtree)
1087
1024
 
1088
 
    def has_branch(self, name=None):
 
1025
    def has_branch(self):
1089
1026
        """Tell if this bzrdir contains a branch.
1090
1027
 
1091
1028
        Note: if you're going to open the branch, you should just go ahead
1093
1030
        branch and discards it, and that's somewhat expensive.)
1094
1031
        """
1095
1032
        try:
1096
 
            self.open_branch(name)
 
1033
            self.open_branch()
1097
1034
            return True
1098
1035
        except errors.NotBranchError:
1099
1036
            return False
1373
1310
        self.create_hook(hooks.HookPoint('pre_open',
1374
1311
            "Invoked before attempting to open a BzrDir with the transport "
1375
1312
            "that the open will use.", (1, 14), None))
1376
 
        self.create_hook(hooks.HookPoint('post_repo_init',
1377
 
            "Invoked after a repository has been initialized. "
1378
 
            "post_repo_init is called with a "
1379
 
            "bzrlib.bzrdir.RepoInitHookParams.",
1380
 
            (2, 2), None))
1381
1313
 
1382
1314
# install the default hooks
1383
1315
BzrDir.hooks = BzrDirHooks()
1384
1316
 
1385
1317
 
1386
 
class RepoInitHookParams(object):
1387
 
    """Object holding parameters passed to *_repo_init hooks.
1388
 
 
1389
 
    There are 4 fields that hooks may wish to access:
1390
 
 
1391
 
    :ivar repository: Repository created
1392
 
    :ivar format: Repository format
1393
 
    :ivar bzrdir: The bzrdir for the repository
1394
 
    :ivar shared: The repository is shared
1395
 
    """
1396
 
 
1397
 
    def __init__(self, repository, format, a_bzrdir, shared):
1398
 
        """Create a group of RepoInitHook parameters.
1399
 
 
1400
 
        :param repository: Repository created
1401
 
        :param format: Repository format
1402
 
        :param bzrdir: The bzrdir for the repository
1403
 
        :param shared: The repository is shared
1404
 
        """
1405
 
        self.repository = repository
1406
 
        self.format = format
1407
 
        self.bzrdir = a_bzrdir
1408
 
        self.shared = shared
1409
 
 
1410
 
    def __eq__(self, other):
1411
 
        return self.__dict__ == other.__dict__
1412
 
 
1413
 
    def __repr__(self):
1414
 
        if self.repository:
1415
 
            return "<%s for %s>" % (self.__class__.__name__,
1416
 
                self.repository)
1417
 
        else:
1418
 
            return "<%s for %s>" % (self.__class__.__name__,
1419
 
                self.bzrdir)
1420
 
 
1421
 
 
1422
1318
class BzrDirPreSplitOut(BzrDir):
1423
1319
    """A common class for the all-in-one formats."""
1424
1320
 
1463
1359
            tree.clone(result)
1464
1360
        return result
1465
1361
 
1466
 
    def create_branch(self, name=None):
 
1362
    def create_branch(self):
1467
1363
        """See BzrDir.create_branch."""
1468
 
        return self._format.get_branch_format().initialize(self, name=name)
 
1364
        return self._format.get_branch_format().initialize(self)
1469
1365
 
1470
 
    def destroy_branch(self, name=None):
 
1366
    def destroy_branch(self):
1471
1367
        """See BzrDir.destroy_branch."""
1472
1368
        raise errors.UnsupportedOperation(self.destroy_branch, self)
1473
1369
 
1529
1425
        raise errors.UnsupportedOperation(self.destroy_workingtree_metadata,
1530
1426
                                          self)
1531
1427
 
1532
 
    def get_branch_transport(self, branch_format, name=None):
 
1428
    def get_branch_transport(self, branch_format):
1533
1429
        """See BzrDir.get_branch_transport()."""
1534
 
        if name is not None:
1535
 
            raise errors.NoColocatedBranchSupport(self)
1536
1430
        if branch_format is None:
1537
1431
            return self.transport
1538
1432
        try:
1571
1465
            format = BzrDirFormat.get_default_format()
1572
1466
        return not isinstance(self._format, format.__class__)
1573
1467
 
1574
 
    def open_branch(self, name=None, unsupported=False,
1575
 
                    ignore_fallbacks=False):
 
1468
    def open_branch(self, unsupported=False, ignore_fallbacks=False):
1576
1469
        """See BzrDir.open_branch."""
1577
1470
        from bzrlib.branch import BzrBranchFormat4
1578
1471
        format = BzrBranchFormat4()
1579
1472
        self._check_supported(format, unsupported)
1580
 
        return format.open(self, name, _found=True)
 
1473
        return format.open(self, _found=True)
1581
1474
 
1582
1475
    def sprout(self, url, revision_id=None, force_new_repo=False,
1583
1476
               possible_transports=None, accelerator_tree=None,
1700
1593
        """See BzrDir.can_convert_format()."""
1701
1594
        return True
1702
1595
 
1703
 
    def create_branch(self, name=None):
 
1596
    def create_branch(self):
1704
1597
        """See BzrDir.create_branch."""
1705
 
        return self._format.get_branch_format().initialize(self, name=name)
 
1598
        return self._format.get_branch_format().initialize(self)
1706
1599
 
1707
 
    def destroy_branch(self, name=None):
 
1600
    def destroy_branch(self):
1708
1601
        """See BzrDir.create_branch."""
1709
 
        if name is not None:
1710
 
            raise errors.NoColocatedBranchSupport(self)
1711
1602
        self.transport.delete_tree('branch')
1712
1603
 
1713
1604
    def create_repository(self, shared=False):
1756
1647
        format = BranchFormat.find_format(self)
1757
1648
        return format.get_reference(self)
1758
1649
 
1759
 
    def get_branch_transport(self, branch_format, name=None):
 
1650
    def get_branch_transport(self, branch_format):
1760
1651
        """See BzrDir.get_branch_transport()."""
1761
 
        if name is not None:
1762
 
            raise errors.NoColocatedBranchSupport(self)
1763
1652
        # XXX: this shouldn't implicitly create the directory if it's just
1764
1653
        # promising to get a transport -- mbp 20090727
1765
1654
        if branch_format is None:
1836
1725
                return True
1837
1726
        except errors.NoRepositoryPresent:
1838
1727
            pass
1839
 
        for branch in self.list_branches():
1840
 
            if not isinstance(branch._format,
 
1728
        try:
 
1729
            if not isinstance(self.open_branch()._format,
1841
1730
                              format.get_branch_format().__class__):
1842
1731
                # the branch needs an upgrade.
1843
1732
                return True
 
1733
        except errors.NotBranchError:
 
1734
            pass
1844
1735
        try:
1845
1736
            my_wt = self.open_workingtree(recommend_upgrade=False)
1846
1737
            if not isinstance(my_wt._format,
1851
1742
            pass
1852
1743
        return False
1853
1744
 
1854
 
    def open_branch(self, name=None, unsupported=False,
1855
 
                    ignore_fallbacks=False):
 
1745
    def open_branch(self, unsupported=False, ignore_fallbacks=False):
1856
1746
        """See BzrDir.open_branch."""
1857
1747
        format = self.find_branch_format()
1858
1748
        self._check_supported(format, unsupported)
1859
 
        return format.open(self, name=name,
1860
 
            _found=True, ignore_fallbacks=ignore_fallbacks)
 
1749
        return format.open(self, _found=True, ignore_fallbacks=ignore_fallbacks)
1861
1750
 
1862
1751
    def open_repository(self, unsupported=False):
1863
1752
        """See BzrDir.open_repository."""
1895
1784
    Once a format is deprecated, just deprecate the initialize and open
1896
1785
    methods on the format class. Do not deprecate the object, as the
1897
1786
    object will be created every system load.
1898
 
 
1899
 
    :cvar colocated_branches: Whether this formats supports colocated branches.
1900
1787
    """
1901
1788
 
1902
1789
    _default_format = None
1919
1806
 
1920
1807
    _lock_file_name = 'branch-lock'
1921
1808
 
1922
 
    colocated_branches = False
1923
 
    """Whether co-located branches are supported for this control dir format.
1924
 
    """
1925
 
 
1926
1809
    # _lock_class must be set in subclasses to the lock type, typ.
1927
1810
    # TransportLock or LockDir
1928
1811
 
2735
2618
            if isinstance(self.bzrdir.transport, local.LocalTransport):
2736
2619
                self.bzrdir.get_workingtree_transport(None).delete('stat-cache')
2737
2620
            self._convert_to_weaves()
2738
 
            return BzrDir.open(self.bzrdir.user_url)
 
2621
            return BzrDir.open(self.bzrdir.root_transport.base)
2739
2622
        finally:
2740
2623
            self.pb.finished()
2741
2624
 
2988
2871
        try:
2989
2872
            ui.ui_factory.note('starting upgrade from format 5 to 6')
2990
2873
            self._convert_to_prefixed()
2991
 
            return BzrDir.open(self.bzrdir.user_url)
 
2874
            return BzrDir.open(self.bzrdir.root_transport.base)
2992
2875
        finally:
2993
2876
            pb.finished()
2994
2877
 
3116
2999
            BzrDirMetaFormat1().get_format_string(),
3117
3000
            mode=self.file_mode)
3118
3001
        self.pb.finished()
3119
 
        return BzrDir.open(self.bzrdir.user_url)
 
3002
        return BzrDir.open(self.bzrdir.root_transport.base)
3120
3003
 
3121
3004
    def make_lock(self, name):
3122
3005
        """Make a lock for the new control dir name."""
3171
3054
                ui.ui_factory.note('starting repository conversion')
3172
3055
                converter = CopyConverter(self.target_format.repository_format)
3173
3056
                converter.convert(repo, pb)
3174
 
        for branch in self.bzrdir.list_branches():
 
3057
        try:
 
3058
            branch = self.bzrdir.open_branch()
 
3059
        except errors.NotBranchError:
 
3060
            pass
 
3061
        else:
3175
3062
            # TODO: conversions of Branch and Tree should be done by
3176
3063
            # InterXFormat lookups/some sort of registry.
3177
3064
            # Avoid circular imports
3703
3590
            try:
3704
3591
                stack_on = urlutils.rebase_url(self._stack_on,
3705
3592
                    self._stack_on_pwd,
3706
 
                    branch.user_url)
 
3593
                    branch.bzrdir.root_transport.base)
3707
3594
            except errors.InvalidRebaseURLs:
3708
3595
                stack_on = self._get_full_stack_on()
3709
3596
        try: