1
# Copyright (C) 2005 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
from bzrlib.trace import mutter, note, warning
18
from bzrlib.errors import NotBranchError
19
from bzrlib.branch import Branch
20
from bzrlib.osutils import quotefn
22
def glob_expand_for_win32(file_list):
25
expanded_file_list = []
26
for possible_glob in file_list:
27
glob_files = glob.glob(possible_glob)
30
# special case to let the normal code path handle
31
# files that do not exists
32
expanded_file_list.append(possible_glob)
34
expanded_file_list += glob_files
35
return expanded_file_list
38
def add_reporter_null(path, kind, entry):
39
"""Absorb add reports and do nothing."""
42
def add_reporter_print(path, kind, entry):
43
"""Print a line to stdout for each file that's added."""
44
print "added", quotefn(path)
46
def _prepare_file_list(file_list):
47
"""Prepare a file list for use by smart_add_*."""
49
if sys.platform == 'win32':
50
file_list = glob_expand_for_win32(file_list)
53
file_list = list(file_list)
57
def smart_add(file_list, recurse=True, reporter=add_reporter_null):
58
"""Add files to version, optionally recursing into directories.
60
This is designed more towards DWIM for humans than API simplicity.
61
For the specific behaviour see the help for cmd_add().
63
Returns the number of files added.
65
file_list = _prepare_file_list(file_list)
66
b = Branch(file_list[0], find_root=True)
67
return smart_add_branch(b, file_list, recurse, reporter)
70
def smart_add_branch(branch, file_list, recurse=True, reporter=add_reporter_null):
71
"""Add files to version, optionally recursing into directories.
73
This is designed more towards DWIM for humans than API simplicity.
74
For the specific behaviour see the help for cmd_add().
76
This yields a sequence of (path, kind, file_id) for added files.
78
Returns the number of files added.
82
from bzrlib.osutils import quotefn
83
from bzrlib.errors import BadFileKindError, ForbiddenFileError
87
assert isinstance(recurse, bool)
89
file_list = _prepare_file_list(file_list)
90
user_list = file_list[:]
91
inv = branch.read_working_inventory()
92
tree = branch.working_tree()
96
rf = branch.relpath(f)
97
af = branch.abspath(rf)
99
kind = bzrlib.osutils.file_kind(af)
101
if kind != 'file' and kind != 'directory':
103
raise BadFileKindError("cannot add %s of type %s" % (f, kind))
105
warning("skipping %s (can't add file of kind '%s')", f, kind)
108
mutter("smart add of %r, abs=%r" % (f, af))
110
if bzrlib.branch.is_control_file(af):
111
raise ForbiddenFileError('cannot add control file %s' % f)
113
versioned = (inv.path2id(rf) != None)
115
if kind == 'directory':
117
sub_branch = Branch(af, find_root=False)
119
except NotBranchError:
125
mutter("branch root doesn't need to be added")
128
mutter("%r is already versioned" % f)
130
mutter("%r is a bzr tree" %f)
132
entry = inv.add_path(rf, kind=kind)
133
mutter("added %r kind %r file_id={%s}" % (rf, kind, entry.file_id))
135
reporter(rf, kind, entry)
137
if kind == 'directory' and recurse and not sub_tree:
138
for subf in os.listdir(af):
139
subp = os.path.join(rf, subf)
140
if subf == bzrlib.BZRDIR:
141
mutter("skip control directory %r" % subp)
142
elif tree.is_ignored(subp):
143
mutter("skip ignored sub-file %r" % subp)
145
mutter("queue to add sub-file %r" % subp)
146
file_list.append(branch.abspath(subp))
149
mutter('added %d entries', count)
152
branch._write_inventory(inv)
added
removed
bzrlib/add.py
