1
# Copyright (C) 2005 by Aaron Bentley
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
# TODO: Move this into builtins
19
# TODO: 'bzr resolve' should accept a directory name and work from that
26
from bzrlib.commands import register_command
27
from bzrlib.errors import BzrCommandError, NotConflicted, UnsupportedOperation
28
from bzrlib.option import Option
29
from bzrlib.osutils import rename, delete_any
30
from bzrlib.rio import Stanza
33
CONFLICT_SUFFIXES = ('.THIS', '.BASE', '.OTHER')
36
class cmd_conflicts(bzrlib.commands.Command):
37
"""List files with conflicts.
39
Merge will do its best to combine the changes in two branches, but there
40
are some kinds of problems only a human can fix. When it encounters those,
41
it will mark a conflict. A conflict means that you need to fix something,
42
before you should commit.
44
Use bzr resolve when you have fixed a problem.
46
(conflicts are determined by the presence of .BASE .TREE, and .OTHER
52
from bzrlib.workingtree import WorkingTree
53
wt = WorkingTree.open_containing(u'.')[0]
54
for conflict in wt.conflicts():
58
class cmd_resolve(bzrlib.commands.Command):
59
"""Mark a conflict as resolved.
61
Merge will do its best to combine the changes in two branches, but there
62
are some kinds of problems only a human can fix. When it encounters those,
63
it will mark a conflict. A conflict means that you need to fix something,
64
before you should commit.
66
Once you have fixed a problem, use "bzr resolve FILE.." to mark
67
individual files as fixed, or "bzr resolve --all" to mark all conflicts as
70
See also bzr conflicts.
72
aliases = ['resolved']
73
takes_args = ['file*']
74
takes_options = [Option('all', help='Resolve all conflicts in this tree')]
75
def run(self, file_list=None, all=False):
76
from bzrlib.workingtree import WorkingTree
79
raise BzrCommandError("If --all is specified, no FILE may be provided")
80
tree = WorkingTree.open_containing('.')[0]
84
raise BzrCommandError("command 'resolve' needs one or more FILE, or --all")
85
tree = WorkingTree.open_containing(file_list[0])[0]
86
to_resolve = [tree.relpath(p) for p in file_list]
87
resolve(tree, to_resolve)
90
def resolve(tree, paths=None, ignore_misses=False):
93
tree_conflicts = tree.conflicts()
95
new_conflicts = ConflictList()
96
selected_conflicts = tree_conflicts
98
new_conflicts, selected_conflicts = \
99
tree_conflicts.select_conflicts(tree, paths, ignore_misses)
101
tree.set_conflicts(new_conflicts)
102
except UnsupportedOperation:
104
selected_conflicts.remove_files(tree)
109
def restore(filename):
111
Restore a conflicted file to the state it was in before merging.
112
Only text restoration supported at present.
116
rename(filename + ".THIS", filename)
119
if e.errno != errno.ENOENT:
122
os.unlink(filename + ".BASE")
125
if e.errno != errno.ENOENT:
128
os.unlink(filename + ".OTHER")
131
if e.errno != errno.ENOENT:
134
raise NotConflicted(filename)
137
class ConflictList(object):
138
"""List of conflicts.
140
Typically obtained from WorkingTree.conflicts()
142
Can be instantiated from stanzas or from Conflict subclasses.
145
def __init__(self, conflicts=None):
146
object.__init__(self)
147
if conflicts is None:
150
self.__list = conflicts
153
return len(self.__list) == 0
156
return len(self.__list)
159
return iter(self.__list)
161
def __getitem__(self, key):
162
return self.__list[key]
164
def append(self, conflict):
165
return self.__list.append(conflict)
167
def __eq__(self, other_list):
168
return list(self) == list(other_list)
170
def __ne__(self, other_list):
171
return not (self == other_list)
174
return "ConflictList(%r)" % self.__list
177
def from_stanzas(stanzas):
178
"""Produce a new ConflictList from an iterable of stanzas"""
179
conflicts = ConflictList()
180
for stanza in stanzas:
181
conflicts.append(Conflict.factory(**stanza.as_dict()))
184
def to_stanzas(self):
185
"""Generator of stanzas"""
186
for conflict in self:
187
yield conflict.as_stanza()
189
def to_strings(self):
190
"""Generate strings for the provided conflicts"""
191
for conflict in self:
194
def remove_files(self, tree):
195
"""Remove the THIS, BASE and OTHER files for listed conflicts"""
196
for conflict in self:
197
if not conflict.has_files:
199
for suffix in CONFLICT_SUFFIXES:
201
delete_any(tree.abspath(conflict.path+suffix))
203
if e.errno != errno.ENOENT:
206
def select_conflicts(self, tree, paths, ignore_misses=False):
207
"""Select the conflicts associated with paths in a tree.
209
File-ids are also used for this.
211
path_set = set(paths)
213
selected_paths = set()
214
new_conflicts = ConflictList()
215
selected_conflicts = ConflictList()
217
file_id = tree.path2id(path)
218
if file_id is not None:
221
for conflict in self:
223
for key in ('path', 'conflict_path'):
224
cpath = getattr(conflict, key, None)
227
if cpath in path_set:
229
selected_paths.add(cpath)
230
for key in ('file_id', 'conflict_file_id'):
231
cfile_id = getattr(conflict, key, None)
235
cpath = ids[cfile_id]
239
selected_paths.add(cpath)
241
selected_conflicts.append(conflict)
243
new_conflicts.append(conflict)
244
if ignore_misses is not True:
245
for path in [p for p in paths if p not in selected_paths]:
246
if not os.path.exists(tree.abspath(path)):
247
print "%s does not exist" % path
249
print "%s is not conflicted" % path
250
return new_conflicts, selected_conflicts
253
class Conflict(object):
254
"""Base class for all types of conflict"""
258
def __init__(self, path, file_id=None):
260
self.file_id = file_id
263
s = Stanza(type=self.typestring, path=self.path)
264
if self.file_id is not None:
265
s.add('file_id', self.file_id)
269
return [type(self), self.path, self.file_id]
271
def __cmp__(self, other):
272
if getattr(other, "_cmp_list", None) is None:
274
return cmp(self._cmp_list(), other._cmp_list())
276
def __eq__(self, other):
277
return self.__cmp__(other) == 0
279
def __ne__(self, other):
280
return not self.__eq__(other)
283
return self.format % self.__dict__
286
rdict = dict(self.__dict__)
287
rdict['class'] = self.__class__.__name__
288
return self.rformat % rdict
291
def factory(type, **kwargs):
293
return ctype[type](**kwargs)
296
class PathConflict(Conflict):
297
"""A conflict was encountered merging file paths"""
299
typestring = 'path conflict'
301
format = 'Path conflict: %(path)s / %(conflict_path)s'
303
rformat = '%(class)s(%(path)r, %(conflict_path)r, %(file_id)r)'
304
def __init__(self, path, conflict_path=None, file_id=None):
305
Conflict.__init__(self, path, file_id)
306
self.conflict_path = conflict_path
309
s = Conflict.as_stanza(self)
310
if self.conflict_path is not None:
311
s.add('conflict_path', self.conflict_path)
315
class ContentsConflict(PathConflict):
316
"""The files are of different types, or not present"""
320
typestring = 'contents conflict'
322
format = 'Contents conflict in %(path)s'
325
class TextConflict(PathConflict):
326
"""The merge algorithm could not resolve all differences encountered."""
330
typestring = 'text conflict'
332
format = 'Text conflict in %(path)s'
335
class HandledConflict(Conflict):
336
"""A path problem that has been provisionally resolved.
337
This is intended to be a base class.
340
rformat = "%(class)s(%(action)r, %(path)r, %(file_id)r)"
342
def __init__(self, action, path, file_id=None):
343
Conflict.__init__(self, path, file_id)
347
return Conflict._cmp_list(self) + [self.action]
350
s = Conflict.as_stanza(self)
351
s.add('action', self.action)
355
class HandledPathConflict(HandledConflict):
356
"""A provisionally-resolved path problem involving two paths.
357
This is intended to be a base class.
360
rformat = "%(class)s(%(action)r, %(path)r, %(conflict_path)r,"\
361
" %(file_id)r, %(conflict_file_id)r)"
363
def __init__(self, action, path, conflict_path, file_id=None,
364
conflict_file_id=None):
365
HandledConflict.__init__(self, action, path, file_id)
366
self.conflict_path = conflict_path
367
self.conflict_file_id = conflict_file_id
370
return HandledConflict._cmp_list(self) + [self.conflict_path,
371
self.conflict_file_id]
374
s = HandledConflict.as_stanza(self)
375
s.add('conflict_path', self.conflict_path)
376
if self.conflict_file_id is not None:
377
s.add('conflict_file_id', self.conflict_file_id)
382
class DuplicateID(HandledPathConflict):
383
"""Two files want the same file_id."""
385
typestring = 'duplicate id'
387
format = 'Conflict adding id to %(conflict_path)s. %(action)s %(path)s.'
390
class DuplicateEntry(HandledPathConflict):
391
"""Two directory entries want to have the same name."""
393
typestring = 'duplicate'
395
format = 'Conflict adding file %(conflict_path)s. %(action)s %(path)s.'
398
class ParentLoop(HandledPathConflict):
399
"""An attempt to create an infinitely-looping directory structure.
400
This is rare, but can be produced like so:
409
typestring = 'parent loop'
411
format = 'Conflict moving %(conflict_path)s into %(path)s. %(action)s.'
414
class UnversionedParent(HandledConflict):
415
"""An attempt to version an file whose parent directory is not versioned.
416
Typically, the result of a merge where one tree unversioned the directory
417
and the other added a versioned file to it.
420
typestring = 'unversioned parent'
422
format = 'Conflict adding versioned files to %(path)s. %(action)s.'
425
class MissingParent(HandledConflict):
426
"""An attempt to add files to a directory that is not present.
427
Typically, the result of a merge where one tree deleted the directory and
428
the other added a file to it.
431
typestring = 'missing parent'
433
format = 'Conflict adding files to %(path)s. %(action)s.'
440
def register_types(*conflict_types):
441
"""Register a Conflict subclass for serialization purposes"""
443
for conflict_type in conflict_types:
444
ctype[conflict_type.typestring] = conflict_type
447
register_types(ContentsConflict, TextConflict, PathConflict, DuplicateID,
448
DuplicateEntry, ParentLoop, UnversionedParent, MissingParent,)
added
removed
bzrlib/conflicts.py
