/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/add.py

Merge from mbp.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/add.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
 
17
from os.path import dirname
 
18
 
 
19
import bzrlib.errors as errors
 
20
from bzrlib.inventory import InventoryEntry
17
21
from bzrlib.trace import mutter, note, warning
18
22
from bzrlib.errors import NotBranchError
19
 
from bzrlib.branch import Branch
20
 
from bzrlib.osutils import quotefn
21
 
from os.path import dirname
 
23
from bzrlib.branch import is_control_file
 
24
import bzrlib.osutils
 
25
from bzrlib.workingtree import WorkingTree
22
26
 
23
27
def glob_expand_for_win32(file_list):
 
28
    if not file_list:
 
29
        return
24
30
    import glob
25
 
    
26
31
    expanded_file_list = []
27
32
    for possible_glob in file_list:
28
33
        glob_files = glob.glob(possible_glob)
40
45
    """Absorb add reports and do nothing."""
41
46
    pass
42
47
 
 
48
 
43
49
def add_reporter_print(path, kind, entry):
44
50
    """Print a line to stdout for each file that's added."""
45
 
    print "added", quotefn(path)
46
 
    
 
51
    print "added", bzrlib.osutils.quotefn(path)
 
52
 
 
53
 
47
54
def _prepare_file_list(file_list):
48
55
    """Prepare a file list for use by smart_add_*."""
49
56
    import sys
50
57
    if sys.platform == 'win32':
51
58
        file_list = glob_expand_for_win32(file_list)
52
59
    if not file_list:
53
 
        file_list = ['.']
 
60
        file_list = [u'.']
54
61
    file_list = list(file_list)
55
62
    return file_list
56
63
 
64
71
    Returns the number of files added.
65
72
    """
66
73
    file_list = _prepare_file_list(file_list)
67
 
    b = Branch.open_containing(file_list[0])
68
 
    return smart_add_branch(b, file_list, recurse, reporter)
 
74
    tree = WorkingTree.open_containing(file_list[0])[0]
 
75
    return smart_add_tree(tree, file_list, recurse, reporter)
69
76
 
70
77
        
71
 
def smart_add_branch(branch, file_list, recurse=True, reporter=add_reporter_null):
 
78
def smart_add_tree(tree, file_list, recurse=True, reporter=add_reporter_null):
72
79
    """Add files to version, optionally recursing into directories.
73
80
 
74
81
    This is designed more towards DWIM for humans than API simplicity.
75
82
    For the specific behaviour see the help for cmd_add().
76
83
 
77
 
    This yields a sequence of (path, kind, file_id) for added files.
 
84
    This calls reporter with each (path, kind, file_id) of added files.
78
85
 
79
86
    Returns the number of files added.
80
87
    """
81
88
    import os
82
89
    from bzrlib.errors import BadFileKindError, ForbiddenFileError
83
 
    import bzrlib.branch
84
 
 
85
90
    assert isinstance(recurse, bool)
86
91
 
87
92
    file_list = _prepare_file_list(file_list)
88
93
    user_list = file_list[:]
89
 
    inv = branch.read_working_inventory()
90
 
    tree = branch.working_tree()
 
94
    inv = tree.read_working_inventory()
91
95
    count = 0
92
96
 
93
97
    for f in file_list:
94
 
        rf = branch.relpath(f)
95
 
        af = branch.abspath(rf)
 
98
        rf = tree.relpath(f)
 
99
        af = tree.abspath(rf)
96
100
 
97
101
        kind = bzrlib.osutils.file_kind(af)
98
102
 
99
 
        if kind != 'file' and kind != 'directory':
 
103
        if not InventoryEntry.versionable_kind(kind):
100
104
            if f in user_list:
101
105
                raise BadFileKindError("cannot add %s of type %s" % (f, kind))
102
106
            else:
103
107
                warning("skipping %s (can't add file of kind '%s')", f, kind)
104
108
                continue
105
109
 
106
 
        mutter("smart add of %r, abs=%r" % (f, af))
 
110
        mutter("smart add of %r, abs=%r", f, af)
107
111
        
108
 
        if bzrlib.branch.is_control_file(af):
 
112
        if is_control_file(af):
109
113
            raise ForbiddenFileError('cannot add control file %s' % f)
110
114
            
111
115
        versioned = (inv.path2id(rf) != None)
112
116
 
113
117
        if kind == 'directory':
114
118
            try:
115
 
                sub_branch = Branch.open(af)
 
119
                sub_branch = WorkingTree(af)
116
120
                sub_tree = True
117
121
            except NotBranchError:
118
122
                sub_tree = False
 
123
            except errors.UnsupportedFormatError:
 
124
                sub_tree = True
119
125
        else:
120
126
            sub_tree = False
121
127
 
122
128
        if rf == '':
123
 
            mutter("branch root doesn't need to be added")
 
129
            mutter("tree root doesn't need to be added")
124
130
            sub_tree = False
125
131
        elif versioned:
126
 
            mutter("%r is already versioned" % f)
 
132
            mutter("%r is already versioned", f)
127
133
        elif sub_tree:
128
 
            mutter("%r is a bzr tree" %f)
 
134
            mutter("%r is a bzr tree", f)
129
135
        else:
130
 
            count += __add_one(branch, inv, rf, kind, reporter)
 
136
            count += __add_one(tree, inv, rf, kind, reporter)
131
137
 
132
138
        if kind == 'directory' and recurse and not sub_tree:
133
139
            for subf in os.listdir(af):
134
140
                subp = os.path.join(rf, subf)
135
141
                if subf == bzrlib.BZRDIR:
136
 
                    mutter("skip control directory %r" % subp)
 
142
                    mutter("skip control directory %r", subp)
137
143
                elif tree.is_ignored(subp):
138
 
                    mutter("skip ignored sub-file %r" % subp)
 
144
                    mutter("skip ignored sub-file %r", subp)
139
145
                else:
140
 
                    mutter("queue to add sub-file %r" % subp)
141
 
                    file_list.append(branch.abspath(subp))
 
146
                    mutter("queue to add sub-file %r", subp)
 
147
                    file_list.append(tree.abspath(subp))
142
148
 
143
149
 
144
150
    mutter('added %d entries', count)
145
151
    
146
152
    if count > 0:
147
 
        branch._write_inventory(inv)
 
153
        tree._write_inventory(inv)
148
154
 
149
155
    return count
150
156
 
151
 
def __add_one(branch, inv, path, kind, reporter):
 
157
def __add_one(tree, inv, path, kind, reporter):
152
158
    """Add a file or directory, automatically add unversioned parents."""
153
159
 
154
160
    # Nothing to do if path is already versioned.
155
 
    # This is safe from infinite recursion because the branch root is
 
161
    # This is safe from infinite recursion because the tree root is
156
162
    # always versioned.
157
163
    if inv.path2id(path) != None:
158
164
        return 0
159
165
 
160
166
    # add parent
161
 
    count = __add_one(branch, inv, dirname(path), 'directory', reporter)
 
167
    count = __add_one(tree, inv, dirname(path), 'directory', reporter)
162
168
 
163
169
    entry = inv.add_path(path, kind=kind)
164
 
    mutter("added %r kind %r file_id={%s}" % (path, kind, entry.file_id))
 
170
    mutter("added %r kind %r file_id={%s}", path, kind, entry.file_id)
165
171
    reporter(path, kind, entry)
166
172
 
167
173
    return count + 1