~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/workingtree.py

  • Committer: Zearin
  • Date: 2010-11-12 22:08:18 UTC
  • mto: (5570.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 5572.
  • Revision ID: zearin@users.sourceforge.net-20101112220818-mb62len4zyxr8qvd
Fixed capitalization of XML and HTTP.  Fixed by hand and only where appropriate (e.g., left http://some/url lowercase, but capitalized "When making an HTTP request…").

Show diffs side-by-side

added added

removed removed

Lines of Context:
49
49
    branch,
50
50
    bzrdir,
51
51
    conflicts as _mod_conflicts,
 
52
    controldir,
52
53
    errors,
53
54
    generate_ids,
54
55
    globbing,
61
62
    revisiontree,
62
63
    trace,
63
64
    transform,
 
65
    transport,
64
66
    ui,
65
67
    views,
66
68
    xml5,
67
69
    xml7,
68
70
    )
69
 
import bzrlib.branch
70
 
from bzrlib.transport import get_transport
71
71
from bzrlib.workingtree_4 import (
72
72
    WorkingTreeFormat4,
73
73
    WorkingTreeFormat5,
77
77
 
78
78
from bzrlib import symbol_versioning
79
79
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
80
from bzrlib.lock import LogicalLockResult
80
81
from bzrlib.lockable_files import LockableFiles
81
82
from bzrlib.lockdir import LockDir
82
83
import bzrlib.mutabletree
168
169
 
169
170
 
170
171
class WorkingTree(bzrlib.mutabletree.MutableTree,
171
 
    bzrdir.ControlComponent):
 
172
    controldir.ControlComponent):
172
173
    """Working copy tree.
173
174
 
174
175
    The inventory is held in the `Branch` working-inventory, and the
176
177
 
177
178
    It is possible for a `WorkingTree` to have a filename which is
178
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).
179
183
    """
180
184
 
181
185
    # override this to set the strategy for storing views
346
350
        if path is None:
347
351
            path = osutils.getcwd()
348
352
        control, relpath = bzrdir.BzrDir.open_containing(path)
349
 
 
350
353
        return control.open_workingtree(), relpath
351
354
 
352
355
    @staticmethod
 
356
    def open_containing_paths(file_list, default_directory=None,
 
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
        if default_directory is None:
 
366
            default_directory = u'.'
 
367
        # recommended replacement for builtins.internal_tree_files
 
368
        if file_list is None or len(file_list) == 0:
 
369
            tree = WorkingTree.open_containing(default_directory)[0]
 
370
            # XXX: doesn't really belong here, and seems to have the strange
 
371
            # side effect of making it return a bunch of files, not the whole
 
372
            # tree -- mbp 20100716
 
373
            if tree.supports_views() and apply_view:
 
374
                view_files = tree.views.lookup_view()
 
375
                if view_files:
 
376
                    file_list = view_files
 
377
                    view_str = views.view_display_str(view_files)
 
378
                    note("Ignoring files outside view. View is %s" % view_str)
 
379
            return tree, file_list
 
380
        if default_directory == u'.':
 
381
            seed = file_list[0]
 
382
        else:
 
383
            seed = default_directory
 
384
            file_list = [osutils.pathjoin(default_directory, f)
 
385
                         for f in file_list]
 
386
        tree = WorkingTree.open_containing(seed)[0]
 
387
        return tree, tree.safe_relpath_files(file_list, canonicalize,
 
388
                                             apply_view=apply_view)
 
389
 
 
390
    def safe_relpath_files(self, file_list, canonicalize=True, apply_view=True):
 
391
        """Convert file_list into a list of relpaths in tree.
 
392
 
 
393
        :param self: A tree to operate on.
 
394
        :param file_list: A list of user provided paths or None.
 
395
        :param apply_view: if True and a view is set, apply it or check that
 
396
            specified files are within it
 
397
        :return: A list of relative paths.
 
398
        :raises errors.PathNotChild: When a provided path is in a different self
 
399
            than self.
 
400
        """
 
401
        if file_list is None:
 
402
            return None
 
403
        if self.supports_views() and apply_view:
 
404
            view_files = self.views.lookup_view()
 
405
        else:
 
406
            view_files = []
 
407
        new_list = []
 
408
        # self.relpath exists as a "thunk" to osutils, but canonical_relpath
 
409
        # doesn't - fix that up here before we enter the loop.
 
410
        if canonicalize:
 
411
            fixer = lambda p: osutils.canonical_relpath(self.basedir, p)
 
412
        else:
 
413
            fixer = self.relpath
 
414
        for filename in file_list:
 
415
            relpath = fixer(osutils.dereference_path(filename))
 
416
            if view_files and not osutils.is_inside_any(view_files, relpath):
 
417
                raise errors.FileOutsideView(filename, view_files)
 
418
            new_list.append(relpath)
 
419
        return new_list
 
420
 
 
421
    @staticmethod
353
422
    def open_downlevel(path=None):
354
423
        """Open an unsupported working tree.
355
424
 
368
437
                return True, None
369
438
            else:
370
439
                return True, tree
371
 
        transport = get_transport(location)
372
 
        iterator = bzrdir.BzrDir.find_bzrdirs(transport, evaluate=evaluate,
 
440
        t = transport.get_transport(location)
 
441
        iterator = bzrdir.BzrDir.find_bzrdirs(t, evaluate=evaluate,
373
442
                                              list_current=list_current)
374
 
        return [t for t in iterator if t is not None]
 
443
        return [tr for tr in iterator if tr is not None]
375
444
 
376
445
    # should be deprecated - this is slow and in any case treating them as a
377
446
    # container is (we now know) bad style -- mbp 20070302
462
531
        return (file_obj, stat_value)
463
532
 
464
533
    def get_file_text(self, file_id, path=None, filtered=True):
465
 
        return self.get_file(file_id, path=path, filtered=filtered).read()
 
534
        my_file = self.get_file(file_id, path=path, filtered=filtered)
 
535
        try:
 
536
            return my_file.read()
 
537
        finally:
 
538
            my_file.close()
466
539
 
467
540
    def get_file_byname(self, filename, filtered=True):
468
541
        path = self.abspath(filename)
522
595
 
523
596
        # Now we have the parents of this content
524
597
        annotator = self.branch.repository.texts.get_annotator()
525
 
        text = self.get_file(file_id).read()
 
598
        text = self.get_file_text(file_id)
526
599
        this_key =(file_id, default_revision)
527
600
        annotator.add_special_text(this_key, file_parent_keys, text)
528
601
        annotations = [(key[-1], line)
1202
1275
                # absolute path
1203
1276
                fap = from_dir_abspath + '/' + f
1204
1277
 
1205
 
                f_ie = inv.get_child(from_dir_id, f)
 
1278
                dir_ie = inv[from_dir_id]
 
1279
                if dir_ie.kind == 'directory':
 
1280
                    f_ie = dir_ie.children.get(f)
 
1281
                else:
 
1282
                    f_ie = None
1206
1283
                if f_ie:
1207
1284
                    c = 'V'
1208
1285
                elif self.is_ignored(fp[1:]):
1209
1286
                    c = 'I'
1210
1287
                else:
1211
 
                    # we may not have found this file, because of a unicode issue
 
1288
                    # we may not have found this file, because of a unicode
 
1289
                    # issue, or because the directory was actually a symlink.
1212
1290
                    f_norm, can_access = osutils.normalized_filename(f)
1213
1291
                    if f == f_norm or not can_access:
1214
1292
                        # No change, so treat this file normally
1257
1335
                stack.pop()
1258
1336
 
1259
1337
    @needs_tree_write_lock
1260
 
    def move(self, from_paths, to_dir=None, after=False, **kwargs):
 
1338
    def move(self, from_paths, to_dir=None, after=False):
1261
1339
        """Rename files.
1262
1340
 
1263
1341
        to_dir must exist in the inventory.
1297
1375
 
1298
1376
        # check for deprecated use of signature
1299
1377
        if to_dir is None:
1300
 
            to_dir = kwargs.get('to_name', None)
1301
 
            if to_dir is None:
1302
 
                raise TypeError('You must supply a target directory')
1303
 
            else:
1304
 
                symbol_versioning.warn('The parameter to_name was deprecated'
1305
 
                                       ' in version 0.13. Use to_dir instead',
1306
 
                                       DeprecationWarning)
1307
 
 
 
1378
            raise TypeError('You must supply a target directory')
1308
1379
        # check destination directory
1309
1380
        if isinstance(from_paths, basestring):
1310
1381
            raise ValueError()
1600
1671
 
1601
1672
    @needs_write_lock
1602
1673
    def pull(self, source, overwrite=False, stop_revision=None,
1603
 
             change_reporter=None, possible_transports=None, local=False):
 
1674
             change_reporter=None, possible_transports=None, local=False,
 
1675
             show_base=False):
1604
1676
        source.lock_read()
1605
1677
        try:
1606
1678
            old_revision_info = self.branch.last_revision_info()
1620
1692
                                basis_tree,
1621
1693
                                this_tree=self,
1622
1694
                                pb=None,
1623
 
                                change_reporter=change_reporter)
 
