40
31
expanded_file_list += glob_files
41
32
return expanded_file_list
44
def add_reporter_null(path, kind, entry):
45
"""Absorb add reports and do nothing."""
49
def add_reporter_print(path, kind, entry):
50
"""Print a line to stdout for each file that's added."""
51
print "added", bzrlib.osutils.quotefn(path)
54
def _prepare_file_list(file_list):
55
"""Prepare a file list for use by smart_add_*."""
34
def smart_add(file_list, verbose=True, recurse=True):
35
"""Add files to version, optionally recursing into directories.
37
This is designed more towards DWIM for humans than API simplicity.
38
For the specific behaviour see the help for cmd_add().
42
from bzrlib.osutils import quotefn, kind_marker
43
from bzrlib.errors import BadFileKindError, ForbiddenFileError
57
47
if sys.platform == 'win32':
58
48
file_list = glob_expand_for_win32(file_list)
61
file_list = list(file_list)
65
def smart_add(file_list, recurse=True, reporter=add_reporter_null):
66
"""Add files to version, optionally recursing into directories.
68
This is designed more towards DWIM for humans than API simplicity.
69
For the specific behaviour see the help for cmd_add().
71
Returns the number of files added.
73
file_list = _prepare_file_list(file_list)
74
tree = WorkingTree.open_containing(file_list[0])[0]
75
return smart_add_tree(tree, file_list, recurse, reporter)
78
def smart_add_tree(tree, file_list, recurse=True, reporter=add_reporter_null):
79
"""Add files to version, optionally recursing into directories.
81
This is designed more towards DWIM for humans than API simplicity.
82
For the specific behaviour see the help for cmd_add().
84
This calls reporter with each (path, kind, file_id) of added files.
86
Returns the number of files added.
89
from bzrlib.errors import BadFileKindError, ForbiddenFileError
90
assert isinstance(recurse, bool)
92
file_list = _prepare_file_list(file_list)
93
53
user_list = file_list[:]
94
inv = tree.read_working_inventory()
54
assert not isinstance(file_list, basestring)
55
b = bzrlib.branch.Branch(file_list[0], find_root=True)
56
inv = b.read_working_inventory()
57
tree = b.working_tree()
97
60
for f in file_list:
101
64
kind = bzrlib.osutils.file_kind(af)
103
if not InventoryEntry.versionable_kind(kind):
66
if kind != 'file' and kind != 'directory':
104
67
if f in user_list:
105
68
raise BadFileKindError("cannot add %s of type %s" % (f, kind))
107
warning("skipping %s (can't add file of kind '%s')", f, kind)
70
print "skipping %s (can't add file of kind '%s')" % (f, kind)
110
mutter("smart add of %r, abs=%r", f, af)
73
mutter("smart add of %r, abs=%r" % (f, af))
112
if is_control_file(af):
75
if bzrlib.branch.is_control_file(af):
113
76
raise ForbiddenFileError('cannot add control file %s' % f)
115
78
versioned = (inv.path2id(rf) != None)
117
if kind == 'directory':
119
sub_branch = WorkingTree(af)
121
except NotBranchError:
123
except errors.UnsupportedFormatError:
129
mutter("tree root doesn't need to be added")
81
mutter("branch root doesn't need to be added")
132
mutter("%r is already versioned", f)
134
mutter("%r is a bzr tree", f)
83
mutter("%r is already versioned" % f)
136
count += __add_one(tree, inv, rf, kind, reporter)
138
if kind == 'directory' and recurse and not sub_tree:
85
file_id = bzrlib.branch.gen_file_id(rf)
86
inv.add_path(rf, kind=kind, file_id=file_id)
87
mutter("added %r kind %r file_id={%s}" % (rf, kind, file_id))
90
print 'added', quotefn(f)
92
if kind == 'directory' and recurse:
139
93
for subf in os.listdir(af):
140
94
subp = os.path.join(rf, subf)
141
95
if subf == bzrlib.BZRDIR:
142
mutter("skip control directory %r", subp)
96
mutter("skip control directory %r" % subp)
143
97
elif tree.is_ignored(subp):
144
mutter("skip ignored sub-file %r", subp)
98
mutter("skip ignored sub-file %r" % subp)
146
mutter("queue to add sub-file %r", subp)
147
file_list.append(tree.abspath(subp))
150
mutter('added %d entries', count)
100
mutter("queue to add sub-file %r" % subp)
101
file_list.append(b.abspath(subp))
153
tree._write_inventory(inv)
157
def __add_one(tree, inv, path, kind, reporter):
158
"""Add a file or directory, automatically add unversioned parents."""
160
# Nothing to do if path is already versioned.
161
# This is safe from infinite recursion because the tree root is
163
if inv.path2id(path) != None:
167
count = __add_one(tree, inv, dirname(path), 'directory', reporter)
169
entry = inv.add_path(path, kind=kind)
170
mutter("added %r kind %r file_id={%s}", path, kind, entry.file_id)
171
reporter(path, kind, entry)
105
note('added %d' % count)
106
b._write_inventory(inv)
108
print "nothing new to add"
109
# should this return 1 to the shell?
added
removed
bzrlib/add.py
