← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/add.py

  • Committer: Robert Collins
  • Date: 2005-08-25 06:20:21 UTC
  • mto: (974.1.50) (1185.1.10) (1092.3.1)
  • mto: This revision was merged to the branch mainline in revision 1139.
  • Revision ID: robertc@robertcollins.net-20050825062021-d7d1b19582ceb297
should run tests before committing, tsk. move an import to fix

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 bzrlib.trace import mutter, note, warning
 
17
from trace import mutter, note
18
18
from bzrlib.errors import NotBranchError
19
19
from bzrlib.branch import Branch
20
 
from bzrlib.osutils import quotefn
21
20
 
22
21
def glob_expand_for_win32(file_list):
23
22
    import glob
34
33
            expanded_file_list += glob_files
35
34
    return expanded_file_list
36
35
 
37
 
 
38
 
def add_reporter_null(path, kind, entry):
39
 
    """Absorb add reports and do nothing."""
 
36
def _NullAddCallback(entry):
40
37
    pass
41
38
 
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)
 
39
def _PrintAddCallback(entry):
 
40
    from bzrlib.osutils import quotefn
 
41
    print "added", quotefn(entry.name)
45
42
    
46
43
def _prepare_file_list(file_list):
47
44
    """Prepare a file list for use by smart_add_*."""
51
48
    if not file_list:
52
49
        file_list = ['.']
53
50
    file_list = list(file_list)
 
51
    assert not isinstance(file_list, basestring)
54
52
    return file_list
55
53
 
56
 
 
57
 
def smart_add(file_list, recurse=True, reporter=add_reporter_null):
 
54
def smart_add(file_list, verbose=True, recurse=True, callback=_NullAddCallback):
58
55
    """Add files to version, optionally recursing into directories.
59
56
 
60
57
    This is designed more towards DWIM for humans than API simplicity.
61
58
    For the specific behaviour see the help for cmd_add().
62
 
 
63
 
    Returns the number of files added.
64
59
    """
65
60
    file_list = _prepare_file_list(file_list)
66
61
    b = Branch(file_list[0], find_root=True)
67
 
    return smart_add_branch(b, file_list, recurse, reporter)
68
 
 
 
62
    return smart_add_branch(b, file_list, verbose, recurse)
69
63
        
70
 
def smart_add_branch(branch, file_list, recurse=True, reporter=add_reporter_null):
 
64
def smart_add_branch(branch, file_list, verbose=True, recurse=True,
 
65
                     callback=_NullAddCallback):
71
66
    """Add files to version, optionally recursing into directories.
72
67
 
73
68
    This is designed more towards DWIM for humans than API simplicity.
74
69
    For the specific behaviour see the help for cmd_add().
75
 
 
76
 
    This yields a sequence of (path, kind, file_id) for added files.
77
 
 
78
 
    Returns the number of files added.
79
70
    """
80
71
    import os
81
72
    import sys
84
75
    import bzrlib.branch
85
76
    import bzrlib.osutils
86
77
 
87
 
    assert isinstance(recurse, bool)
88
 
 
89
78
    file_list = _prepare_file_list(file_list)
90
79
    user_list = file_list[:]
91
80
    inv = branch.read_working_inventory()
102
91
            if f in user_list:
103
92
                raise BadFileKindError("cannot add %s of type %s" % (f, kind))
104
93
            else:
105
 
                warning("skipping %s (can't add file of kind '%s')", f, kind)
 
94
                print "skipping %s (can't add file of kind '%s')" % (f, kind)
106
95
                continue
107
96
 
108
97
        mutter("smart add of %r, abs=%r" % (f, af))
132
121
            entry = inv.add_path(rf, kind=kind)
133
122
            mutter("added %r kind %r file_id={%s}" % (rf, kind, entry.file_id))
134
123
            count += 1 
135
 
            reporter(rf, kind, entry)
 
124
            callback(entry)
136
125
 
137
126
        if kind == 'directory' and recurse and not sub_tree:
138
127
            for subf in os.listdir(af):
145
134
                    mutter("queue to add sub-file %r" % subp)
146
135
                    file_list.append(branch.abspath(subp))
147
136
 
148
 
 
149
 
    mutter('added %d entries', count)
150
 
    
151
137
    if count > 0:
 
138
        if verbose:
 
139
            note('added %d' % count)
152
140
        branch._write_inventory(inv)
153
 
 
154
 
    return count
 
141
    else:
 
142
        print "nothing new to add"
 
143
        # should this return 1 to the shell?