1695
                                change_reporter=change_reporter,
 
1696
                                show_base=show_base)
1624
1697
                    basis_root_id = basis_tree.get_root_id()
1625
1698
                    new_root_id = new_basis_tree.get_root_id()
1626
1699
                    if basis_root_id != new_root_id:
1798
1871
            raise errors.ObjectNotLocked(self)
1799
1872
 
1800
1873
    def lock_read(self):
1801
 
        """See Branch.lock_read, and WorkingTree.unlock."""
 
1874
        """Lock the tree for reading.
 
1875
 
 
1876
        This also locks the branch, and can be unlocked via self.unlock().
 
1877
 
 
1878
        :return: A bzrlib.lock.LogicalLockResult.
 
1879
        """
1802
1880
        if not self.is_locked():
1803
1881
            self._reset_data()
1804
1882
        self.branch.lock_read()
1805
1883
        try:
1806
 
            return self._control_files.lock_read()
 
1884
            self._control_files.lock_read()
 
1885
            return LogicalLockResult(self.unlock)
1807
1886
        except:
1808
1887
            self.branch.unlock()
1809
1888
            raise
1810
1889
 
1811
1890
    def lock_tree_write(self):
1812
 
        """See MutableTree.lock_tree_write, and WorkingTree.unlock."""
 
