← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/add.py

  • Committer: Aaron Bentley
  • Date: 2005-08-25 13:10:25 UTC
  • mfrom: (974.1.38)
  • mto: (1092.1.42) (1185.3.4)
  • mto: This revision was merged to the branch mainline in revision 1178.
  • Revision ID: abentley@panoramicfeedback.com-20050825131025-2aa94bcbbd646a00
Fixed return value when not an ImmutableStore

Show diffs side-by-side

added added

removed removed

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