← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/add.py

  • Committer: John Arbash Meinel
  • Date: 2006-04-25 15:05:42 UTC
  • mfrom: (1185.85.85 bzr-encoding)
  • mto: This revision was merged to the branch mainline in revision 1752.
  • Revision ID: john@arbash-meinel.com-20060425150542-c7b518dca9928691
[merge] the old bzr-encoding changes, reparenting them on bzr.dev

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 sys
17
18
from os.path import dirname
18
19
 
19
20
import bzrlib.errors as errors
43
44
 
44
45
def _prepare_file_list(file_list):
45
46
    """Prepare a file list for use by smart_add_*."""
46
 
    import sys
47
47
    if sys.platform == 'win32':
48
48
        file_list = glob_expand_for_win32(file_list)
49
49
    if not file_list:
52
52
    return file_list
53
53
 
54
54
 
55
 
def add_action_null(inv, path, kind):
56
 
    """Absorb add actions and do nothing."""
57
 
    pass
58
 
 
59
 
def add_action_print(inv, path, kind):
60
 
    """Print a line to stdout for each file that would be added."""
61
 
    print "added", bzrlib.osutils.quotefn(path)
62
 
 
63
 
def add_action_add(inv, path, kind):
64
 
    """Add each file to the given inventory. Produce no output."""
65
 
    entry = inv.add_path(path, kind=kind)
66
 
    mutter("added %r kind %r file_id={%s}" % (path, kind, entry.file_id))
67
 
 
68
 
def add_action_add_and_print(inv, path, kind):
69
 
    """Add each file to the given inventory, and print a line to stdout."""
70
 
    add_action_add(inv, path, kind)
71
 
    add_action_print(inv, path, kind)
72
 
 
73
 
 
74
 
def smart_add(file_list, recurse=True, action=add_action_add):
 
55
class AddAction(object):
 
56
    """A class which defines what action to take when adding a file."""
 
57
 
 
58
    should_add = False
 
59
    should_print = False
 
60
 
 
61
    def __init__(self, to_file=None, should_add=None, should_print=None):
 
62
        self._to_file = to_file
 
63
        if to_file is None:
 
64
            self._to_file = sys.stdout
 
65
        if should_add is not None:
 
66
            self.should_add = should_add
 
67
        if should_print is not None:
 
68
            self.should_print = should_print
 
69
 
 
70
    def __call__(self, inv, path, kind):
 
71
        """Add path to inventory.
 
72
 
 
73
        The default action does nothing.
 
74
 
 
75
        :param inv: The inventory we are working with.
 
76
        :param path: The path being added
 
77
        :param kind: The kind of the object being added.
 
78
        """
 
79
        if self.should_add:
 
80
            self._add_to_inv(inv, path, kind)
 
81
        if self.should_print:
 
82
            self._print(inv, path, kind)
 
83
 
 
84
    def _print(self, inv, path, kind):
 
85
        self._to_file.write('added ')
 
86
        self._to_file.write(bzrlib.osutils.quotefn(path))
 
87
        self._to_file.write('\n')
 
88
 
 
89
    def _add_to_inv(self, inv, path, kind):
 
90
        entry = inv.add_path(path, kind=kind)
 
91
        mutter("added %r kind %r file_id={%s}", path, kind, entry.file_id)
 
92
 
 
93
 
 
94
# TODO: jam 20050105 These could be used for compatibility
 
95
#       however, they bind against the current stdout, not the
 
96
#       one which exists at the time they are called, so they
 
97
#       don't work for the test suite.
 
98
# deprecated
 
99
add_action_null = AddAction()
 
100
add_action_add = AddAction(should_add=True)
 
101
add_action_print = AddAction(should_print=True)
 
102
add_action_add_and_print = AddAction(should_add=True, should_print=True)
 
103
 
 
104
 
 
105
def smart_add(file_list, recurse=True, action=None):
75
106
    """Add files to version, optionally recursing into directories.
76
107
 
77
108
    This is designed more towards DWIM for humans than API simplicity.
78
109
    For the specific behaviour see the help for cmd_add().
79
110
 
80
111
    Returns the number of files added.
 
112
 
81
113
    """
82
114
    file_list = _prepare_file_list(file_list)
83
115
    tree = WorkingTree.open_containing(file_list[0])[0]
84
 
    return smart_add_tree(tree, file_list, recurse, action)
85
 
 
86
 
def smart_add_tree(tree, file_list, recurse=True, action=add_action_add):
 
116
    return smart_add_tree(tree, file_list, recurse, action=action)
 
117
 
 
118
 
 
119
def smart_add_tree(tree, file_list, recurse=True, action=None):
87
120
    """Add files to version, optionally recursing into directories.
88
121
 
89
122
    This is designed more towards DWIM for humans than API simplicity.
96
129
    import os, errno
97
130
    from bzrlib.errors import BadFileKindError, ForbiddenFileError
98
131
    assert isinstance(recurse, bool)
 
132
    if action is None:
 
133
        action = AddAction(should_add=True)
99
134
 
100
135
    file_list = _prepare_file_list(file_list)
101
136
    user_list = file_list[:]
173
208
 
174
209
    return added, ignored
175
210
 
 
211
 
176
212
def __add_one(tree, inv, path, kind, action):
177
213
    """Add a file or directory, automatically add unversioned parents."""
178
214