1891
        """See MutableTree.lock_tree_write, and WorkingTree.unlock.
 
1892
 
 
1893
        :return: A bzrlib.lock.LogicalLockResult.
 
1894
        """
1813
1895
        if not self.is_locked():
1814
1896
            self._reset_data()
1815
1897
        self.branch.lock_read()
1816
1898
        try:
1817
 
            return self._control_files.lock_write()
 
1899
            self._control_files.lock_write()
 
1900
            return LogicalLockResult(self.unlock)
1818
1901
        except:
1819
1902
            self.branch.unlock()
1820
1903
            raise
1821
1904
 
1822
1905
    def lock_write(self):
1823
 
        """See MutableTree.lock_write, and WorkingTree.unlock."""
 
1906
        """See MutableTree.lock_write, and WorkingTree.unlock.
 
1907
 
 
1908
        :return: A bzrlib.lock.LogicalLockResult.
 
1909
        """
1824
1910
        if not self.is_locked():
1825
1911
            self._reset_data()
1826
1912
        self.branch.lock_write()
1827
1913
        try:
1828
 
            return self._control_files.lock_write()
 
1914
            self._control_files.lock_write()
 
1915
            return LogicalLockResult(self.unlock)
1829
1916
        except:
1830
1917
            self.branch.unlock()
1831
1918
            raise
1948
2035
 
1949
2036
        inv_delta = []
1950
2037
 
1951
 
        new_files=set()
 
2038
        all_files = set() # specified and nested files 
1952
2039
        unknown_nested_files=set()
1953
2040
        if to_file is None:
1954
2041
            to_file = sys.stdout
