← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/add.py

  • Committer: Martin Pool
  • Date: 2005-07-20 17:31:25 UTC
  • Revision ID: mbp@sourcefrog.net-20050720173125-e5974e73bd5fc31e
- add back update-hashes command for testing/profiling

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
import os, sys
 
18
import bzrlib
 
19
 
17
20
from trace import mutter, note
18
 
from bzrlib.errors import NotBranchError
19
 
from bzrlib.branch import Branch
20
21
 
21
22
def glob_expand_for_win32(file_list):
22
23
    import glob
33
34
            expanded_file_list += glob_files
34
35
    return expanded_file_list
35
36
 
36
 
def _NullAddCallback(entry):
37
 
    pass
38
 
 
39
 
def _PrintAddCallback(entry):
40
 
    from bzrlib.osutils import quotefn
41
 
    print "added", quotefn(entry.name)
 
37
def smart_add(file_list, verbose=True, recurse=True):
 
38
    """Add files to version, optionally recursing into directories.
 
39
 
 
40
    This is designed more towards DWIM for humans than API simplicity.
 
41
    For the specific behaviour see the help for cmd_add().
 
42
    """
 
43
    from bzrlib.osutils import quotefn, kind_marker
 
44
    from bzrlib.errors import BadFileKindError, ForbiddenFileError
 
45
 
 
46
    assert file_list
42
47
    
43
 
def _prepare_file_list(file_list):
44
 
    """Prepare a file list for use by smart_add_*."""
45
 
    import sys
46
48
    if sys.platform == 'win32':
47
49
        file_list = glob_expand_for_win32(file_list)
48
 
    if not file_list:
49
 
        file_list = ['.']
50
 
    file_list = list(file_list)
 
50
    
 
51
    user_list = file_list[:]
51
52
    assert not isinstance(file_list, basestring)
52
 
    return file_list
53
 
 
54
 
def smart_add(file_list, verbose=True, recurse=True, callback=_NullAddCallback):
55
 
    """Add files to version, optionally recursing into directories.
56
 
 
57
 
    This is designed more towards DWIM for humans than API simplicity.
58
 
    For the specific behaviour see the help for cmd_add().
59
 
    """
60
 
    file_list = _prepare_file_list(file_list)
61
 
    b = Branch(file_list[0], find_root=True)
62
 
    return smart_add_branch(b, file_list, verbose, recurse)
63
 
        
64
 
def smart_add_branch(branch, file_list, verbose=True, recurse=True,
65
 
                     callback=_NullAddCallback):
66
 
    """Add files to version, optionally recursing into directories.
67
 
 
68
 
    This is designed more towards DWIM for humans than API simplicity.
69
 
    For the specific behaviour see the help for cmd_add().
70
 
    """
71
 
    import os
72
 
    import sys
73
 
    from bzrlib.osutils import quotefn
74
 
    from bzrlib.errors import BadFileKindError, ForbiddenFileError
75
 
    import bzrlib.branch
76
 
    import bzrlib.osutils
77
 
 
78
 
    file_list = _prepare_file_list(file_list)
79
 
    user_list = file_list[:]
80
 
    inv = branch.read_working_inventory()
81
 
    tree = branch.working_tree()
 
53
    b = bzrlib.branch.Branch(file_list[0], find_root=True)
 
54
    inv = b.read_working_inventory()
 
55
    tree = b.working_tree()
82
56
    count = 0
83
57
 
84
58
    for f in file_list:
85
 
        rf = branch.relpath(f)
86
 
        af = branch.abspath(rf)
 
59
        rf = b.relpath(f)
 
60
        af = b.abspath(rf)
87
61
 
88
62
        kind = bzrlib.osutils.file_kind(af)
89
63
 
101
75
            
102
76
        versioned = (inv.path2id(rf) != None)
103
77
 
104
 
        if kind == 'directory':
105
 
            try:
106
 
                sub_branch = Branch(af, find_root=False)
107
 
                sub_tree = True
108
 
            except NotBranchError:
109
 
                sub_tree = False
110
 
        else:
111
 
            sub_tree = False
112
 
 
113
78
        if rf == '':
114
79
            mutter("branch root doesn't need to be added")
115
 
            sub_tree = False
116
80
        elif versioned:
117
81
            mutter("%r is already versioned" % f)
118
 
        elif sub_tree:
119
 
            mutter("%r is a bzr tree" %f)
120
82
        else:
121
 
            entry = inv.add_path(rf, kind=kind)
122
 
            mutter("added %r kind %r file_id={%s}" % (rf, kind, entry.file_id))
 
83
            file_id = bzrlib.branch.gen_file_id(rf)
 
84
            inv.add_path(rf, kind=kind, file_id=file_id)
 
85
            mutter("added %r kind %r file_id={%s}" % (rf, kind, file_id))
123
86
            count += 1 
124
 
            callback(entry)
125
 
 
126
 
        if kind == 'directory' and recurse and not sub_tree:
 
87
 
 
88
            print 'added', quotefn(f)
 
89
 
 
90
        if kind == 'directory' and recurse:
127
91
            for subf in os.listdir(af):
128
92
                subp = os.path.join(rf, subf)
129
93
                if subf == bzrlib.BZRDIR:
132
96
                    mutter("skip ignored sub-file %r" % subp)
133
97
                else:
134
98
                    mutter("queue to add sub-file %r" % subp)
135
 
                    file_list.append(branch.abspath(subp))
 
99
                    file_list.append(b.abspath(subp))
136
100
 
137
101
    if count > 0:
138
102
        if verbose:
139
103
            note('added %d' % count)
140
 
        branch._write_inventory(inv)
141
 
    else:
142
 
        print "nothing new to add"
143
 
        # should this return 1 to the shell?
 
104
        b._write_inventory(inv)