← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to zap.py

  • Committer: Aaron Bentley
  • Date: 2007-01-16 22:17:34 UTC
  • Revision ID: abentley@panoramicfeedback.com-20070116221734-ijxdmpekl9swpf4x
Tweak colordiff/shelf changes

Show diffs side-by-side

added added

removed removed

Lines of Context:
zap.py
1
1
from shutil import rmtree
2
2
 
 
3
from bzrlib.branch import Branch
3
4
from bzrlib.errors import NoWorkingTree, NotLocalUrl, NotBranchError
4
 
from bzrlib.delta import compare_trees
5
5
from bzrlib.workingtree import WorkingTree
6
6
 
7
 
from errors import NotCheckout, UncommittedCheckout
8
 
 
9
 
 
10
 
def zap(path):
 
7
from errors import (NotCheckout, UncommittedCheckout, ParentMissingRevisions, 
 
8
                    NoParent)
 
9
 
 
10
 
 
11
def zap(path, remove_branch=False):
11
12
    try:
12
13
        wt = WorkingTree.open(path)
13
14
    except (NoWorkingTree, NotBranchError):
14
15
        raise NotCheckout(path)
15
16
    tree_base = wt.bzrdir.transport.base
16
 
    branch_base = wt.branch.bzrdir.transport.base
 
17
    branch = wt.branch
 
18
    branch_base = branch.bzrdir.transport.base
17
19
    if tree_base == branch_base:
18
20
        raise NotCheckout(path)
19
 
    delta = compare_trees(wt.basis_tree(), wt, want_unchanged=False)
 
21
    delta = wt.changes_from(wt.basis_tree(), want_unchanged=False)
20
22
    if delta.has_changed():
21
23
        raise UncommittedCheckout()
 
24
    if remove_branch:
 
25
        parent_loc = branch.get_parent()
 
26
        if parent_loc is None:
 
27
            raise NoParent()
 
28
        parent = Branch.open(parent_loc)
 
29
        p_ancestry = parent.repository.get_ancestry(parent.last_revision())
 
30
        if branch.last_revision() not in p_ancestry:
 
31
            raise ParentMissingRevisions(branch.get_parent())
22
32
    rmtree(path)
 
33
    if remove_branch:
 
34
        t = branch.bzrdir.transport
 
35
        while t.base != branch_base:
 
36
            t = t.clone('..')
 
37
        t = t.clone('..')
 
38
        t.delete_tree('')
23
39
 
24
40
 
25
41
def test_suite():
39
55
            BranchReferenceFormat().initialize(checkout, wt.branch)
40
56
            return checkout.create_workingtree()
41
57
 
 
58
        def make_checkout2(self):
 
59
            wt = self.make_checkout()
 
60
            wt2 = wt.branch.bzrdir.sprout('source2').open_workingtree()
 
61
            os.mkdir('checkout2')
 
62
            checkout = BzrDirMetaFormat1().initialize('checkout2')
 
63
            BranchReferenceFormat().initialize(checkout, wt2.branch)
 
64
            return checkout.create_workingtree()
 
65
 
42
66
        def test_is_checkout(self):
43
67
            self.assertRaises(NotCheckout, zap, '.')
44
68
            wt = BzrDir.create_standalone_workingtree('.')
49
73
            self.assertIs(True, os.path.exists('checkout'))
50
74
            zap('checkout')
51
75
            self.assertIs(False, os.path.exists('checkout'))
 
76
            self.assertIs(True, os.path.exists('source'))
 
77
 
 
78
        def test_zap_branch(self):
 
79
            self.make_checkout2()
 
80
            base = WorkingTree.open('checkout').branch.base
 
81
            self.assertIs(True, os.path.exists('checkout'))
 
82
            self.assertRaises(NoParent, zap, 'checkout', remove_branch=True)
 
83
            self.assertIs(True, os.path.exists('checkout'))
 
84
            self.assertIs(True, os.path.exists('source'))
 
85
            zap('checkout2', remove_branch=True)
 
86
            self.assertIs(False, os.path.exists('checkout2'))
 
87
            self.assertIs(False, os.path.exists('source2'))
52
88
 
53
89
        def test_checks_modified(self):
54
90
            checkout = self.make_checkout()