← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to zap.py

  • Committer: Aaron Bentley
  • Date: 2006-03-24 19:01:30 UTC
  • Revision ID: abentley@panoramicfeedback.com-20060324190130-2208c693486a8b33
Added apache index scraping to the branches command

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
4
3
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, ParentMissingRevisions, 
8
 
                    NoParent)
9
 
 
10
 
 
11
 
def zap(path, remove_branch=False):
 
7
from errors import NotCheckout, UncommittedCheckout
 
8
 
 
9
 
 
10
def zap(path):
12
11
    try:
13
12
        wt = WorkingTree.open(path)
14
13
    except (NoWorkingTree, NotBranchError):
15
14
        raise NotCheckout(path)
16
15
    tree_base = wt.bzrdir.transport.base
17
 
    branch = wt.branch
18
 
    branch_base = branch.bzrdir.transport.base
 
16
    branch_base = wt.branch.bzrdir.transport.base
19
17
    if tree_base == branch_base:
20
18
        raise NotCheckout(path)
21
 
    delta = wt.changes_from(wt.basis_tree(), want_unchanged=False)
 
19
    delta = compare_trees(wt.basis_tree(), wt, want_unchanged=False)
22
20
    if delta.has_changed():
23
21
        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())
32
22
    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('')
39
23
 
40
24
 
41
25
def test_suite():
55
39
            BranchReferenceFormat().initialize(checkout, wt.branch)
56
40
            return checkout.create_workingtree()
57
41
 
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
 
 
66
42
        def test_is_checkout(self):
67
43
            self.assertRaises(NotCheckout, zap, '.')
68
44
            wt = BzrDir.create_standalone_workingtree('.')
73
49
            self.assertIs(True, os.path.exists('checkout'))
74
50
            zap('checkout')
75
51
            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'))
88
52
 
89
53
        def test_checks_modified(self):
90
54
            checkout = self.make_checkout()