/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to backend/update.py

  • Committer: Szilveszter Farkas (Phanatic)
  • Date: 2006-06-21 20:14:00 UTC
  • mto: (0.14.1 main) (93.1.1 win32.bialix)
  • mto: This revision was merged to the branch mainline in revision 83.
  • Revision ID: Szilveszter.Farkas@gmail.com-20060621201400-ef7ddfe95b6687a0
2006-06-20  Szilveszter Farkas <Szilveszter.Farkas@gmail.com>

    * backend/init.py: moved pull() to backend/update.py
    * backend/update.py: implemented update() and missing()
        
2006-06-19  Szilveszter Farkas <Szilveszter.Farkas@gmail.com>
        
    * backend/errors.py: renamed NoPushLocationKnown to NoLocationKnown
    * backend/init.py: implemented pull()

Show diffs side-by-side

added added

removed removed

Lines of Context:
backend/update.py
 
1
# Copyright (C) 2006 by Szilveszter Farkas (Phanatic)
 
2
# Some parts of the code are:
 
3
# Copyright (C) 2005, 2006 by Canonical Ltd
 
4
 
 
5
# This program is free software; you can redistribute it and/or modify
 
6
# it under the terms of the GNU General Public License as published by
 
7
# the Free Software Foundation; either version 2 of the License, or
 
8
# (at your option) any later version.
 
9
 
 
10
# This program is distributed in the hope that it will be useful,
 
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
# GNU General Public License for more details.
 
14
 
 
15
# You should have received a copy of the GNU General Public License
 
16
# along with this program; if not, write to the Free Software
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
18
 
 
19
import bzrlib.errors as errors
 
20
 
 
21
from bzrlib.workingtree import WorkingTree
 
22
 
 
23
from errors import NoLocationKnown
 
24
 
 
25
def missing(branch, other_branch=None, reverse=False):
 
26
    """ Get the number of missing or extra revisions of local branch.
 
27
    
 
28
    :param branch: path to local branch
 
29
    
 
30
    :param other_branch: location of the other branch
 
31
    
 
32
    :param reverse: reverse the order of revisions
 
33
    
 
34
    :return: number of missing revisions (if < 0, then extra revisions * -1)
 
35
    """
 
36
    import bzrlib
 
37
    from bzrlib.missing import find_unmerged
 
38
    
 
39
    local_branch = bzrlib.branch.Branch.open_containing(branch)[0]
 
40
    parent = local_branch.get_parent()
 
41
    if other_branch is None:
 
42
        other_branch = parent
 
43
        if other_branch is None:
 
44
            raise NoLocationKnown
 
45
    remote_branch = bzrlib.branch.Branch.open(other_branch)
 
46
    if remote_branch.base == local_branch.base:
 
47
        remote_branch = local_branch
 
48
    local_branch.lock_read()
 
49
 
 
50
    ret = 0
 
51
 
 
52
    try:
 
53
        remote_branch.lock_read()
 
54
        try:
 
55
            local_extra, remote_extra = find_unmerged(local_branch, remote_branch)
 
56
 
 
57
            if reverse is False:
 
58
                local_extra.reverse()
 
59
                remote_extra.reverse()
 
60
 
 
61
            if local_extra:
 
62
                ret = len(local_extra) * -1
 
63
 
 
64
            if remote_extra:
 
65
                ret = len(remote_extra)
 
66
 
 
67
            if not remote_extra and not local_extra:
 
68
                ret = 0
 
69
 
 
70
        finally:
 
71
            remote_branch.unlock()
 
72
    finally:
 
73
        local_branch.unlock()
 
74
 
 
75
    if not ret and parent is None and other_branch is not None:
 
76
        local_branch.lock_write()
 
77
        try:
 
78
            # handle race conditions - a parent might be set while we run.
 
79
            if local_branch.get_parent() is None:
 
80
                local_branch.set_parent(remote_branch.base)
 
81
        finally:
 
82
            local_branch.unlock()
 
83
    
 
84
    return ret
 
85
 
 
86
 
 
87
def pull(branch, location=None, remember=False, overwrite=False, revision=None):
 
88
    """ Pull revisions from another branch.
 
89
    
 
90
    :param branch: the local branch where you want to pull
 
91
    
 
92
    :param location: location of branch you'd like to pull from (can be a bundle, too)
 
93
    
 
94
    :param remeber: if True, the location will be stored
 
95
    
 
96
    :param overwrite: if True, forget local changes, and update the branch to match the remote one
 
97
    
 
98
    :param revision: if specified, only the given revision will be pulled (revno)
 
99
    
 
100
    :return: number of revisions pulled
 
101
    """
 
102
    from bzrlib.branch import Branch
 
103
    from bzrlib.bundle import read_bundle_from_url
 
104
    from bzrlib.bundle.apply_bundle import install_bundle
 
105
    
 
106
    try:
 
107
        tree_to = WorkingTree.open_containing(branch)[0]
 
108
        branch_to = tree_to.branch
 
109
    except errors.NoWorkingTree:
 
110
        tree_to = None
 
111
        branch_to = Branch.open_containing(branch)[0]
 
112
 
 
113
    reader = None
 
114
    if location is not None:
 
115
        try:
 
116
            reader = read_bundle_from_url(location)
 
117
        except errors.NotABundle:
 
118
            pass # Continue on considering this url a Branch
 
119
 
 
120
    stored_loc = branch_to.get_parent()
 
121
    if location is None:
 
122
        if stored_loc is None:
 
123
            raise NoLocationKnown
 
124
        else:
 
125
            location = stored_loc
 
126
 
 
127
    if reader is not None:
 
128
        install_bundle(branch_to.repository, reader)
 
129
        branch_from = branch_to
 
130
    else:
 
131
        branch_from = Branch.open(location)
 
132
 
 
133
        if branch_to.get_parent() is None or remember:
 
134
            branch_to.set_parent(branch_from.base)
 
135
 
 
136
    rev_id = None
 
137
    
 
138
    if revision is not None:
 
139
        rev_id = branch_from.get_rev_id(revision)
 
140
    else:
 
141
        if reader is not None:
 
142
            rev_id = reader.info.target
 
143
        
 
144
    old_rh = branch_to.revision_history()
 
145
    if tree_to is not None:
 
146
        count = tree_to.pull(branch_from, overwrite, rev_id)
 
147
    else:
 
148
        count = branch_to.pull(branch_from, overwrite, rev_id)
 
149
    
 
150
    return count
 
151
 
 
152
def update(location):
 
153
    """ Update a tree to have the latest code committed to its branch.
 
154
    
 
155
    :param location: the path to the branch/working tree
 
156
    
 
157
    :return: None if tree is up to date, 1 if there are conflicts, 0 if updated without having conflicts
 
158
    """
 
159
    tree = WorkingTree.open_containing(location)[0]
 
160
    tree.lock_write()
 
161
    try:
 
162
        if tree.last_revision() == tree.branch.last_revision():
 
163
            # may be up to date, check master too.
 
164
            master = tree.branch.get_master_branch()
 
165
            if master is None or master.last_revision == tree.last_revision():
 
166
                # tree is up to date.
 
167
                return None
 
168
        conflicts = tree.update()
 
169
        if conflicts != 0:
 
170
            return 1
 
171
        else:
 
172
            return 0
 
173
    finally:
 
174
        tree.unlock()
 
175