1177
867
format.require_stacking()
1180
def checkout_metadir(self):
1181
return self.cloning_metadir()
1183
def sprout(self, url, revision_id=None, force_new_repo=False,
1184
recurse='down', possible_transports=None,
1185
accelerator_tree=None, hardlink=False, stacked=False,
1186
source_branch=None, create_tree_if_local=True):
1187
"""Create a copy of this bzrdir prepared for use as a new line of
1190
If url's last component does not exist, it will be created.
1192
Attributes related to the identity of the source branch like
1193
branch nickname will be cleaned, a working tree is created
1194
whether one existed before or not; and a local branch is always
1197
if revision_id is not None, then the clone operation may tune
1198
itself to download less data.
1199
:param accelerator_tree: A tree which can be used for retrieving file
1200
contents more quickly than the revision tree, i.e. a workingtree.
1201
The revision tree will be used for cases where accelerator_tree's
1202
content is different.
1203
:param hardlink: If true, hard-link files from accelerator_tree,
1205
:param stacked: If true, create a stacked branch referring to the
1206
location of this control directory.
1207
:param create_tree_if_local: If true, a working-tree will be created
1208
when working locally.
1210
target_transport = get_transport(url, possible_transports)
1211
target_transport.ensure_base()
1212
cloning_format = self.cloning_metadir(stacked)
1213
# Create/update the result branch
1214
result = cloning_format.initialize_on_transport(target_transport)
1215
# if a stacked branch wasn't requested, we don't create one
1216
# even if the origin was stacked
1217
stacked_branch_url = None
1218
if source_branch is not None:
1220
stacked_branch_url = self.root_transport.base
1221
source_repository = source_branch.repository
1224
source_branch = self.open_branch()
1225
source_repository = source_branch.repository
1227
stacked_branch_url = self.root_transport.base
1228
except errors.NotBranchError:
1229
source_branch = None
1231
source_repository = self.open_repository()
1232
except errors.NoRepositoryPresent:
1233
source_repository = None
1234
repository_policy = result.determine_repository_policy(
1235
force_new_repo, stacked_branch_url, require_stacking=stacked)
1236
result_repo, is_new_repo = repository_policy.acquire_repository()
1237
if is_new_repo and revision_id is not None and not stacked:
1238
fetch_spec = graph.PendingAncestryResult(
1239
[revision_id], source_repository)
1242
if source_repository is not None:
1243
# Fetch while stacked to prevent unstacked fetch from
1245
if fetch_spec is None:
1246
result_repo.fetch(source_repository, revision_id=revision_id)
1248
result_repo.fetch(source_repository, fetch_spec=fetch_spec)
1250
if source_branch is None:
1251
# this is for sprouting a bzrdir without a branch; is that
1253
# Not especially, but it's part of the contract.
1254
result_branch = result.create_branch()
1256
result_branch = source_branch.sprout(result,
1257
revision_id=revision_id, repository_policy=repository_policy)
1258
mutter("created new branch %r" % (result_branch,))
1260
# Create/update the result working tree
1261
if (create_tree_if_local and
1262
isinstance(target_transport, local.LocalTransport) and
1263
(result_repo is None or result_repo.make_working_trees())):
1264
wt = result.create_workingtree(accelerator_tree=accelerator_tree,
1268
if wt.path2id('') is None:
1270
wt.set_root_id(self.open_workingtree.get_root_id())
1271
except errors.NoWorkingTree:
1277
if recurse == 'down':
1279
basis = wt.basis_tree()
1281
subtrees = basis.iter_references()
1282
elif result_branch is not None:
1283
basis = result_branch.basis_tree()
1285
subtrees = basis.iter_references()
1286
elif source_branch is not None:
1287
basis = source_branch.basis_tree()
1289
subtrees = basis.iter_references()
1294
for path, file_id in subtrees:
1295
target = urlutils.join(url, urlutils.escape(path))
1296
sublocation = source_branch.reference_parent(file_id, path)
1297
sublocation.bzrdir.sprout(target,
1298
basis.get_reference_revision(file_id, path),
1299
force_new_repo=force_new_repo, recurse=recurse,
1302
if basis is not None:
1306
def push_branch(self, source, revision_id=None, overwrite=False,
1307
remember=False, create_prefix=False):
1308
"""Push the source branch into this BzrDir."""
1310
# If we can open a branch, use its direct repository, otherwise see
1311
# if there is a repository without a branch.
1313
br_to = self.open_branch()
1314
except errors.NotBranchError:
1315
# Didn't find a branch, can we find a repository?
1316
repository_to = self.find_repository()
1318
# Found a branch, so we must have found a repository
1319
repository_to = br_to.repository
1321
push_result = PushResult()
1322
push_result.source_branch = source
1324
# We have a repository but no branch, copy the revisions, and then
1326
repository_to.fetch(source.repository, revision_id=revision_id)
1327
br_to = source.clone(self, revision_id=revision_id)
1328
if source.get_push_location() is None or remember:
1329
source.set_push_location(br_to.base)
1330
push_result.stacked_on = None
1331
push_result.branch_push_result = None
1332
push_result.old_revno = None
1333
push_result.old_revid = _mod_revision.NULL_REVISION
1334
push_result.target_branch = br_to
1335
push_result.master_branch = None
1336
push_result.workingtree_updated = False
1338
# We have successfully opened the branch, remember if necessary:
1339
if source.get_push_location() is None or remember:
1340
source.set_push_location(br_to.base)
1342
tree_to = self.open_workingtree()
1343
except errors.NotLocalUrl:
1344
push_result.branch_push_result = source.push(br_to,
1345
overwrite, stop_revision=revision_id)
1346
push_result.workingtree_updated = False
1347
except errors.NoWorkingTree:
1348
push_result.branch_push_result = source.push(br_to,
1349
overwrite, stop_revision=revision_id)
1350
push_result.workingtree_updated = None # Not applicable
1352
tree_to.lock_write()
1354
push_result.branch_push_result = source.push(
1355
tree_to.branch, overwrite, stop_revision=revision_id)
1359
push_result.workingtree_updated = True
1360
push_result.old_revno = push_result.branch_push_result.old_revno
1361
push_result.old_revid = push_result.branch_push_result.old_revid
1362
push_result.target_branch = \
1363
push_result.branch_push_result.target_branch
871
def create(cls, base, format=None, possible_transports=None):
872
"""Create a new BzrDir at the url 'base'.
874
:param format: If supplied, the format of branch to create. If not
875
supplied, the default is used.
876
:param possible_transports: If supplied, a list of transports that
877
can be reused to share a remote connection.
879
if cls is not BzrDir:
880
raise AssertionError("BzrDir.create always creates the"
881
"default format, not one of %r" % cls)
882
t = _mod_transport.get_transport(base, possible_transports)
885
format = controldir.ControlDirFormat.get_default_format()
886
return format.initialize_on_transport(t)
888
def get_branch_transport(self, branch_format, name=None):
889
"""Get the transport for use by branch format in this BzrDir.
891
Note that bzr dirs that do not support format strings will raise
892
IncompatibleFormat if the branch format they are given has
893
a format string, and vice versa.
895
If branch_format is None, the transport is returned with no
896
checking. If it is not None, then the returned transport is
897
guaranteed to point to an existing directory ready for use.
899
raise NotImplementedError(self.get_branch_transport)
901
def get_repository_transport(self, repository_format):
902
"""Get the transport for use by repository format in this BzrDir.
904
Note that bzr dirs that do not support format strings will raise
905
IncompatibleFormat if the repository format they are given has
906
a format string, and vice versa.
908
If repository_format is None, the transport is returned with no
909
checking. If it is not None, then the returned transport is
910
guaranteed to point to an existing directory ready for use.
912
raise NotImplementedError(self.get_repository_transport)
914
def get_workingtree_transport(self, tree_format):
915
"""Get the transport for use by workingtree format in this BzrDir.
917
Note that bzr dirs that do not support format strings will raise
918
IncompatibleFormat if the workingtree format they are given has a
919
format string, and vice versa.
921
If workingtree_format is None, the transport is returned with no
922
checking. If it is not None, then the returned transport is
923
guaranteed to point to an existing directory ready for use.
925
raise NotImplementedError(self.get_workingtree_transport)
1367
928
class BzrDirHooks(hooks.Hooks):
1422
class BzrDirPreSplitOut(BzrDir):
1423
"""A common class for the all-in-one formats."""
1425
def __init__(self, _transport, _format):
1426
"""See BzrDir.__init__."""
1427
super(BzrDirPreSplitOut, self).__init__(_transport, _format)
1428
self._control_files = lockable_files.LockableFiles(
1429
self.get_branch_transport(None),
1430
self._format._lock_file_name,
1431
self._format._lock_class)
1433
def break_lock(self):
1434
"""Pre-splitout bzrdirs do not suffer from stale locks."""
1435
raise NotImplementedError(self.break_lock)
1437
def cloning_metadir(self, require_stacking=False):
1438
"""Produce a metadir suitable for cloning with."""
1439
if require_stacking:
1440
return format_registry.make_bzrdir('1.6')
1441
return self._format.__class__()
1443
def clone(self, url, revision_id=None, force_new_repo=False,
1444
preserve_stacking=False):
1445
"""See BzrDir.clone().
1447
force_new_repo has no effect, since this family of formats always
1448
require a new repository.
1449
preserve_stacking has no effect, since no source branch using this
1450
family of formats can be stacked, so there is no stacking to preserve.
1452
self._make_tail(url)
1453
result = self._format._initialize_for_clone(url)
1454
self.open_repository().clone(result, revision_id=revision_id)
1455
from_branch = self.open_branch()
1456
from_branch.clone(result, revision_id=revision_id)
1458
tree = self.open_workingtree()
1459
except errors.NotLocalUrl:
1460
# make a new one, this format always has to have one.
1461
result._init_workingtree()
1466
def create_branch(self, name=None):
1467
"""See BzrDir.create_branch."""
1468
return self._format.get_branch_format().initialize(self, name=name)
1470
def destroy_branch(self, name=None):
1471
"""See BzrDir.destroy_branch."""
1472
raise errors.UnsupportedOperation(self.destroy_branch, self)
1474
def create_repository(self, shared=False):
1475
"""See BzrDir.create_repository."""
1477
raise errors.IncompatibleFormat('shared repository', self._format)
1478
return self.open_repository()
1480
def destroy_repository(self):
1481
"""See BzrDir.destroy_repository."""
1482
raise errors.UnsupportedOperation(self.destroy_repository, self)
1484
def create_workingtree(self, revision_id=None, from_branch=None,
1485
accelerator_tree=None, hardlink=False):
1486
"""See BzrDir.create_workingtree."""
1487
# The workingtree is sometimes created when the bzrdir is created,
1488
# but not when cloning.
1490
# this looks buggy but is not -really-
1491
# because this format creates the workingtree when the bzrdir is
1493
# clone and sprout will have set the revision_id
1494
# and that will have set it for us, its only
1495
# specific uses of create_workingtree in isolation
1496
# that can do wonky stuff here, and that only
1497
# happens for creating checkouts, which cannot be
1498
# done on this format anyway. So - acceptable wart.
1500
warning("can't support hardlinked working trees in %r"
1503
result = self.open_workingtree(recommend_upgrade=False)
1504
except errors.NoSuchFile:
1505
result = self._init_workingtree()
1506
if revision_id is not None:
1507
if revision_id == _mod_revision.NULL_REVISION:
1508
result.set_parent_ids([])
1510
result.set_parent_ids([revision_id])
1513
def _init_workingtree(self):
1514
from bzrlib.workingtree import WorkingTreeFormat2
1516
return WorkingTreeFormat2().initialize(self)
1517
except errors.NotLocalUrl:
1518
# Even though we can't access the working tree, we need to
1519
# create its control files.
1520
return WorkingTreeFormat2()._stub_initialize_on_transport(
1521
self.transport, self._control_files._file_mode)
1523
def destroy_workingtree(self):
1524
"""See BzrDir.destroy_workingtree."""
1525
raise errors.UnsupportedOperation(self.destroy_workingtree, self)
1527
def destroy_workingtree_metadata(self):
1528
"""See BzrDir.destroy_workingtree_metadata."""
1529
raise errors.UnsupportedOperation(self.destroy_workingtree_metadata,
1532
def get_branch_transport(self, branch_format, name=None):
1533
"""See BzrDir.get_branch_transport()."""
1534
if name is not None:
1535
raise errors.NoColocatedBranchSupport(self)
1536
if branch_format is None:
1537
return self.transport
1539
branch_format.get_format_string()
1540
except NotImplementedError:
1541
return self.transport
1542
raise errors.IncompatibleFormat(branch_format, self._format)
1544
def get_repository_transport(self, repository_format):
1545
"""See BzrDir.get_repository_transport()."""
1546
if repository_format is None:
1547
return self.transport
1549
repository_format.get_format_string()
1550
except NotImplementedError:
1551
return self.transport
1552
raise errors.IncompatibleFormat(repository_format, self._format)
1554
def get_workingtree_transport(self, workingtree_format):
1555
"""See BzrDir.get_workingtree_transport()."""
1556
if workingtree_format is None:
1557
return self.transport
1559
workingtree_format.get_format_string()
1560
except NotImplementedError:
1561
return self.transport
1562
raise errors.IncompatibleFormat(workingtree_format, self._format)
1564
def needs_format_conversion(self, format=None):
1565
"""See BzrDir.needs_format_conversion()."""
1566
# if the format is not the same as the system default,
1567
# an upgrade is needed.
1569
symbol_versioning.warn(symbol_versioning.deprecated_in((1, 13, 0))
1570
% 'needs_format_conversion(format=None)')
1571
format = BzrDirFormat.get_default_format()
1572
return not isinstance(self._format, format.__class__)
1574
def open_branch(self, name=None, unsupported=False,
1575
ignore_fallbacks=False):
1576
"""See BzrDir.open_branch."""
1577
from bzrlib.branch import BzrBranchFormat4
1578
format = BzrBranchFormat4()
1579
self._check_supported(format, unsupported)
1580
return format.open(self, name, _found=True)
1582
def sprout(self, url, revision_id=None, force_new_repo=False,
1583
possible_transports=None, accelerator_tree=None,
1584
hardlink=False, stacked=False, create_tree_if_local=True,
1585
source_branch=None):
1586
"""See BzrDir.sprout()."""
1587
if source_branch is not None:
1588
my_branch = self.open_branch()
1589
if source_branch.base != my_branch.base:
1590
raise AssertionError(
1591
"source branch %r is not within %r with branch %r" %
1592
(source_branch, self, my_branch))
1594
raise errors.UnstackableBranchFormat(
1595
self._format, self.root_transport.base)
1596
if not create_tree_if_local:
1597
raise errors.MustHaveWorkingTree(
1598
self._format, self.root_transport.base)
1599
from bzrlib.workingtree import WorkingTreeFormat2
1600
self._make_tail(url)
1601
result = self._format._initialize_for_clone(url)
1603
self.open_repository().clone(result, revision_id=revision_id)
1604
except errors.NoRepositoryPresent:
1607
self.open_branch().sprout(result, revision_id=revision_id)
1608
except errors.NotBranchError:
1611
# we always want a working tree
1612
WorkingTreeFormat2().initialize(result,
1613
accelerator_tree=accelerator_tree,
1618
class BzrDir4(BzrDirPreSplitOut):
1619
"""A .bzr version 4 control object.
1621
This is a deprecated format and may be removed after sept 2006.
1624
def create_repository(self, shared=False):
1625
"""See BzrDir.create_repository."""
1626
return self._format.repository_format.initialize(self, shared)
1628
def needs_format_conversion(self, format=None):
1629
"""Format 4 dirs are always in need of conversion."""
1631
symbol_versioning.warn(symbol_versioning.deprecated_in((1, 13, 0))
1632
% 'needs_format_conversion(format=None)')
1635
def open_repository(self):
1636
"""See BzrDir.open_repository."""
1637
from bzrlib.repofmt.weaverepo import RepositoryFormat4
1638
return RepositoryFormat4().open(self, _found=True)
1641
class BzrDir5(BzrDirPreSplitOut):
1642
"""A .bzr version 5 control object.
1644
This is a deprecated format and may be removed after sept 2006.
1647
def has_workingtree(self):
1648
"""See BzrDir.has_workingtree."""
1651
def open_repository(self):
1652
"""See BzrDir.open_repository."""
1653
from bzrlib.repofmt.weaverepo import RepositoryFormat5
1654
return RepositoryFormat5().open(self, _found=True)
1656
def open_workingtree(self, _unsupported=False,
1657
recommend_upgrade=True):
1658
"""See BzrDir.create_workingtree."""
1659
from bzrlib.workingtree import WorkingTreeFormat2
1660
wt_format = WorkingTreeFormat2()
1661
# we don't warn here about upgrades; that ought to be handled for the
1663
return wt_format.open(self, _found=True)
1666
class BzrDir6(BzrDirPreSplitOut):
1667
"""A .bzr version 6 control object.
1669
This is a deprecated format and may be removed after sept 2006.
1672
def has_workingtree(self):
1673
"""See BzrDir.has_workingtree."""
1676
def open_repository(self):
1677
"""See BzrDir.open_repository."""
1678
from bzrlib.repofmt.weaverepo import RepositoryFormat6
1679
return RepositoryFormat6().open(self, _found=True)
1681
def open_workingtree(self, _unsupported=False,
1682
recommend_upgrade=True):
1683
"""See BzrDir.create_workingtree."""
1684
# we don't warn here about upgrades; that ought to be handled for the
1686
from bzrlib.workingtree import WorkingTreeFormat2
1687
return WorkingTreeFormat2().open(self, _found=True)
1690
983
class BzrDirMeta1(BzrDir):
1691
984
"""A .bzr meta version 1 control object.
2266
def unregister_format(klass, format):
2267
del klass._formats[format.get_format_string()]
2270
def unregister_control_format(klass, format):
2271
klass._control_formats.remove(format)
2274
class BzrDirFormat4(BzrDirFormat):
2275
"""Bzr dir format 4.
2277
This format is a combined format for working tree, branch and repository.
2279
- Format 1 working trees [always]
2280
- Format 4 branches [always]
2281
- Format 4 repositories [always]
2283
This format is deprecated: it indexes texts using a text it which is
2284
removed in format 5; write support for this format has been removed.
2287
_lock_class = lockable_files.TransportLock
2289
def get_format_string(self):
2290
"""See BzrDirFormat.get_format_string()."""
2291
return "Bazaar-NG branch, format 0.0.4\n"
2293
def get_format_description(self):
2294
"""See BzrDirFormat.get_format_description()."""
2295
return "All-in-one format 4"
2297
def get_converter(self, format=None):
2298
"""See BzrDirFormat.get_converter()."""
2299
# there is one and only one upgrade path here.
2300
return ConvertBzrDir4To5()
2302
def initialize_on_transport(self, transport):
2303
"""Format 4 branches cannot be created."""
2304
raise errors.UninitializableFormat(self)
2306
def is_supported(self):
2307
"""Format 4 is not supported.
2309
It is not supported because the model changed from 4 to 5 and the
2310
conversion logic is expensive - so doing it on the fly was not
2315
def network_name(self):
2316
return self.get_format_string()
2318
def _open(self, transport):
2319
"""See BzrDirFormat._open."""
2320
return BzrDir4(transport, self)
2322
def __return_repository_format(self):
2323
"""Circular import protection."""
2324
from bzrlib.repofmt.weaverepo import RepositoryFormat4
2325
return RepositoryFormat4()
2326
repository_format = property(__return_repository_format)
2329
class BzrDirFormatAllInOne(BzrDirFormat):
2330
"""Common class for formats before meta-dirs."""
2332
def initialize_on_transport_ex(self, transport, use_existing_dir=False,
2333
create_prefix=False, force_new_repo=False, stacked_on=None,
2334
stack_on_pwd=None, repo_format_name=None, make_working_trees=None,
2336
"""See BzrDirFormat.initialize_on_transport_ex."""
2337
require_stacking = (stacked_on is not None)
2338
# Format 5 cannot stack, but we've been asked to - actually init
2340
if require_stacking:
2341
format = BzrDirMetaFormat1()
2342
return format.initialize_on_transport_ex(transport,
2343
use_existing_dir=use_existing_dir, create_prefix=create_prefix,
2344
force_new_repo=force_new_repo, stacked_on=stacked_on,
2345
stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
2346
make_working_trees=make_working_trees, shared_repo=shared_repo)
2347
return BzrDirFormat.initialize_on_transport_ex(self, transport,
2348
use_existing_dir=use_existing_dir, create_prefix=create_prefix,
2349
force_new_repo=force_new_repo, stacked_on=stacked_on,
2350
stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
2351
make_working_trees=make_working_trees, shared_repo=shared_repo)
2354
class BzrDirFormat5(BzrDirFormatAllInOne):
2355
"""Bzr control format 5.
2357
This format is a combined format for working tree, branch and repository.
2359
- Format 2 working trees [always]
2360
- Format 4 branches [always]
2361
- Format 5 repositories [always]
2362
Unhashed stores in the repository.
2365
_lock_class = lockable_files.TransportLock
2367
def get_format_string(self):
2368
"""See BzrDirFormat.get_format_string()."""
2369
return "Bazaar-NG branch, format 5\n"
2371
def get_branch_format(self):
2372
from bzrlib import branch
2373
return branch.BzrBranchFormat4()
2375
def get_format_description(self):
2376
"""See BzrDirFormat.get_format_description()."""
2377
return "All-in-one format 5"
2379
def get_converter(self, format=None):
2380
"""See BzrDirFormat.get_converter()."""
2381
# there is one and only one upgrade path here.
2382
return ConvertBzrDir5To6()
2384
def _initialize_for_clone(self, url):
2385
return self.initialize_on_transport(get_transport(url), _cloning=True)
2387
def initialize_on_transport(self, transport, _cloning=False):
2388
"""Format 5 dirs always have working tree, branch and repository.
2390
Except when they are being cloned.
2392
from bzrlib.branch import BzrBranchFormat4
2393
from bzrlib.repofmt.weaverepo import RepositoryFormat5
2394
result = (super(BzrDirFormat5, self).initialize_on_transport(transport))
2395
RepositoryFormat5().initialize(result, _internal=True)
2397
branch = BzrBranchFormat4().initialize(result)
2398
result._init_workingtree()
2401
def network_name(self):
2402
return self.get_format_string()
2404
def _open(self, transport):
2405
"""See BzrDirFormat._open."""
2406
return BzrDir5(transport, self)
2408
def __return_repository_format(self):
2409
"""Circular import protection."""
2410
from bzrlib.repofmt.weaverepo import RepositoryFormat5
2411
return RepositoryFormat5()
2412
repository_format = property(__return_repository_format)
2415
class BzrDirFormat6(BzrDirFormatAllInOne):
2416
"""Bzr control format 6.
2418
This format is a combined format for working tree, branch and repository.
2420
- Format 2 working trees [always]
2421
- Format 4 branches [always]
2422
- Format 6 repositories [always]
2425
_lock_class = lockable_files.TransportLock
2427
def get_format_string(self):
2428
"""See BzrDirFormat.get_format_string()."""
2429
return "Bazaar-NG branch, format 6\n"
2431
def get_format_description(self):
2432
"""See BzrDirFormat.get_format_description()."""
2433
return "All-in-one format 6"
2435
def get_branch_format(self):
2436
from bzrlib import branch
2437
return branch.BzrBranchFormat4()
2439
def get_converter(self, format=None):
2440
"""See BzrDirFormat.get_converter()."""
2441
# there is one and only one upgrade path here.
2442
return ConvertBzrDir6ToMeta()
2444
def _initialize_for_clone(self, url):
2445
return self.initialize_on_transport(get_transport(url), _cloning=True)
2447
def initialize_on_transport(self, transport, _cloning=False):
2448
"""Format 6 dirs always have working tree, branch and repository.
2450
Except when they are being cloned.
2452
from bzrlib.branch import BzrBranchFormat4
2453
from bzrlib.repofmt.weaverepo import RepositoryFormat6
2454
result = super(BzrDirFormat6, self).initialize_on_transport(transport)
2455
RepositoryFormat6().initialize(result, _internal=True)
2457
branch = BzrBranchFormat4().initialize(result)
2458
result._init_workingtree()
2461
def network_name(self):
2462
return self.get_format_string()
2464
def _open(self, transport):
2465
"""See BzrDirFormat._open."""
2466
return BzrDir6(transport, self)
2468
def __return_repository_format(self):
2469
"""Circular import protection."""
2470
from bzrlib.repofmt.weaverepo import RepositoryFormat6
2471
return RepositoryFormat6()
2472
repository_format = property(__return_repository_format)
2475
1463
class BzrDirMetaFormat1(BzrDirFormat):
2476
1464
"""Bzr meta control format 1
2677
1671
__set_workingtree_format)
2680
network_format_registry = registry.FormatRegistry()
2681
"""Registry of formats indexed by their network name.
2683
The network name for a BzrDirFormat is an identifier that can be used when
2684
referring to formats with smart server operations. See
2685
BzrDirFormat.network_name() for more detail.
2689
# Register bzr control format
2690
BzrDirFormat.register_control_format(BzrDirFormat)
2692
1674
# Register bzr formats
2693
BzrDirFormat.register_format(BzrDirFormat4())
2694
BzrDirFormat.register_format(BzrDirFormat5())
2695
BzrDirFormat.register_format(BzrDirFormat6())
2696
__default_format = BzrDirMetaFormat1()
2697
BzrDirFormat.register_format(__default_format)
2698
BzrDirFormat._default_format = __default_format
2701
class Converter(object):
2702
"""Converts a disk format object from one format to another."""
2704
def convert(self, to_convert, pb):
2705
"""Perform the conversion of to_convert, giving feedback via pb.
2707
:param to_convert: The disk object to convert.
2708
:param pb: a progress bar to use for progress information.
2711
def step(self, message):
2712
"""Update the pb by a step."""
2714
self.pb.update(message, self.count, self.total)
2717
class ConvertBzrDir4To5(Converter):
2718
"""Converts format 4 bzr dirs to format 5."""
2721
super(ConvertBzrDir4To5, self).__init__()
2722
self.converted_revs = set()
2723
self.absent_revisions = set()
2727
def convert(self, to_convert, pb):
2728
"""See Converter.convert()."""
2729
self.bzrdir = to_convert
2731
warnings.warn("pb parameter to convert() is deprecated")
2732
self.pb = ui.ui_factory.nested_progress_bar()
2734
ui.ui_factory.note('starting upgrade from format 4 to 5')
2735
if isinstance(self.bzrdir.transport, local.LocalTransport):
2736
self.bzrdir.get_workingtree_transport(None).delete('stat-cache')
2737
self._convert_to_weaves()
2738
return BzrDir.open(self.bzrdir.user_url)
2742
def _convert_to_weaves(self):
2743
ui.ui_factory.note('note: upgrade may be faster if all store files are ungzipped first')
2746
stat = self.bzrdir.transport.stat('weaves')
2747
if not S_ISDIR(stat.st_mode):
2748
self.bzrdir.transport.delete('weaves')
2749
self.bzrdir.transport.mkdir('weaves')
2750
except errors.NoSuchFile:
2751
self.bzrdir.transport.mkdir('weaves')
2752
# deliberately not a WeaveFile as we want to build it up slowly.
2753
self.inv_weave = Weave('inventory')
2754
# holds in-memory weaves for all files
2755
self.text_weaves = {}
2756
self.bzrdir.transport.delete('branch-format')
2757
self.branch = self.bzrdir.open_branch()
2758
self._convert_working_inv()
2759
rev_history = self.branch.revision_history()
2760
# to_read is a stack holding the revisions we still need to process;
2761
# appending to it adds new highest-priority revisions
2762
self.known_revisions = set(rev_history)
2763
self.to_read = rev_history[-1:]
2765
rev_id = self.to_read.pop()
2766
if (rev_id not in self.revisions
2767
and rev_id not in self.absent_revisions):
2768
self._load_one_rev(rev_id)
2770
to_import = self._make_order()
2771
for i, rev_id in enumerate(to_import):
2772
self.pb.update('converting revision', i, len(to_import))
2773
self._convert_one_rev(rev_id)
2775
self._write_all_weaves()
2776
self._write_all_revs()
2777
ui.ui_factory.note('upgraded to weaves:')
2778
ui.ui_factory.note(' %6d revisions and inventories' % len(self.revisions))
2779
ui.ui_factory.note(' %6d revisions not present' % len(self.absent_revisions))
2780
ui.ui_factory.note(' %6d texts' % self.text_count)
2781
self._cleanup_spare_files_after_format4()
2782
self.branch._transport.put_bytes(
2784
BzrDirFormat5().get_format_string(),
2785
mode=self.bzrdir._get_file_mode())
2787
def _cleanup_spare_files_after_format4(self):
2788
# FIXME working tree upgrade foo.
2789
for n in 'merged-patches', 'pending-merged-patches':
2791
## assert os.path.getsize(p) == 0
2792
self.bzrdir.transport.delete(n)
2793
except errors.NoSuchFile:
2795
self.bzrdir.transport.delete_tree('inventory-store')
2796
self.bzrdir.transport.delete_tree('text-store')
2798
def _convert_working_inv(self):
2799
inv = xml4.serializer_v4.read_inventory(
2800
self.branch._transport.get('inventory'))
2801
new_inv_xml = xml5.serializer_v5.write_inventory_to_string(inv, working=True)
2802
self.branch._transport.put_bytes('inventory', new_inv_xml,
2803
mode=self.bzrdir._get_file_mode())
2805
def _write_all_weaves(self):
2806
controlweaves = WeaveStore(self.bzrdir.transport, prefixed=False)
2807
weave_transport = self.bzrdir.transport.clone('weaves')
2808
weaves = WeaveStore(weave_transport, prefixed=False)
2809
transaction = WriteTransaction()
2813
for file_id, file_weave in self.text_weaves.items():
2814
self.pb.update('writing weave', i, len(self.text_weaves))
2815
weaves._put_weave(file_id, file_weave, transaction)
2817
self.pb.update('inventory', 0, 1)
2818
controlweaves._put_weave('inventory', self.inv_weave, transaction)
2819
self.pb.update('inventory', 1, 1)
2823
def _write_all_revs(self):
2824
"""Write all revisions out in new form."""
2825
self.bzrdir.transport.delete_tree('revision-store')
2826
self.bzrdir.transport.mkdir('revision-store')
2827
revision_transport = self.bzrdir.transport.clone('revision-store')
2829
from bzrlib.xml5 import serializer_v5
2830
from bzrlib.repofmt.weaverepo import RevisionTextStore
2831
revision_store = RevisionTextStore(revision_transport,
2832
serializer_v5, False, versionedfile.PrefixMapper(),
2833
lambda:True, lambda:True)
2835
for i, rev_id in enumerate(self.converted_revs):
2836
self.pb.update('write revision', i, len(self.converted_revs))
2837
text = serializer_v5.write_revision_to_string(
2838
self.revisions[rev_id])
2840
revision_store.add_lines(key, None, osutils.split_lines(text))
2844
def _load_one_rev(self, rev_id):
2845
"""Load a revision object into memory.
2847
Any parents not either loaded or abandoned get queued to be
2849
self.pb.update('loading revision',
2850
len(self.revisions),
2851
len(self.known_revisions))
2852
if not self.branch.repository.has_revision(rev_id):
2854
ui.ui_factory.note('revision {%s} not present in branch; '
2855
'will be converted as a ghost' %
2857
self.absent_revisions.add(rev_id)
2859
rev = self.branch.repository.get_revision(rev_id)
2860
for parent_id in rev.parent_ids:
2861
self.known_revisions.add(parent_id)
2862
self.to_read.append(parent_id)
2863
self.revisions[rev_id] = rev
2865
def _load_old_inventory(self, rev_id):
2866
old_inv_xml = self.branch.repository.inventory_store.get(rev_id).read()
2867
inv = xml4.serializer_v4.read_inventory_from_string(old_inv_xml)
2868
inv.revision_id = rev_id
2869
rev = self.revisions[rev_id]
2872
def _load_updated_inventory(self, rev_id):
2873
inv_xml = self.inv_weave.get_text(rev_id)
2874
inv = xml5.serializer_v5.read_inventory_from_string(inv_xml, rev_id)
2877
def _convert_one_rev(self, rev_id):
2878
"""Convert revision and all referenced objects to new format."""
2879
rev = self.revisions[rev_id]
2880
inv = self._load_old_inventory(rev_id)
2881
present_parents = [p for p in rev.parent_ids
2882
if p not in self.absent_revisions]
2883
self._convert_revision_contents(rev, inv, present_parents)
2884
self._store_new_inv(rev, inv, present_parents)
2885
self.converted_revs.add(rev_id)
2887
def _store_new_inv(self, rev, inv, present_parents):
2888
new_inv_xml = xml5.serializer_v5.write_inventory_to_string(inv)
2889
new_inv_sha1 = sha_string(new_inv_xml)
2890
self.inv_weave.add_lines(rev.revision_id,
2892
new_inv_xml.splitlines(True))
2893
rev.inventory_sha1 = new_inv_sha1
2895
def _convert_revision_contents(self, rev, inv, present_parents):
2896
"""Convert all the files within a revision.
2898
Also upgrade the inventory to refer to the text revision ids."""
2899
rev_id = rev.revision_id
2900
mutter('converting texts of revision {%s}',
2902
parent_invs = map(self._load_updated_inventory, present_parents)
2903
entries = inv.iter_entries()
2905
for path, ie in entries:
2906
self._convert_file_version(rev, ie, parent_invs)
2908
def _convert_file_version(self, rev, ie, parent_invs):
2909
"""Convert one version of one file.
2911
The file needs to be added into the weave if it is a merge
2912
of >=2 parents or if it's changed from its parent.
2914
file_id = ie.file_id
2915
rev_id = rev.revision_id
2916
w = self.text_weaves.get(file_id)
2919
self.text_weaves[file_id] = w
2920
text_changed = False
2921
parent_candiate_entries = ie.parent_candidates(parent_invs)
2922
heads = graph.Graph(self).heads(parent_candiate_entries.keys())
2923
# XXX: Note that this is unordered - and this is tolerable because
2924
# the previous code was also unordered.
2925
previous_entries = dict((head, parent_candiate_entries[head]) for head
2927
self.snapshot_ie(previous_entries, ie, w, rev_id)
2930
def get_parent_map(self, revision_ids):
2931
"""See graph.StackedParentsProvider.get_parent_map"""
2932
return dict((revision_id, self.revisions[revision_id])
2933
for revision_id in revision_ids
2934
if revision_id in self.revisions)
2936
def snapshot_ie(self, previous_revisions, ie, w, rev_id):
2937
# TODO: convert this logic, which is ~= snapshot to
2938
# a call to:. This needs the path figured out. rather than a work_tree
2939
# a v4 revision_tree can be given, or something that looks enough like
2940
# one to give the file content to the entry if it needs it.
2941
# and we need something that looks like a weave store for snapshot to
2943
#ie.snapshot(rev, PATH, previous_revisions, REVISION_TREE, InMemoryWeaveStore(self.text_weaves))
2944
if len(previous_revisions) == 1:
2945
previous_ie = previous_revisions.values()[0]
2946
if ie._unchanged(previous_ie):
2947
ie.revision = previous_ie.revision
2950
text = self.branch.repository._text_store.get(ie.text_id)
2951
file_lines = text.readlines()
2952
w.add_lines(rev_id, previous_revisions, file_lines)
2953
self.text_count += 1
2955
w.add_lines(rev_id, previous_revisions, [])
2956
ie.revision = rev_id
2958
def _make_order(self):
2959
"""Return a suitable order for importing revisions.
2961
The order must be such that an revision is imported after all
2962
its (present) parents.
2964
todo = set(self.revisions.keys())
2965
done = self.absent_revisions.copy()
2968
# scan through looking for a revision whose parents
2970
for rev_id in sorted(list(todo)):
2971
rev = self.revisions[rev_id]
2972
parent_ids = set(rev.parent_ids)
2973
if parent_ids.issubset(done):
2974
# can take this one now
2975
order.append(rev_id)
2981
class ConvertBzrDir5To6(Converter):
2982
"""Converts format 5 bzr dirs to format 6."""
2984
def convert(self, to_convert, pb):
2985
"""See Converter.convert()."""
2986
self.bzrdir = to_convert
2987
pb = ui.ui_factory.nested_progress_bar()
2989
ui.ui_factory.note('starting upgrade from format 5 to 6')
2990
self._convert_to_prefixed()
2991
return BzrDir.open(self.bzrdir.user_url)
2995
def _convert_to_prefixed(self):
2996
from bzrlib.store import TransportStore
2997
self.bzrdir.transport.delete('branch-format')
2998
for store_name in ["weaves", "revision-store"]:
2999
ui.ui_factory.note("adding prefixes to %s" % store_name)
3000
store_transport = self.bzrdir.transport.clone(store_name)
3001
store = TransportStore(store_transport, prefixed=True)
3002
for urlfilename in store_transport.list_dir('.'):
3003
filename = urlutils.unescape(urlfilename)
3004
if (filename.endswith(".weave") or
3005
filename.endswith(".gz") or
3006
filename.endswith(".sig")):
3007
file_id, suffix = os.path.splitext(filename)
3011
new_name = store._mapper.map((file_id,)) + suffix
3012
# FIXME keep track of the dirs made RBC 20060121
3014
store_transport.move(filename, new_name)
3015
except errors.NoSuchFile: # catches missing dirs strangely enough
3016
store_transport.mkdir(osutils.dirname(new_name))
3017
store_transport.move(filename, new_name)
3018
self.bzrdir.transport.put_bytes(
3020
BzrDirFormat6().get_format_string(),
3021
mode=self.bzrdir._get_file_mode())
3024
class ConvertBzrDir6ToMeta(Converter):
3025
"""Converts format 6 bzr dirs to metadirs."""
3027
def convert(self, to_convert, pb):
3028
"""See Converter.convert()."""
3029
from bzrlib.repofmt.weaverepo import RepositoryFormat7
3030
from bzrlib.branch import BzrBranchFormat5
3031
self.bzrdir = to_convert
3032
self.pb = ui.ui_factory.nested_progress_bar()
3034
self.total = 20 # the steps we know about
3035
self.garbage_inventories = []
3036
self.dir_mode = self.bzrdir._get_dir_mode()
3037
self.file_mode = self.bzrdir._get_file_mode()
3039
ui.ui_factory.note('starting upgrade from format 6 to metadir')
3040
self.bzrdir.transport.put_bytes(
3042
"Converting to format 6",
3043
mode=self.file_mode)
3044
# its faster to move specific files around than to open and use the apis...
3045
# first off, nuke ancestry.weave, it was never used.
3047
self.step('Removing ancestry.weave')
3048
self.bzrdir.transport.delete('ancestry.weave')
3049
except errors.NoSuchFile:
3051
# find out whats there
3052
self.step('Finding branch files')
3053
last_revision = self.bzrdir.open_branch().last_revision()
3054
bzrcontents = self.bzrdir.transport.list_dir('.')
3055
for name in bzrcontents:
3056
if name.startswith('basis-inventory.'):
3057
self.garbage_inventories.append(name)
3058
# create new directories for repository, working tree and branch
3059
repository_names = [('inventory.weave', True),
3060
('revision-store', True),
3062
self.step('Upgrading repository ')
3063
self.bzrdir.transport.mkdir('repository', mode=self.dir_mode)
3064
self.make_lock('repository')
3065
# we hard code the formats here because we are converting into
3066
# the meta format. The meta format upgrader can take this to a
3067
# future format within each component.
3068
self.put_format('repository', RepositoryFormat7())
3069
for entry in repository_names:
3070
self.move_entry('repository', entry)
3072
self.step('Upgrading branch ')
3073
self.bzrdir.transport.mkdir('branch', mode=self.dir_mode)
3074
self.make_lock('branch')
3075
self.put_format('branch', BzrBranchFormat5())
3076
branch_files = [('revision-history', True),
3077
('branch-name', True),
3079
for entry in branch_files:
3080
self.move_entry('branch', entry)
3082
checkout_files = [('pending-merges', True),
3083
('inventory', True),
3084
('stat-cache', False)]
3085
# If a mandatory checkout file is not present, the branch does not have
3086
# a functional checkout. Do not create a checkout in the converted
3088
for name, mandatory in checkout_files:
3089
if mandatory and name not in bzrcontents:
3090
has_checkout = False
3094
if not has_checkout:
3095
ui.ui_factory.note('No working tree.')
3096
# If some checkout files are there, we may as well get rid of them.
3097
for name, mandatory in checkout_files:
3098
if name in bzrcontents:
3099
self.bzrdir.transport.delete(name)
3101
from bzrlib.workingtree import WorkingTreeFormat3
3102
self.step('Upgrading working tree')
3103
self.bzrdir.transport.mkdir('checkout', mode=self.dir_mode)
3104
self.make_lock('checkout')
3106
'checkout', WorkingTreeFormat3())
3107
self.bzrdir.transport.delete_multi(
3108
self.garbage_inventories, self.pb)
3109
for entry in checkout_files:
3110
self.move_entry('checkout', entry)
3111
if last_revision is not None:
3112
self.bzrdir.transport.put_bytes(
3113
'checkout/last-revision', last_revision)
3114
self.bzrdir.transport.put_bytes(
3116
BzrDirMetaFormat1().get_format_string(),
3117
mode=self.file_mode)
3119
return BzrDir.open(self.bzrdir.user_url)
3121
def make_lock(self, name):
3122
"""Make a lock for the new control dir name."""
3123
self.step('Make %s lock' % name)
3124
ld = lockdir.LockDir(self.bzrdir.transport,
3126
file_modebits=self.file_mode,
3127
dir_modebits=self.dir_mode)
3130
def move_entry(self, new_dir, entry):
3131
"""Move then entry name into new_dir."""
3133
mandatory = entry[1]
3134
self.step('Moving %s' % name)
3136
self.bzrdir.transport.move(name, '%s/%s' % (new_dir, name))
3137
except errors.NoSuchFile:
3141
def put_format(self, dirname, format):
3142
self.bzrdir.transport.put_bytes('%s/format' % dirname,
3143
format.get_format_string(),
3147
class ConvertMetaToMeta(Converter):
1675
BzrProber.formats.register_lazy(
1676
"Bazaar-NG branch, format 0.0.4\n", "bzrlib.bzrdir_weave",
1678
BzrProber.formats.register_lazy(
1679
"Bazaar-NG branch, format 5\n", "bzrlib.bzrdir_weave",
1681
BzrProber.formats.register_lazy(
1682
"Bazaar-NG branch, format 6\n", "bzrlib.bzrdir_weave",
1684
BzrProber.formats.register(BzrDirMetaFormat1.get_format_string(),
1686
controldir.ControlDirFormat._default_format = BzrDirMetaFormat1()
1689
class ConvertMetaToMeta(controldir.Converter):
3148
1690
"""Converts the components of metadirs."""
3150
1692
def __init__(self, target_format):
3223
1765
return to_convert
3226
# This is not in remote.py because it's relatively small, and needs to be
3227
# registered. Putting it in remote.py creates a circular import problem.
3228
# we can make it a lazy object if the control formats is turned into something
3230
class RemoteBzrDirFormat(BzrDirMetaFormat1):
3231
"""Format representing bzrdirs accessed via a smart server"""
3234
BzrDirMetaFormat1.__init__(self)
3235
# XXX: It's a bit ugly that the network name is here, because we'd
3236
# like to believe that format objects are stateless or at least
3237
# immutable, However, we do at least avoid mutating the name after
3238
# it's returned. See <https://bugs.edge.launchpad.net/bzr/+bug/504102>
3239
self._network_name = None
3242
return "%s(_network_name=%r)" % (self.__class__.__name__,
3245
def get_format_description(self):
3246
if self._network_name:
3247
real_format = network_format_registry.get(self._network_name)
3248
return 'Remote: ' + real_format.get_format_description()
3249
return 'bzr remote bzrdir'
3251
def get_format_string(self):
3252
raise NotImplementedError(self.get_format_string)
3254
def network_name(self):
3255
if self._network_name:
3256
return self._network_name
3258
raise AssertionError("No network name set.")
3261
def probe_transport(klass, transport):
3262
"""Return a RemoteBzrDirFormat object if it looks possible."""
3264
medium = transport.get_smart_medium()
3265
except (NotImplementedError, AttributeError,
3266
errors.TransportNotPossible, errors.NoSmartMedium,
3267
errors.SmartProtocolError):
3268
# no smart server, so not a branch for this format type.
3269
raise errors.NotBranchError(path=transport.base)
3271
# Decline to open it if the server doesn't support our required
3272
# version (3) so that the VFS-based transport will do it.
3273
if medium.should_probe():
3275
server_version = medium.protocol_version()
3276
except errors.SmartProtocolError:
3277
# Apparently there's no usable smart server there, even though
3278
# the medium supports the smart protocol.
3279
raise errors.NotBranchError(path=transport.base)
3280
if server_version != '2':
3281
raise errors.NotBranchError(path=transport.base)
3284
def initialize_on_transport(self, transport):
3286
# hand off the request to the smart server
3287
client_medium = transport.get_smart_medium()
3288
except errors.NoSmartMedium:
3289
# TODO: lookup the local format from a server hint.
3290
local_dir_format = BzrDirMetaFormat1()
3291
return local_dir_format.initialize_on_transport(transport)
3292
client = _SmartClient(client_medium)
3293
path = client.remote_path_from_transport(transport)
3295
response = client.call('BzrDirFormat.initialize', path)
3296
except errors.ErrorFromSmartServer, err:
3297
remote._translate_error(err, path=path)
3298
if response[0] != 'ok':
3299
raise errors.SmartProtocolError('unexpected response code %s' % (response,))
3300
format = RemoteBzrDirFormat()
3301
self._supply_sub_formats_to(format)
3302
return remote.RemoteBzrDir(transport, format)
3304
def parse_NoneTrueFalse(self, arg):
3311
raise AssertionError("invalid arg %r" % arg)
3313
def _serialize_NoneTrueFalse(self, arg):
3320
def _serialize_NoneString(self, arg):
3323
def initialize_on_transport_ex(self, transport, use_existing_dir=False,
3324
create_prefix=False, force_new_repo=False, stacked_on=None,
3325
stack_on_pwd=None, repo_format_name=None, make_working_trees=None,
3328
# hand off the request to the smart server
3329
client_medium = transport.get_smart_medium()
3330
except errors.NoSmartMedium:
3333
# Decline to open it if the server doesn't support our required
3334
# version (3) so that the VFS-based transport will do it.
3335
if client_medium.should_probe():
3337
server_version = client_medium.protocol_version()
3338
if server_version != '2':
3342
except errors.SmartProtocolError:
3343
# Apparently there's no usable smart server there, even though
3344
# the medium supports the smart protocol.
3349
client = _SmartClient(client_medium)
3350
path = client.remote_path_from_transport(transport)
3351
if client_medium._is_remote_before((1, 16)):
3354
# TODO: lookup the local format from a server hint.
3355
local_dir_format = BzrDirMetaFormat1()
3356
self._supply_sub_formats_to(local_dir_format)
3357
return local_dir_format.initialize_on_transport_ex(transport,
3358
use_existing_dir=use_existing_dir, create_prefix=create_prefix,
3359
force_new_repo=force_new_repo, stacked_on=stacked_on,
3360
stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
3361
make_working_trees=make_working_trees, shared_repo=shared_repo,
3363
return self._initialize_on_transport_ex_rpc(client, path, transport,
3364
use_existing_dir, create_prefix, force_new_repo, stacked_on,
3365
stack_on_pwd, repo_format_name, make_working_trees, shared_repo)
3367
def _initialize_on_transport_ex_rpc(self, client, path, transport,
3368
use_existing_dir, create_prefix, force_new_repo, stacked_on,
3369
stack_on_pwd, repo_format_name, make_working_trees, shared_repo):
3371
args.append(self._serialize_NoneTrueFalse(use_existing_dir))
3372
args.append(self._serialize_NoneTrueFalse(create_prefix))
3373
args.append(self._serialize_NoneTrueFalse(force_new_repo))
3374
args.append(self._serialize_NoneString(stacked_on))
3375
# stack_on_pwd is often/usually our transport
3378
stack_on_pwd = transport.relpath(stack_on_pwd)
3379
if not stack_on_pwd:
3381
except errors.PathNotChild:
3383
args.append(self._serialize_NoneString(stack_on_pwd))
3384
args.append(self._serialize_NoneString(repo_format_name))
3385
args.append(self._serialize_NoneTrueFalse(make_working_trees))
3386
args.append(self._serialize_NoneTrueFalse(shared_repo))
3387
request_network_name = self._network_name or \
3388
BzrDirFormat.get_default_format().network_name()
3390
response = client.call('BzrDirFormat.initialize_ex_1.16',
3391
request_network_name, path, *args)
3392
except errors.UnknownSmartMethod:
3393
client._medium._remember_remote_is_before((1,16))
3394
local_dir_format = BzrDirMetaFormat1()
3395
self._supply_sub_formats_to(local_dir_format)
3396
return local_dir_format.initialize_on_transport_ex(transport,
3397
use_existing_dir=use_existing_dir, create_prefix=create_prefix,
3398
force_new_repo=force_new_repo, stacked_on=stacked_on,
3399
stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
3400
make_working_trees=make_working_trees, shared_repo=shared_repo,
3402
except errors.ErrorFromSmartServer, err:
3403
remote._translate_error(err, path=path)
3404
repo_path = response[0]
3405
bzrdir_name = response[6]
3406
require_stacking = response[7]
3407
require_stacking = self.parse_NoneTrueFalse(require_stacking)
3408
format = RemoteBzrDirFormat()
3409
format._network_name = bzrdir_name
3410
self._supply_sub_formats_to(format)
3411
bzrdir = remote.RemoteBzrDir(transport, format, _client=client)
3413
repo_format = remote.response_tuple_to_repo_format(response[1:])
3414
if repo_path == '.':
3417
repo_bzrdir_format = RemoteBzrDirFormat()
3418
repo_bzrdir_format._network_name = response[5]
3419
repo_bzr = remote.RemoteBzrDir(transport.clone(repo_path),
3423
final_stack = response[8] or None
3424
final_stack_pwd = response[9] or None
3426
final_stack_pwd = urlutils.join(
3427
transport.base, final_stack_pwd)
3428
remote_repo = remote.RemoteRepository(repo_bzr, repo_format)
3429
if len(response) > 10:
3430
# Updated server verb that locks remotely.
3431
repo_lock_token = response[10] or None
3432
remote_repo.lock_write(repo_lock_token, _skip_rpc=True)
3434
remote_repo.dont_leave_lock_in_place()
3436
remote_repo.lock_write()
3437
policy = UseExistingRepository(remote_repo, final_stack,
3438
final_stack_pwd, require_stacking)
3439
policy.acquire_repository()
3443
bzrdir._format.set_branch_format(self.get_branch_format())
3444
if require_stacking:
3445
# The repo has already been created, but we need to make sure that
3446
# we'll make a stackable branch.
3447
bzrdir._format.require_stacking(_skip_repo=True)
3448
return remote_repo, bzrdir, require_stacking, policy
3450
def _open(self, transport):
3451
return remote.RemoteBzrDir(transport, self)
3453
def __eq__(self, other):
3454
if not isinstance(other, RemoteBzrDirFormat):
3456
return self.get_format_description() == other.get_format_description()
3458
def __return_repository_format(self):
3459
# Always return a RemoteRepositoryFormat object, but if a specific bzr
3460
# repository format has been asked for, tell the RemoteRepositoryFormat
3461
# that it should use that for init() etc.
3462
result = remote.RemoteRepositoryFormat()
3463
custom_format = getattr(self, '_repository_format', None)
3465
if isinstance(custom_format, remote.RemoteRepositoryFormat):
3466
return custom_format
3468
# We will use the custom format to create repositories over the
3469
# wire; expose its details like rich_root_data for code to
3471
result._custom_format = custom_format
3474
def get_branch_format(self):
3475
result = BzrDirMetaFormat1.get_branch_format(self)
3476
if not isinstance(result, remote.RemoteBranchFormat):
3477
new_result = remote.RemoteBranchFormat()
3478
new_result._custom_format = result
3480
self.set_branch_format(new_result)
3484
repository_format = property(__return_repository_format,
3485
BzrDirMetaFormat1._set_repository_format) #.im_func)
3488
BzrDirFormat.register_control_server_format(RemoteBzrDirFormat)
3491
class BzrDirFormatInfo(object):
3493
def __init__(self, native, deprecated, hidden, experimental):
3494
self.deprecated = deprecated
3495
self.native = native
3496
self.hidden = hidden
3497
self.experimental = experimental
3500
class BzrDirFormatRegistry(registry.Registry):
3501
"""Registry of user-selectable BzrDir subformats.
3503
Differs from BzrDirFormat._control_formats in that it provides sub-formats,
3504
e.g. BzrDirMeta1 with weave repository. Also, it's more user-oriented.
3508
"""Create a BzrDirFormatRegistry."""
3509
self._aliases = set()
3510
self._registration_order = list()
3511
super(BzrDirFormatRegistry, self).__init__()
3514
"""Return a set of the format names which are aliases."""
3515
return frozenset(self._aliases)
3517
def register_metadir(self, key,
3518
repository_format, help, native=True, deprecated=False,
3524
"""Register a metadir subformat.
3526
These all use a BzrDirMetaFormat1 bzrdir, but can be parameterized
3527
by the Repository/Branch/WorkingTreeformats.
3529
:param repository_format: The fully-qualified repository format class
3531
:param branch_format: Fully-qualified branch format class name as
3533
:param tree_format: Fully-qualified tree format class name as
3536
# This should be expanded to support setting WorkingTree and Branch
3537
# formats, once BzrDirMetaFormat1 supports that.
3538
def _load(full_name):
3539
mod_name, factory_name = full_name.rsplit('.', 1)
3541
mod = __import__(mod_name, globals(), locals(),
3543
except ImportError, e:
3544
raise ImportError('failed to load %s: %s' % (full_name, e))
3546
factory = getattr(mod, factory_name)
3547
except AttributeError:
3548
raise AttributeError('no factory %s in module %r'
3553
bd = BzrDirMetaFormat1()
3554
if branch_format is not None:
3555
bd.set_branch_format(_load(branch_format))
3556
if tree_format is not None:
3557
bd.workingtree_format = _load(tree_format)
3558
if repository_format is not None:
3559
bd.repository_format = _load(repository_format)
3561
self.register(key, helper, help, native, deprecated, hidden,
3562
experimental, alias)
3564
def register(self, key, factory, help, native=True, deprecated=False,
3565
hidden=False, experimental=False, alias=False):
3566
"""Register a BzrDirFormat factory.
3568
The factory must be a callable that takes one parameter: the key.
3569
It must produce an instance of the BzrDirFormat when called.
3571
This function mainly exists to prevent the info object from being
3574
registry.Registry.register(self, key, factory, help,
3575
BzrDirFormatInfo(native, deprecated, hidden, experimental))
3577
self._aliases.add(key)
3578
self._registration_order.append(key)
3580
def register_lazy(self, key, module_name, member_name, help, native=True,
3581
deprecated=False, hidden=False, experimental=False, alias=False):
3582
registry.Registry.register_lazy(self, key, module_name, member_name,
3583
help, BzrDirFormatInfo(native, deprecated, hidden, experimental))
3585
self._aliases.add(key)
3586
self._registration_order.append(key)
3588
def set_default(self, key):
3589
"""Set the 'default' key to be a clone of the supplied key.
3591
This method must be called once and only once.
3593
registry.Registry.register(self, 'default', self.get(key),
3594
self.get_help(key), info=self.get_info(key))
3595
self._aliases.add('default')
3597
def set_default_repository(self, key):
3598
"""Set the FormatRegistry default and Repository default.
3600
This is a transitional method while Repository.set_default_format
3603
if 'default' in self:
3604
self.remove('default')
3605
self.set_default(key)
3606
format = self.get('default')()
3608
def make_bzrdir(self, key):
3609
return self.get(key)()
3611
def help_topic(self, topic):
3613
default_realkey = None
3614
default_help = self.get_help('default')
3616
for key in self._registration_order:
3617
if key == 'default':
3619
help = self.get_help(key)
3620
if help == default_help:
3621
default_realkey = key
3623
help_pairs.append((key, help))
3625
def wrapped(key, help, info):
3627
help = '(native) ' + help
3628
return ':%s:\n%s\n\n' % (key,
3629
textwrap.fill(help, initial_indent=' ',
3630
subsequent_indent=' ',
3631
break_long_words=False))
3632
if default_realkey is not None:
3633
output += wrapped(default_realkey, '(default) %s' % default_help,
3634
self.get_info('default'))
3635
deprecated_pairs = []
3636
experimental_pairs = []
3637
for key, help in help_pairs:
3638
info = self.get_info(key)
3641
elif info.deprecated:
3642
deprecated_pairs.append((key, help))
3643
elif info.experimental:
3644
experimental_pairs.append((key, help))
3646
output += wrapped(key, help, info)
3647
output += "\nSee :doc:`formats-help` for more about storage formats."
3649
if len(experimental_pairs) > 0:
3650
other_output += "Experimental formats are shown below.\n\n"
3651
for key, help in experimental_pairs:
3652
info = self.get_info(key)
3653
other_output += wrapped(key, help, info)
3656
"No experimental formats are available.\n\n"
3657
if len(deprecated_pairs) > 0:
3658
other_output += "\nDeprecated formats are shown below.\n\n"
3659
for key, help in deprecated_pairs:
3660
info = self.get_info(key)
3661
other_output += wrapped(key, help, info)
3664
"\nNo deprecated formats are available.\n\n"
3666
"\nSee :doc:`formats-help` for more about storage formats."
3668
if topic == 'other-formats':
1768
controldir.ControlDirFormat.register_server_prober(RemoteBzrProber)
3674
1771
class RepositoryAcquisitionPolicy(object):
added
removed
bzrlib/bzrdir.py