1955
2042
 
 
2043
        files_to_backup = []
 
2044
 
1956
2045
        def recurse_directory_to_add_files(directory):
1957
2046
            # Recurse directory and add all files
1958
2047
            # so we can check if they have changed.
1959
 
            for parent_info, file_infos in\
1960
 
                self.walkdirs(directory):
 
2048
            for parent_info, file_infos in self.walkdirs(directory):
1961
2049
                for relpath, basename, kind, lstat, fileid, kind in file_infos:
1962
2050
                    # Is it versioned or ignored?
1963
 
                    if self.path2id(relpath) or self.is_ignored(relpath):
 
2051
                    if self.path2id(relpath):
1964
2052
                        # Add nested content for deletion.
1965
 
                        new_files.add(relpath)
 
2053
                        all_files.add(relpath)
1966
2054
                    else:
1967
 
                        # Files which are not versioned and not ignored
 
2055
                        # Files which are not versioned
1968
2056
                        # should be treated as unknown.
1969
 
                        unknown_nested_files.add((relpath, None, kind))
 
2057
                        files_to_backup.append(relpath)
1970
2058
 
1971
2059
        for filename in files:
1972
2060
            # Get file name into canonical form.
1973
2061
            abspath = self.abspath(filename)
1974
2062
            filename = self.relpath(abspath)
1975
2063
            if len(filename) > 0:
1976
 
                new_files.add(filename)
 
2064
                all_files.add(filename)
1977
2065
                recurse_directory_to_add_files(filename)
1978
2066
 
1979
 
        files = list(new_files)
 
2067
        files = list(all_files)
1980
2068
 
1981
2069
        if len(files) == 0:
1982
2070
            return # nothing to do
1986
2074
 
1987
2075
        # Bail out if we are going to delete files we shouldn't
1988
2076
        if not keep_files and not force:
1989
 
            has_changed_files = len(unknown_nested_files) > 0
1990
 
            if not has_changed_files:
1991
 
                for (file_id, path, content_change, versioned, parent_id, name,
1992
 
                     kind, executable) in self.iter_changes(self.basis_tree(),
1993
 
                         include_unchanged=True, require_versioned=False,
1994
 
                         want_unversioned=True, specific_files=files):
1995
 
                    if versioned == (False, False):
1996
 
                        # The record is unknown ...
1997
 
                        if not self.is_ignored(path[1]):
1998
 
                            # ... but not ignored
1999
 
                            has_changed_files = True
2000
 
                            break
2001
 
                    elif content_change and (kind[1] is not None):
2002
 
                        # Versioned and changed, but not deleted
2003
 
                        has_changed_files = True
2004
 
                        break
 
2077
            for (file_id, path, content_change, versioned, parent_id, name,
 
2078
                 kind, executable) in self.iter_changes(self.basis_tree(),
 
2079
                     include_unchanged=True, require_versioned=False,
 
2080
                     want_unversioned=True, specific_files=files):
 
2081
                if versioned[0] == False:
 
2082
                    # The record is unknown or newly added
 
2083
                    files_to_backup.append(path[1])
 
2084
                elif (content_change and (kind[1] is not None) and
 
2085
                        osutils.is_inside_any(files, path[1])):
 
2086
                    # Versioned and changed, but not deleted, and still
 
2087
                    # in one of the dirs to be deleted.
 
2088
                    files_to_backup.append(path[1])
2005
2089
 
2006
 
            if has_changed_files:
2007
 
                # Make delta show ALL applicable changes in error message.
2008
 
                tree_delta = self.changes_from(self.basis_tree(),
2009
 
                    require_versioned=False, want_unversioned=True,
2010
 
                    specific_files=files)
2011
 
                for unknown_file in unknown_nested_files:
2012
 
                    if unknown_file not in tree_delta.unversioned:
2013
 
                        tree_delta.unversioned.extend((unknown_file,))
2014
 
                raise errors.BzrRemoveChangedFilesError(tree_delta)
 
2090
        def backup(file_to_backup):
 
