← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/add.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-25 03:09:43 UTC
  • Revision ID: mbp@sourcefrog.net-20050325030943-e2ba06e151908aee
- clean up smart_add code, and make it commit the inventory
  when done.

  still not perfect handling of already-versioned files.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/add.py
32
32
    b = bzrlib.branch.Branch(file_list[0], find_root=True)
33
33
    inv = b.read_working_inventory()
34
34
    tree = b.working_tree()
35
 
    dirty = False
36
 
 
37
 
    def add_one(rf, kind):
38
 
        file_id = bzrlib.branch.gen_file_id(rf)
39
 
        inv.add_path(rf, kind=kind, file_id=file_id)
40
 
        bzrlib.mutter("added %r kind %r file_id={%s}" % (rf, kind, file_id))
41
 
        dirty = True
42
 
        if verbose:
43
 
            bzrlib.textui.show_status('A', kind, quotefn(f))
44
 
        
 
35
    count = 0
45
36
 
46
37
    for f in file_list:
47
38
        rf = b.relpath(f)
48
39
        af = b.abspath(rf)
49
40
 
 
41
        ## TODO: It's OK to add root but only in recursive mode
 
42
 
50
43
        bzrlib.mutter("smart add of %r" % f)
51
44
        
52
45
        if bzrlib.branch.is_control_file(af):
53
46
            bailout("cannot add control file %r" % af)
54
47
 
55
48
        kind = bzrlib.osutils.file_kind(f)
 
49
 
 
50
        if kind != 'file' and kind != 'directory':
 
51
            bailout("can't add file of kind %r" % kind)
 
52
            
56
53
        versioned = (inv.path2id(rf) != None)
57
54
 
58
 
        ## TODO: It's OK to add '.' but only in recursive mode
59
 
 
60
 
        if kind == 'file':
61
 
            if versioned:
62
 
                bzrlib.warning("%r is already versioned" % f)
63
 
                continue
64
 
            else:
65
 
                add_one(rf, kind)
66
 
        elif kind == 'directory':
67
 
            if versioned and not recurse:
68
 
                bzrlib.warning("%r is already versioned" % f)
69
 
                continue
70
 
            
71
 
            if not versioned:
72
 
                add_one(rf, kind)
73
 
 
74
 
            if recurse:
75
 
                for subf in os.listdir(af):
76
 
                    subp = appendpath(rf, subf)
77
 
                    if tree.is_ignored(subp):
78
 
                        mutter("skip ignored sub-file %r" % subp)
79
 
                    else:
80
 
                        mutter("queue to add sub-file %r" % (subp))
81
 
                        file_list.append(subp)
 
55
        if versioned:
 
56
            bzrlib.warning("%r is already versioned" % f)
82
57
        else:
83
 
            bailout("can't smart_add file kind %r" % kind)
84
 
 
85
 
    if dirty:
 
58
            file_id = bzrlib.branch.gen_file_id(rf)
 
59
            inv.add_path(rf, kind=kind, file_id=file_id)
 
60
            bzrlib.mutter("added %r kind %r file_id={%s}" % (rf, kind, file_id))
 
61
            count += 1 
 
62
            if verbose:
 
63
                bzrlib.textui.show_status('A', kind, quotefn(f))
 
64
 
 
65
        if kind == 'directory' and recurse:
 
66
            for subf in os.listdir(af):
 
67
                subp = appendpath(rf, subf)
 
68
                if tree.is_ignored(subp):
 
69
                    mutter("skip ignored sub-file %r" % subp)
 
70
                else:
 
71
                    mutter("queue to add sub-file %r" % (subp))
 
72
                    file_list.append(subp)
 
73
 
 
74
    if count > 0:
 
75
        print '* added %d' % count
86
76
        b._write_inventory(inv)