1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
# Copyright (C) 2008 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""Import command classes."""
# Lists of command names
COMMAND_NAMES = ['blob', 'checkpoint', 'commit', 'progress', 'reset', 'tag']
FILE_COMMAND_NAMES = ['filemodify', 'filedelete', 'filecopy', 'filerename',
'filedeleteall']
# Bazaar file kinds
FILE_KIND = 'file'
SYMLINK_KIND = 'symlink'
class ImportCommand(object):
"""Base class for import commands."""
def __init__(self, name):
self.name = name
# List of field names not to display
self._binary = []
def dump_str(self, names=None, child_lists=None, verbose=False):
"""Dump fields as a string.
:param names: the list of fields to include or
None for all public fields
:param child_lists: dictionary of child command names to
fields for that child command to include
:param verbose: if True, prefix each line with the command class and
display fields as a dictionary; if False, dump just the field
values with tabs between them
"""
interesting = {}
if names is None:
fields = [k for k in self.__dict__.keys() if not k.startswith('_')]
else:
fields = names
for field in fields:
value = self.__dict__.get(field)
if field in self._binary and value is not None:
value = '(...)'
interesting[field] = value
if verbose:
return "%s: %s" % (self.__class__.__name__, interesting)
else:
return "\t".join([str(interesting[k]) for k in fields])
class BlobCommand(ImportCommand):
def __init__(self, mark, data, lineno=0):
ImportCommand.__init__(self, 'blob')
self.mark = mark
self.data = data
self.lineno = lineno
# Provide a unique id in case the mark is missing
if mark is None:
self.id = '@%d' % lineno
else:
self.id = ':' + mark
self._binary = ['data']
class CheckpointCommand(ImportCommand):
def __init__(self):
ImportCommand.__init__(self, 'checkpoint')
class CommitCommand(ImportCommand):
def __init__(self, ref, mark, author, committer, message, from_,
merges, file_iter, lineno=0):
ImportCommand.__init__(self, 'commit')
self.ref = ref
self.mark = mark
self.author = author
self.committer = committer
self.message = message
self.from_ = from_
self.merges = merges
self.file_iter = file_iter
self.lineno = lineno
self._binary = ['file_iter']
# Provide a unique id in case the mark is missing
if mark is None:
self.id = '@%d' % lineno
else:
self.id = ':' + mark
def dump_str(self, names=None, child_lists=None, verbose=False):
result = [ImportCommand.dump_str(self, names, verbose=verbose)]
for f in self.file_iter():
if child_lists is None:
continue
try:
child_names = child_lists[f.name]
except KeyError:
continue
result.append("\t%s" % f.dump_str(child_names, verbose=verbose))
return '\n'.join(result)
class ProgressCommand(ImportCommand):
def __init__(self, message):
ImportCommand.__init__(self, 'progress')
self.message = message
class ResetCommand(ImportCommand):
def __init__(self, ref, from_):
ImportCommand.__init__(self, 'reset')
self.ref = ref
self.from_ = from_
class TagCommand(ImportCommand):
def __init__(self, id, from_, tagger, message):
ImportCommand.__init__(self, 'tag')
self.id = id
self.from_ = from_
self.tagger = tagger
self.message = message
class FileCommand(ImportCommand):
"""Base class for file commands."""
pass
class FileModifyCommand(FileCommand):
def __init__(self, path, kind, is_executable, dataref, data):
# Either dataref or data should be null
FileCommand.__init__(self, 'filemodify')
self.path = path
self.kind = kind
self.is_executable = is_executable
self.dataref = dataref
self.data = data
self._binary = ['data']
class FileDeleteCommand(FileCommand):
def __init__(self, path):
FileCommand.__init__(self, 'filedelete')
self.path = path
class FileCopyCommand(FileCommand):
def __init__(self, src_path, dest_path):
FileCommand.__init__(self, 'filecopy')
self.src_path = src_path
self.dest_path = dest_path
class FileRenameCommand(FileCommand):
def __init__(self, old_path, new_path):
FileCommand.__init__(self, 'filerename')
self.old_path = old_path
self.new_path = new_path
class FileDeleteAllCommand(FileCommand):
def __init__(self):
FileCommand.__init__(self, 'filedeleteall')
|