29
29
WorkingTree.open(dir).
32
# TODO: Give the workingtree sole responsibility for the working inventory;
33
# remove the variable and references to it from the branch. This may require
34
# updating the commit code so as to update the inventory within the working
35
# copy, and making sure there's only one WorkingTree for any directory on disk.
36
# At the moment they may alias the inventory and have old copies of it in
37
# memory. (Now done? -- mbp 20060309)
33
39
from cStringIO import StringIO
96
102
from bzrlib.filters import filtered_input_file
97
103
from bzrlib.trace import mutter, note
98
104
from bzrlib.transport.local import LocalTransport
105
from bzrlib.progress import DummyProgress, ProgressPhase
99
106
from bzrlib.revision import CURRENT_REVISION
100
107
from bzrlib.rio import RioReader, rio_file, Stanza
101
108
from bzrlib.symbol_versioning import (
286
281
self._control_files.break_lock()
287
282
self.branch.break_lock()
289
def _get_check_refs(self):
290
"""Return the references needed to perform a check of this tree.
292
The default implementation returns no refs, and is only suitable for
293
trees that have no local caching and can commit on ghosts at any time.
295
:seealso: bzrlib.check for details about check_refs.
299
284
def requires_rich_root(self):
300
285
return self._format.requires_rich_root
642
625
def _is_executable_from_path_and_stat_from_basis(self, path, stat_result):
643
626
file_id = self.path2id(path)
645
# For unversioned files on win32, we just assume they are not
648
627
return self._inventory[file_id].executable
650
629
def _is_executable_from_path_and_stat_from_stat(self, path, stat_result):
761
740
kind = _mapper(stat_result.st_mode)
762
741
if kind == 'file':
763
return self._file_content_summary(path, stat_result)
742
size = stat_result.st_size
743
# try for a stat cache lookup
744
executable = self._is_executable_from_path_and_stat(path, stat_result)
745
return (kind, size, executable, self._sha_from_stat(
764
747
elif kind == 'directory':
765
748
# perhaps it looks like a plain directory, but it's really a
774
757
return (kind, None, None, None)
776
def _file_content_summary(self, path, stat_result):
777
size = stat_result.st_size
778
executable = self._is_executable_from_path_and_stat(path, stat_result)
779
# try for a stat cache lookup
780
return ('file', size, executable, self._sha_from_stat(
783
759
def _check_parents_for_ghosts(self, revision_ids, allow_leftmost_as_ghost):
784
760
"""Common ghost checking functionality from set_parent_*.
906
882
@needs_write_lock # because merge pulls data into the branch.
907
883
def merge_from_branch(self, branch, to_revision=None, from_revision=None,
908
merge_type=None, force=False):
909
885
"""Merge from a branch into this working tree.
911
887
:param branch: The branch to merge from.
915
891
branch.last_revision().
917
893
from bzrlib.merge import Merger, Merge3Merger
918
merger = Merger(self.branch, this_tree=self)
919
# check that there are no local alterations
920
if not force and self.has_changes():
921
raise errors.UncommittedChanges(self)
922
if to_revision is None:
923
to_revision = _mod_revision.ensure_null(branch.last_revision())
924
merger.other_rev_id = to_revision
925
if _mod_revision.is_null(merger.other_rev_id):
926
raise errors.NoCommits(branch)
927
self.branch.fetch(branch, last_revision=merger.other_rev_id)
928
merger.other_basis = merger.other_rev_id
929
merger.other_tree = self.branch.repository.revision_tree(
931
merger.other_branch = branch
932
if from_revision is None:
935
merger.set_base_revision(from_revision, branch)
936
if merger.base_rev_id == merger.other_rev_id:
937
raise errors.PointlessMerge
938
merger.backup_files = False
939
if merge_type is None:
940
merger.merge_type = Merge3Merger
942
merger.merge_type = merge_type
943
merger.set_interesting_files(None)
944
merger.show_base = False
945
merger.reprocess = False
946
conflicts = merger.do_merge()
894
pb = ui.ui_factory.nested_progress_bar()
896
merger = Merger(self.branch, this_tree=self, pb=pb)
897
merger.pp = ProgressPhase("Merge phase", 5, pb)
898
merger.pp.next_phase()
899
# check that there are no
901
merger.check_basis(check_clean=True, require_commits=False)
902
if to_revision is None:
903
to_revision = _mod_revision.ensure_null(branch.last_revision())
904
merger.other_rev_id = to_revision
905
if _mod_revision.is_null(merger.other_rev_id):
906
raise errors.NoCommits(branch)
907
self.branch.fetch(branch, last_revision=merger.other_rev_id)
908
merger.other_basis = merger.other_rev_id
909
merger.other_tree = self.branch.repository.revision_tree(
911
merger.other_branch = branch
912
merger.pp.next_phase()
913
if from_revision is None:
916
merger.set_base_revision(from_revision, branch)
917
if merger.base_rev_id == merger.other_rev_id:
918
raise errors.PointlessMerge
919
merger.backup_files = False
920
if merge_type is None:
921
merger.merge_type = Merge3Merger
923
merger.merge_type = merge_type
924
merger.set_interesting_files(None)
925
merger.show_base = False
926
merger.reprocess = False
927
conflicts = merger.do_merge()
1097
1080
tree_transport = self.bzrdir.root_transport.clone(sub_path)
1098
1081
if tree_transport.base != branch_transport.base:
1099
1082
tree_bzrdir = format.initialize_on_transport(tree_transport)
1100
branch.BranchReferenceFormat().initialize(tree_bzrdir,
1101
target_branch=new_branch)
1083
branch.BranchReferenceFormat().initialize(tree_bzrdir, new_branch)
1103
1085
tree_bzrdir = branch_bzrdir
1104
1086
wt = tree_bzrdir.create_workingtree(_mod_revision.NULL_REVISION)
1142
1124
This does not include files that have been deleted in this
1143
1125
tree. Skips the control directory.
1145
:param include_root: if True, return an entry for the root
1127
:param include_root: if True, do not return an entry for the root
1146
1128
:param from_dir: start from this directory or None for the root
1147
1129
:param recursive: whether to recurse into subdirectories or not
1602
1584
@needs_write_lock
1603
1585
def pull(self, source, overwrite=False, stop_revision=None,
1604
1586
change_reporter=None, possible_transports=None, local=False):
1587
top_pb = ui.ui_factory.nested_progress_bar()
1605
1588
source.lock_read()
1590
pp = ProgressPhase("Pull phase", 2, top_pb)
1607
1592
old_revision_info = self.branch.last_revision_info()
1608
1593
basis_tree = self.basis_tree()
1609
1594
count = self.branch.pull(source, overwrite, stop_revision,
1620
1607
new_basis_tree,
1622
1609
this_tree=self,
1624
1611
change_reporter=change_reporter)
1625
basis_root_id = basis_tree.get_root_id()
1626
new_root_id = new_basis_tree.get_root_id()
1627
if basis_root_id != new_root_id:
1628
self.set_root_id(new_root_id)
1612
if (basis_tree.inventory.root is None and
1613
new_basis_tree.inventory.root is not None):
1614
self.set_root_id(new_basis_tree.get_root_id())
1630
1617
basis_tree.unlock()
1631
1618
# TODO - dedup parents list with things merged by pull ?
1632
1619
# reuse the revisiontree we merged against to set the new
1735
1723
r"""Check whether the filename matches an ignore pattern.
1737
1725
Patterns containing '/' or '\' need to match the whole path;
1738
others match against only the last component. Patterns starting
1739
with '!' are ignore exceptions. Exceptions take precedence
1740
over regular patterns and cause the filename to not be ignored.
1726
others match against only the last component.
1742
1728
If the file is ignored, returns the pattern which caused it to
1743
1729
be ignored, otherwise None. So this can simply be used as a
1744
1730
boolean if desired."""
1745
1731
if getattr(self, '_ignoreglobster', None) is None:
1746
self._ignoreglobster = globbing.ExceptionGlobster(self.get_ignore_list())
1732
self._ignoreglobster = globbing.Globster(self.get_ignore_list())
1747
1733
return self._ignoreglobster.match(filename)
1749
1735
def kind(self, file_id):
1799
1785
raise errors.ObjectNotLocked(self)
1801
1787
def lock_read(self):
1802
"""Lock the tree for reading.
1804
This also locks the branch, and can be unlocked via self.unlock().
1806
:return: A bzrlib.lock.LogicalLockResult.
1788
"""See Branch.lock_read, and WorkingTree.unlock."""
1808
1789
if not self.is_locked():
1809
1790
self._reset_data()
1810
1791
self.branch.lock_read()
1812
self._control_files.lock_read()
1813
return LogicalLockResult(self.unlock)
1793
return self._control_files.lock_read()
1815
1795
self.branch.unlock()
1818
1798
def lock_tree_write(self):
1819
"""See MutableTree.lock_tree_write, and WorkingTree.unlock.
1821
:return: A bzrlib.lock.LogicalLockResult.
1799
"""See MutableTree.lock_tree_write, and WorkingTree.unlock."""
1823
1800
if not self.is_locked():
1824
1801
self._reset_data()
1825
1802
self.branch.lock_read()
1827
self._control_files.lock_write()
1828
return LogicalLockResult(self.unlock)
1804
return self._control_files.lock_write()
1830
1806
self.branch.unlock()
1833
1809
def lock_write(self):
1834
"""See MutableTree.lock_write, and WorkingTree.unlock.
1836
:return: A bzrlib.lock.LogicalLockResult.
1810
"""See MutableTree.lock_write, and WorkingTree.unlock."""
1838
1811
if not self.is_locked():
1839
1812
self._reset_data()
1840
1813
self.branch.lock_write()
1842
self._control_files.lock_write()
1843
return LogicalLockResult(self.unlock)
1815
return self._control_files.lock_write()
1845
1817
self.branch.unlock()
1911
1879
# revision_id is set. We must check for this full string, because a
1912
1880
# root node id can legitimately look like 'revision_id' but cannot
1913
1881
# contain a '"'.
1914
xml = self.branch.repository._get_inventory_xml(new_revision)
1882
xml = self.branch.repository.get_inventory_xml(new_revision)
1915
1883
firstline = xml.split('\n', 1)[0]
1916
1884
if (not 'revision_id="' in firstline or
1917
1885
'format="7"' not in firstline):
1918
inv = self.branch.repository._serializer.read_inventory_from_string(
1886
inv = self.branch.repository.deserialise_inventory(
1920
1888
xml = self._create_basis_xml_from_inventory(new_revision, inv)
1921
1889
self._write_basis_inventory(xml)
1922
1890
except (errors.NoSuchRevision, errors.RevisionNotPresent):
1966
1930
new_files=set()
1967
1931
unknown_nested_files=set()
1969
to_file = sys.stdout
1971
1933
def recurse_directory_to_add_files(directory):
1972
1934
# Recurse directory and add all files
1973
1935
# so we can check if they have changed.
1974
for parent_info, file_infos in self.walkdirs(directory):
1936
for parent_info, file_infos in\
1937
self.walkdirs(directory):
1975
1938
for relpath, basename, kind, lstat, fileid, kind in file_infos:
1976
1939
# Is it versioned or ignored?
1977
1940
if self.path2id(relpath) or self.is_ignored(relpath):
2012
1975
# ... but not ignored
2013
1976
has_changed_files = True
2015
elif (content_change and (kind[1] is not None) and
2016
osutils.is_inside_any(files, path[1])):
2017
# Versioned and changed, but not deleted, and still
2018
# in one of the dirs to be deleted.
1978
elif content_change and (kind[1] is not None):
1979
# Versioned and changed, but not deleted
2019
1980
has_changed_files = True
2043
2004
new_status = 'I'
2045
2006
new_status = '?'
2046
# XXX: Really should be a more abstract reporter interface
2047
kind_ch = osutils.kind_marker(self.kind(fid))
2048
to_file.write(new_status + ' ' + f + kind_ch + '\n')
2007
textui.show_status(new_status, self.kind(fid), f,
2049
2009
# Unversion file
2050
2010
inv_delta.append((f, None, fid, None))
2051
2011
message = "removed %s" % (f,)
2203
2163
raise NotImplementedError(self.unlock)
2207
def update(self, change_reporter=None, possible_transports=None,
2208
revision=None, old_tip=_marker):
2165
def update(self, change_reporter=None, possible_transports=None):
2209
2166
"""Update a working tree along its branch.
2211
2168
This will update the branch if its bound too, which means we have
2229
2186
- Merge current state -> basis tree of the master w.r.t. the old tree
2231
2188
- Do a 'normal' merge of the old branch basis if it is relevant.
2233
:param revision: The target revision to update to. Must be in the
2235
:param old_tip: If branch.update() has already been run, the value it
2236
returned (old tip of the branch or None). _marker is used
2239
2190
if self.branch.get_bound_location() is not None:
2240
2191
self.lock_write()
2241
update_branch = (old_tip is self._marker)
2192
update_branch = True
2243
2194
self.lock_tree_write()
2244
2195
update_branch = False
2246
2197
if update_branch:
2247
2198
old_tip = self.branch.update(possible_transports)
2249
if old_tip is self._marker:
2251
return self._update_tree(old_tip, change_reporter, revision)
2201
return self._update_tree(old_tip, change_reporter)
2255
2205
@needs_tree_write_lock
2256
def _update_tree(self, old_tip=None, change_reporter=None, revision=None):
2206
def _update_tree(self, old_tip=None, change_reporter=None):
2257
2207
"""Update a tree to the master branch.
2259
2209
:param old_tip: if supplied, the previous tip revision the branch,
2269
2219
# We MUST save it even if an error occurs, because otherwise the users
2270
2220
# local work is unreferenced and will appear to have been lost.
2274
2224
last_rev = self.get_parent_ids()[0]
2275
2225
except IndexError:
2276
2226
last_rev = _mod_revision.NULL_REVISION
2277
if revision is None:
2278
revision = self.branch.last_revision()
2280
old_tip = old_tip or _mod_revision.NULL_REVISION
2282
if not _mod_revision.is_null(old_tip) and old_tip != last_rev:
2283
# the branch we are bound to was updated
2284
# merge those changes in first
2285
base_tree = self.basis_tree()
2286
other_tree = self.branch.repository.revision_tree(old_tip)
2287
nb_conflicts = merge.merge_inner(self.branch, other_tree,
2288
base_tree, this_tree=self,
2289
change_reporter=change_reporter)
2291
self.add_parent_tree((old_tip, other_tree))
2292
trace.note('Rerun update after fixing the conflicts.')
2295
if last_rev != _mod_revision.ensure_null(revision):
2296
# the working tree is up to date with the branch
2297
# we can merge the specified revision from master
2298
to_tree = self.branch.repository.revision_tree(revision)
2299
to_root_id = to_tree.get_root_id()
2227
if last_rev != _mod_revision.ensure_null(self.branch.last_revision()):
2228
# merge tree state up to new branch tip.
2301
2229
basis = self.basis_tree()
2302
2230
basis.lock_read()
2304
if (basis.inventory.root is None
2305
or basis.inventory.root.file_id != to_root_id):
2306
self.set_root_id(to_root_id)
2232
to_tree = self.branch.basis_tree()
2233
if basis.inventory.root is None:
2234
self.set_root_id(to_tree.get_root_id())
2236
result += merge.merge_inner(
2241
change_reporter=change_reporter)
2311
# determine the branch point
2312
graph = self.branch.repository.get_graph()
2313
base_rev_id = graph.find_unique_lca(self.branch.last_revision(),
2315
base_tree = self.branch.repository.revision_tree(base_rev_id)
2317
nb_conflicts = merge.merge_inner(self.branch, to_tree, base_tree,
2319
change_reporter=change_reporter)
2320
self.set_last_revision(revision)
2321
2244
# TODO - dedup parents list with things merged by pull ?
2322
2245
# reuse the tree we've updated to to set the basis:
2323
parent_trees = [(revision, to_tree)]
2246
parent_trees = [(self.branch.last_revision(), to_tree)]
2324
2247
merges = self.get_parent_ids()[1:]
2325
2248
# Ideally we ask the tree for the trees here, that way the working
2326
2249
# tree can decide whether to give us the entire tree or give us a
2330
2253
for parent in merges:
2331
2254
parent_trees.append(
2332
2255
(parent, self.branch.repository.revision_tree(parent)))
2333
if not _mod_revision.is_null(old_tip):
2256
if (old_tip is not None and not _mod_revision.is_null(old_tip)):
2334
2257
parent_trees.append(
2335
2258
(old_tip, self.branch.repository.revision_tree(old_tip)))
2336
2259
self.set_parent_trees(parent_trees)
2337
2260
last_rev = parent_trees[0][0]
2262
# the working tree had the same last-revision as the master
2263
# branch did. We may still have pivot local work from the local
2264
# branch into old_tip:
2265
if (old_tip is not None and not _mod_revision.is_null(old_tip)):
2266
self.add_parent_tree_id(old_tip)
2267
if (old_tip is not None and not _mod_revision.is_null(old_tip)
2268
and old_tip != last_rev):
2269
# our last revision was not the prior branch last revision
2270
# and we have converted that last revision to a pending merge.
2271
# base is somewhere between the branch tip now
2272
# and the now pending merge
2274
# Since we just modified the working tree and inventory, flush out
2275
# the current state, before we modify it again.
2276
# TODO: jam 20070214 WorkingTree3 doesn't require this, dirstate
2277
# requires it only because TreeTransform directly munges the
2278
# inventory and calls tree._write_inventory(). Ultimately we
2279
# should be able to remove this extra flush.
2281
graph = self.branch.repository.get_graph()
2282
base_rev_id = graph.find_unique_lca(self.branch.last_revision(),
2284
base_tree = self.branch.repository.revision_tree(base_rev_id)
2285
other_tree = self.branch.repository.revision_tree(old_tip)
2286
result += merge.merge_inner(
2291
change_reporter=change_reporter)
2340
2294
def _write_hashcache_if_dirty(self):
2341
2295
"""Write out the hashcache if it is dirty."""
2582
2536
return un_resolved, resolved
2584
2538
@needs_read_lock
2585
def _check(self, references):
2586
"""Check the tree for consistency.
2588
:param references: A dict with keys matching the items returned by
2589
self._get_check_refs(), and values from looking those keys up in
2592
2540
tree_basis = self.basis_tree()
2593
2541
tree_basis.lock_read()
2595
repo_basis = references[('trees', self.last_revision())]
2543
repo_basis = self.branch.repository.revision_tree(
2544
self.last_revision())
2596
2545
if len(list(repo_basis.iter_changes(tree_basis))) > 0:
2597
2546
raise errors.BzrCheckError(
2598
2547
"Mismatched basis inventory content.")
2643
2593
if self._inventory is None:
2644
2594
self.read_working_inventory()
2646
def _get_check_refs(self):
2647
"""Return the references needed to perform a check of this tree."""
2648
return [('trees', self.last_revision())]
2650
2596
def lock_tree_write(self):
2651
2597
"""See WorkingTree.lock_tree_write().
2653
2599
In Format2 WorkingTrees we have a single lock for the branch and tree
2654
2600
so lock_tree_write() degrades to lock_write().
2656
:return: An object with an unlock method which will release the lock
2659
2602
self.branch.lock_write()
2661
self._control_files.lock_write()
2604
return self._control_files.lock_write()
2664
2606
self.branch.unlock()
2713
2655
mode=self.bzrdir._get_file_mode())
2716
def _get_check_refs(self):
2717
"""Return the references needed to perform a check of this tree."""
2718
return [('trees', self.last_revision())]
2720
2658
@needs_tree_write_lock
2721
2659
def set_conflicts(self, conflicts):
2722
2660
self._put_rio('conflicts', conflicts.to_stanzas(),
2799
2737
"""Return the format for the working tree object in a_bzrdir."""
2801
2739
transport = a_bzrdir.get_workingtree_transport(None)
2802
format_string = transport.get_bytes("format")
2740
format_string = transport.get("format").read()
2803
2741
return klass._formats[format_string]
2804
2742
except errors.NoSuchFile:
2805
2743
raise errors.NoWorkingTree(base=transport.base)
3069
3007
return self.get_format_string()
3072
__default_format = WorkingTreeFormat6()
3010
__default_format = WorkingTreeFormat4()
3073
3011
WorkingTreeFormat.register_format(__default_format)
3012
WorkingTreeFormat.register_format(WorkingTreeFormat6())
3074
3013
WorkingTreeFormat.register_format(WorkingTreeFormat5())
3075
WorkingTreeFormat.register_format(WorkingTreeFormat4())
3076
3014
WorkingTreeFormat.register_format(WorkingTreeFormat3())
3077
3015
WorkingTreeFormat.set_default_format(__default_format)
3078
3016
# formats which have no format string are not discoverable