~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/workingtree.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-29 22:03:03 UTC
  • mfrom: (5416.2.6 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100929220303-cr95h8iwtggco721
(mbp) Add 'break-lock --force'

Show diffs side-by-side

added added

removed removed

Lines of Context:
29
29
WorkingTree.open(dir).
30
30
"""
31
31
 
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)
38
32
 
39
33
from cStringIO import StringIO
40
34
import os
55
49
    branch,
56
50
    bzrdir,
57
51
    conflicts as _mod_conflicts,
 
52
    controldir,
58
53
    errors,
59
54
    generate_ids,
60
55
    globbing,
67
62
    revisiontree,
68
63
    trace,
69
64
    transform,
 
65
    transport,
70
66
    ui,
71
67
    views,
72
68
    xml5,
73
69
    xml7,
74
70
    )
75
 
import bzrlib.branch
76
 
from bzrlib.transport import get_transport
77
71
from bzrlib.workingtree_4 import (
78
72
    WorkingTreeFormat4,
79
73
    WorkingTreeFormat5,
83
77
 
84
78
from bzrlib import symbol_versioning
85
79
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
80
from bzrlib.lock import LogicalLockResult
86
81
from bzrlib.lockable_files import LockableFiles
87
82
from bzrlib.lockdir import LockDir
88
83
import bzrlib.mutabletree
101
96
from bzrlib.filters import filtered_input_file
102
97
from bzrlib.trace import mutter, note
103
98
from bzrlib.transport.local import LocalTransport
104
 
from bzrlib.progress import ProgressPhase
105
99
from bzrlib.revision import CURRENT_REVISION
106
100
from bzrlib.rio import RioReader, rio_file, Stanza
107
101
from bzrlib.symbol_versioning import (
174
168
        return ''
175
169
 
176
170
 
177
 
class WorkingTree(bzrlib.mutabletree.MutableTree):
 
171
class WorkingTree(bzrlib.mutabletree.MutableTree,
 
172
    controldir.ControlComponent):
178
173
    """Working copy tree.
179
174
 
180
175
    The inventory is held in the `Branch` working-inventory, and the
182
177
 
183
178
    It is possible for a `WorkingTree` to have a filename which is
184
179
    not listed in the Inventory and vice versa.
 
180
 
 
181
    :ivar basedir: The root of the tree on disk. This is a unicode path object
 
182
        (as opposed to a URL).
185
183
    """
186
184
 
187
185
    # override this to set the strategy for storing views
253
251
        self._rules_searcher = None
254
252
        self.views = self._make_views()
255
253
 
 
254
    @property
 
255
    def user_transport(self):
 
256
        return self.bzrdir.user_transport
 
257
 
 
258
    @property
 
259
    def control_transport(self):
 
260
        return self._transport
 
261
 
256
262
    def _detect_case_handling(self):
257
263
        wt_trans = self.bzrdir.get_workingtree_transport(None)
258
264
        try:
344
350
        if path is None:
345
351
            path = osutils.getcwd()
346
352
        control, relpath = bzrdir.BzrDir.open_containing(path)
347
 
 
348
353
        return control.open_workingtree(), relpath
349
354
 
350
355
    @staticmethod
 
356
    def open_containing_paths(file_list, default_directory='.',
 
357
        canonicalize=True, apply_view=True):
 
358
        """Open the WorkingTree that contains a set of paths.
 
359
 
 
360
        Fail if the paths given are not all in a single tree.
 
361
 
 
362
        This is used for the many command-line interfaces that take a list of
 
363
        any number of files and that require they all be in the same tree.
 
364
        """
 
365
        # recommended replacement for builtins.internal_tree_files
 
366
        if file_list is None or len(file_list) == 0:
 
367
            tree = WorkingTree.open_containing(default_directory)[0]
 
368
            # XXX: doesn't really belong here, and seems to have the strange
 
369
            # side effect of making it return a bunch of files, not the whole
 
370
            # tree -- mbp 20100716
 
371
            if tree.supports_views() and apply_view:
 
372
                view_files = tree.views.lookup_view()
 
373
                if view_files:
 
374
                    file_list = view_files
 
375
                    view_str = views.view_display_str(view_files)
 
376
                    note("Ignoring files outside view. View is %s" % view_str)
 
377
            return tree, file_list
 
378
        tree = WorkingTree.open_containing(file_list[0])[0]
 
379
        return tree, tree.safe_relpath_files(file_list, canonicalize,
 
380
            apply_view=apply_view)
 
381
 
 
382
    def safe_relpath_files(self, file_list, canonicalize=True, apply_view=True):
 
383
        """Convert file_list into a list of relpaths in tree.
 
384
 
 
385
        :param self: A tree to operate on.
 
386
        :param file_list: A list of user provided paths or None.
 
387
        :param apply_view: if True and a view is set, apply it or check that
 
388
            specified files are within it
 
389
        :return: A list of relative paths.
 
390
        :raises errors.PathNotChild: When a provided path is in a different self
 
391
            than self.
 
392
        """
 
393
        if file_list is None:
 
394
            return None
 
395
        if self.supports_views() and apply_view:
 
396
            view_files = self.views.lookup_view()
 
397
        else:
 
398
            view_files = []
 
399
        new_list = []
 
400
        # self.relpath exists as a "thunk" to osutils, but canonical_relpath
 
401
        # doesn't - fix that up here before we enter the loop.
 
402
        if canonicalize:
 
403
            fixer = lambda p: osutils.canonical_relpath(self.basedir, p)
 
404
        else:
 
405
            fixer = self.relpath
 
406
        for filename in file_list:
 
407
            relpath = fixer(osutils.dereference_path(filename))
 
408
            if view_files and not osutils.is_inside_any(view_files, relpath):
 
409
                raise errors.FileOutsideView(filename, view_files)
 
410
            new_list.append(relpath)
 
411
        return new_list
 
412
 
 
413
    @staticmethod
351
414
    def open_downlevel(path=None):
352
415
        """Open an unsupported working tree.
353
416
 
366
429
                return True, None
367
430
            else:
368
431
                return True, tree
369
 
        transport = get_transport(location)
370
 
        iterator = bzrdir.BzrDir.find_bzrdirs(transport, evaluate=evaluate,
 
432
        t = transport.get_transport(location)
 
433
        iterator = bzrdir.BzrDir.find_bzrdirs(t, evaluate=evaluate,
371
434
                                              list_current=list_current)
372
 
        return [t for t in iterator if t is not None]
 
435
        return [tr for tr in iterator if tr is not None]
373
436
 
374
437
    # should be deprecated - this is slow and in any case treating them as a
375
438
    # container is (we now know) bad style -- mbp 20070302
460
523
        return (file_obj, stat_value)
461
524
 
462
525
    def get_file_text(self, file_id, path=None, filtered=True):
463
 
        return self.get_file(file_id, path=path, filtered=filtered).read()
 
526
        my_file = self.get_file(file_id, path=path, filtered=filtered)
 
527
        try:
 
528
            return my_file.read()
 
529
        finally:
 
530
            my_file.close()
464
531
 
465
532
    def get_file_byname(self, filename, filtered=True):
466
533
        path = self.abspath(filename)
520
587
 
521
588
        # Now we have the parents of this content
522
589
        annotator = self.branch.repository.texts.get_annotator()
523
 
        text = self.get_file(file_id).read()
 
590
        text = self.get_file_text(file_id)
524
591
        this_key =(file_id, default_revision)
525
592
        annotator.add_special_text(this_key, file_parent_keys, text)
526
593
        annotations = [(key[-1], line)
1094
1161
        tree_transport = self.bzrdir.root_transport.clone(sub_path)
1095
1162
        if tree_transport.base != branch_transport.base:
1096
1163
            tree_bzrdir = format.initialize_on_transport(tree_transport)
1097
 
            branch.BranchReferenceFormat().initialize(tree_bzrdir, new_branch)
 
1164
            branch.BranchReferenceFormat().initialize(tree_bzrdir,
 
1165
                target_branch=new_branch)
1098
1166
        else:
1099
1167
            tree_bzrdir = branch_bzrdir
1100
1168
        wt = tree_bzrdir.create_workingtree(_mod_revision.NULL_REVISION)
1138
1206
        This does not include files that have been deleted in this
1139
1207
        tree. Skips the control directory.
1140
1208
 
1141
 
        :param include_root: if True, do not return an entry for the root
 
1209
        :param include_root: if True, return an entry for the root
1142
1210
        :param from_dir: start from this directory or None for the root
1143
1211
        :param recursive: whether to recurse into subdirectories or not
1144
1212
        """
1199
1267
                # absolute path
1200
1268
                fap = from_dir_abspath + '/' + f
1201
1269
 
1202
 
                f_ie = inv.get_child(from_dir_id, f)
 
1270
                dir_ie = inv[from_dir_id]
 
1271
                if dir_ie.kind == 'directory':
 
1272
                    f_ie = dir_ie.children.get(f)
 
1273
                else:
 
1274
                    f_ie = None
1203
1275
                if f_ie:
1204
1276
                    c = 'V'
1205
1277
                elif self.is_ignored(fp[1:]):
1206
1278
                    c = 'I'
1207
1279
                else:
1208
 
                    # we may not have found this file, because of a unicode issue
 
1280
                    # we may not have found this file, because of a unicode
 
1281
                    # issue, or because the directory was actually a symlink.
1209
1282
                    f_norm, can_access = osutils.normalized_filename(f)
1210
1283
                    if f == f_norm or not can_access:
1211
1284
                        # No change, so treat this file normally
1254
1327
                stack.pop()
1255
1328
 
1256
1329
    @needs_tree_write_lock
1257
 
    def move(self, from_paths, to_dir=None, after=False, **kwargs):
 
1330
    def move(self, from_paths, to_dir=None, after=False):
1258
1331
        """Rename files.
1259
1332
 
1260
1333
        to_dir must exist in the inventory.
1294
1367
 
1295
1368
        # check for deprecated use of signature
1296
1369
        if to_dir is None:
1297
 
            to_dir = kwargs.get('to_name', None)
1298
 
            if to_dir is None:
1299
 
                raise TypeError('You must supply a target directory')
1300
 
            else:
1301
 
                symbol_versioning.warn('The parameter to_name was deprecated'
1302
 
                                       ' in version 0.13. Use to_dir instead',
1303
 
                                       DeprecationWarning)
1304
 
 
 
1370
            raise TypeError('You must supply a target directory')
1305
1371
        # check destination directory
1306
1372
        if isinstance(from_paths, basestring):
1307
1373
            raise ValueError()
1795
1861
            raise errors.ObjectNotLocked(self)
1796
1862
 
1797
1863
    def lock_read(self):
1798
 
        """See Branch.lock_read, and WorkingTree.unlock."""
 
1864
        """Lock the tree for reading.
 
1865
 
 
1866
        This also locks the branch, and can be unlocked via self.unlock().
 
1867
 
 
1868
        :return: A bzrlib.lock.LogicalLockResult.
 
1869
        """
1799
1870
        if not self.is_locked():
1800
1871
            self._reset_data()
1801
1872
        self.branch.lock_read()
1802
1873
        try:
1803
 
            return self._control_files.lock_read()
 
1874
            self._control_files.lock_read()
 
1875
            return LogicalLockResult(self.unlock)
1804
1876
        except:
1805
1877
            self.branch.unlock()
1806
1878
            raise
1807
1879
 
1808
1880
    def lock_tree_write(self):
1809
 
        """See MutableTree.lock_tree_write, and WorkingTree.unlock."""
 
1881
        """See MutableTree.lock_tree_write, and WorkingTree.unlock.
 
1882
 
 
1883
        :return: A bzrlib.lock.LogicalLockResult.
 
1884
        """
1810
1885
        if not self.is_locked():
1811
1886
            self._reset_data()
1812
1887
        self.branch.lock_read()
1813
1888
        try:
1814
 
            return self._control_files.lock_write()
 
1889
            self._control_files.lock_write()
 
1890
            return LogicalLockResult(self.unlock)
1815
1891
        except:
1816
1892
            self.branch.unlock()
1817
1893
            raise
1818
1894
 
1819
1895
    def lock_write(self):
1820
 
        """See MutableTree.lock_write, and WorkingTree.unlock."""
 
1896
        """See MutableTree.lock_write, and WorkingTree.unlock.
 
1897
 
 
1898
        :return: A bzrlib.lock.LogicalLockResult.
 
1899
        """
1821
1900
        if not self.is_locked():
1822
1901
            self._reset_data()
1823
1902
        self.branch.lock_write()
1824
1903
        try:
1825
 
            return self._control_files.lock_write()
 
1904
            self._control_files.lock_write()
 
1905
            return LogicalLockResult(self.unlock)
1826
1906
        except:
1827
1907
            self.branch.unlock()
1828
1908
            raise
1945
2025
 
1946
2026
        inv_delta = []
1947
2027
 
1948
 
        new_files=set()
 
2028
        all_files = set() # specified and nested files 
1949
2029
        unknown_nested_files=set()
1950
2030
        if to_file is None:
1951
2031
            to_file = sys.stdout
1952
2032
 
 
2033
        files_to_backup = []
 
2034
 
1953
2035
        def recurse_directory_to_add_files(directory):
1954
2036
            # Recurse directory and add all files
1955
2037
            # so we can check if they have changed.
1956
 
            for parent_info, file_infos in\
1957
 
                self.walkdirs(directory):
 
2038
            for parent_info, file_infos in self.walkdirs(directory):
1958
2039
                for relpath, basename, kind, lstat, fileid, kind in file_infos:
1959
2040
                    # Is it versioned or ignored?
1960
 
                    if self.path2id(relpath) or self.is_ignored(relpath):
 
2041
                    if self.path2id(relpath):
1961
2042
                        # Add nested content for deletion.
1962
 
                        new_files.add(relpath)
 
2043
                        all_files.add(relpath)
1963
2044
                    else:
1964
 
                        # Files which are not versioned and not ignored
 
2045
                        # Files which are not versioned
1965
2046
                        # should be treated as unknown.
1966
 
                        unknown_nested_files.add((relpath, None, kind))
 
2047
                        files_to_backup.append(relpath)
1967
2048
 
1968
2049
        for filename in files:
1969
2050
            # Get file name into canonical form.
1970
2051
            abspath = self.abspath(filename)
1971
2052
            filename = self.relpath(abspath)
1972
2053
            if len(filename) > 0:
1973
 
                new_files.add(filename)
 
2054
                all_files.add(filename)
1974
2055
                recurse_directory_to_add_files(filename)
1975
2056
 
1976
 
        files = list(new_files)
 
2057
        files = list(all_files)
1977
2058
 
1978
2059
        if len(files) == 0:
1979
2060
            return # nothing to do
1983
2064
 
1984
2065
        # Bail out if we are going to delete files we shouldn't
1985
2066
        if not keep_files and not force:
1986
 
            has_changed_files = len(unknown_nested_files) > 0
1987
 
            if not has_changed_files:
1988
 
                for (file_id, path, content_change, versioned, parent_id, name,
1989
 
                     kind, executable) in self.iter_changes(self.basis_tree(),
1990
 
                         include_unchanged=True, require_versioned=False,
1991
 
                         want_unversioned=True, specific_files=files):
1992
 
                    if versioned == (False, False):
1993
 
                        # The record is unknown ...
1994
 
                        if not self.is_ignored(path[1]):
1995
 
                            # ... but not ignored
1996
 
                            has_changed_files = True
1997
 
                            break
1998
 
                    elif content_change and (kind[1] is not None):
1999
 
                        # Versioned and changed, but not deleted
2000
 
                        has_changed_files = True
2001
 
                        break
 
2067
            for (file_id, path, content_change, versioned, parent_id, name,
 
2068
                 kind, executable) in self.iter_changes(self.basis_tree(),
 
2069
                     include_unchanged=True, require_versioned=False,
 
2070
                     want_unversioned=True, specific_files=files):
 
2071
                if versioned[0] == False:
 
2072
                    # The record is unknown or newly added
 
2073
                    files_to_backup.append(path[1])
 
2074
                elif (content_change and (kind[1] is not None) and
 
2075
                        osutils.is_inside_any(files, path[1])):
 
2076
                    # Versioned and changed, but not deleted, and still
 
2077
                    # in one of the dirs to be deleted.
 
2078
                    files_to_backup.append(path[1])
2002
2079
 
2003
 
            if has_changed_files:
2004
 
                # Make delta show ALL applicable changes in error message.
2005
 
                tree_delta = self.changes_from(self.basis_tree(),
2006
 
                    require_versioned=False, want_unversioned=True,
2007
 
                    specific_files=files)
2008
 
                for unknown_file in unknown_nested_files:
2009
 
                    if unknown_file not in tree_delta.unversioned:
2010
 
                        tree_delta.unversioned.extend((unknown_file,))
2011
 
                raise errors.BzrRemoveChangedFilesError(tree_delta)
 
2080
        def backup(file_to_backup):
 
2081
            backup_name = self.bzrdir.generate_backup_name(file_to_backup)
 
2082
            osutils.rename(abs_path, self.abspath(backup_name))
 
2083
            return "removed %s (but kept a copy: %s)" % (file_to_backup, backup_name)
2012
2084
 
2013
2085
        # Build inv_delta and delete files where applicable,
2014
2086
        # do this before any modifications to inventory.
2038
2110
                        len(os.listdir(abs_path)) > 0):
2039
2111
                        if force:
2040
2112
                            osutils.rmtree(abs_path)
 
2113
                            message = "deleted %s" % (f,)
2041
2114
                        else:
2042
 
                            message = "%s is not an empty directory "\
2043
 
                                "and won't be deleted." % (f,)
 
2115
                            message = backup(f)
2044
2116
                    else:
2045
 
                        osutils.delete_any(abs_path)
2046
 
                        message = "deleted %s" % (f,)
 
2117
                        if f in files_to_backup:
 
2118
                            message = backup(f)
 
2119
                        else:
 
2120
                            osutils.delete_any(abs_path)
 
2121
                            message = "deleted %s" % (f,)
2047
2122
                elif message is not None:
2048
2123
                    # Only care if we haven't done anything yet.
2049
2124
                    message = "%s does not exist." % (f,)
2257
2332
            last_rev = _mod_revision.NULL_REVISION
2258
2333
        if revision is None:
2259
2334
            revision = self.branch.last_revision()
2260
 
        else:
2261
 
            if revision not in self.branch.revision_history():
2262
 
                raise errors.NoSuchRevision(self.branch, revision)
2263
2335
 
2264
2336
        old_tip = old_tip or _mod_revision.NULL_REVISION
2265
2337
 
2636
2708
 
2637
2709
        In Format2 WorkingTrees we have a single lock for the branch and tree
2638
2710
        so lock_tree_write() degrades to lock_write().
 
2711
 
 
2712
        :return: An object with an unlock method which will release the lock
 
2713
            obtained.
2639
2714
        """
2640
2715
        self.branch.lock_write()
2641
2716
        try:
2642
 
            return self._control_files.lock_write()
 
2717
            self._control_files.lock_write()
 
2718
            return self
2643
2719
        except:
2644
2720
            self.branch.unlock()
2645
2721
            raise