2091
            backup_name = self.bzrdir._available_backup_name(file_to_backup)
 
2092
            osutils.rename(abs_path, self.abspath(backup_name))
 
2093
            return "removed %s (but kept a copy: %s)" % (file_to_backup,
 
2094
                                                         backup_name)
2015
2095
 
2016
2096
        # Build inv_delta and delete files where applicable,
2017
2097
        # do this before any modifications to inventory.
2041
2121
                        len(os.listdir(abs_path)) > 0):
2042
2122
                        if force:
2043
2123
                            osutils.rmtree(abs_path)
 
2124
                            message = "deleted %s" % (f,)
2044
2125
                        else:
2045
 
                            message = "%s is not an empty directory "\
2046
 
                                "and won't be deleted." % (f,)
 
2126
                            message = backup(f)
2047
2127
                    else:
2048
 
                        osutils.delete_any(abs_path)
2049
 
                        message = "deleted %s" % (f,)
 
2128
                        if f in files_to_backup:
 
2129
                            message = backup(f)
 
2130
                        else:
 
2131
                            osutils.delete_any(abs_path)
 
2132
                            message = "deleted %s" % (f,)
2050
2133
                elif message is not None:
2051
2134
                    # Only care if we haven't done anything yet.
2052
2135
                    message = "%s does not exist." % (f,)
2189
2272
    _marker = object()
2190
2273
 
2191
2274
    def update(self, change_reporter=None, possible_transports=None,
2192
 
               revision=None, old_tip=_marker):
 
2275
               revision=None, old_tip=_marker, show_base=False):
2193
2276
        """Update a working tree along its branch.
2194
2277
 
2195
2278
        This will update the branch if its bound too, which means we have
2232
2315
            else:
2233
2316
                if old_tip is self._marker:
2234
2317
                    old_tip = None
2235
 
            return self._update_tree(old_tip, change_reporter, revision)
 
2318
            return self._update_tree(old_tip, change_reporter, revision, show_base)
2236
2319
        finally:
2237
2320
            self.unlock()
2238
2321
 
2239
2322
    @needs_tree_write_lock
2240
 
    def _update_tree(self, old_tip=None, change_reporter=None, revision=None):
 
2323
    def _update_tree(self, old_tip=None, change_reporter=None, revision=None,
 
2324
                     show_base=False):
2241
2325
        """Update a tree to the master branch.
2242
2326
 
2243
2327
        :param old_tip: if supplied, the previous tip revision the branch,
2270
2354
            other_tree = self.branch.repository.revision_tree(old_tip)
2271
2355
            nb_conflicts = merge.merge_inner(self.branch, other_tree,
2272
2356
                                             base_tree, this_tree=self,
2273
 
                                             change_reporter=change_reporter)
 
2357
                                             change_reporter=change_reporter,
 
2358
                                             show_base=show_base)
2274
2359
            if nb_conflicts:
2275
2360
                self.add_parent_tree((old_tip, other_tree))
2276
2361
                trace.note('Rerun update after fixing the conflicts.')
2300
2385
 
2301
2386
            nb_conflicts = merge.merge_inner(self.branch, to_tree, base_tree,
2302
2387
                                             this_tree=self,
2303
 
                                             change_reporter=change_reporter)
 
2388
                                             change_reporter=change_reporter,
 
2389
                                             show_base=show_base)
2304
2390
            self.set_last_revision(revision)
2305
2391
            # TODO - dedup parents list with things merged by pull ?
2306
2392
            # reuse the tree we've updated to to set the basis:
2636
2722
 
2637
2723
        In Format2 WorkingTrees we have a single lock for the branch and tree
2638
2724
        so lock_tree_write() degrades to lock_write().
 
2725
 
 
2726
        :return: An object with an unlock method which will release the lock
 
2727
            obtained.
2639
2728
        """
2640
2729
        self.branch.lock_write()
2641
2730
        try:
2642
 
            return self._control_files.lock_write()
 
2731
            self._control_files.lock_write()
 
2732
            return self
2643
2733
        except:
2644
2734
            self.branch.unlock()
2645
2735
            raise