|
|
|
__all__ = ['Mark', 'YAMLError', 'MarkedYAMLError'] |
|
|
|
class Mark: |
|
|
|
def __init__(self, name, index, line, column, buffer, pointer): |
|
self.name = name |
|
self.index = index |
|
self.line = line |
|
self.column = column |
|
self.buffer = buffer |
|
self.pointer = pointer |
|
|
|
def get_snippet(self, indent=4, max_length=75): |
|
if self.buffer is None: |
|
return None |
|
head = '' |
|
start = self.pointer |
|
while start > 0 and self.buffer[start-1] not in '\0\r\n\x85\u2028\u2029': |
|
start -= 1 |
|
if self.pointer-start > max_length/2-1: |
|
head = ' ... ' |
|
start += 5 |
|
break |
|
tail = '' |
|
end = self.pointer |
|
while end < len(self.buffer) and self.buffer[end] not in '\0\r\n\x85\u2028\u2029': |
|
end += 1 |
|
if end-self.pointer > max_length/2-1: |
|
tail = ' ... ' |
|
end -= 5 |
|
break |
|
snippet = self.buffer[start:end] |
|
return ' '*indent + head + snippet + tail + '\n' \ |
|
+ ' '*(indent+self.pointer-start+len(head)) + '^' |
|
|
|
def __str__(self): |
|
snippet = self.get_snippet() |
|
where = " in \"%s\", line %d, column %d" \ |
|
% (self.name, self.line+1, self.column+1) |
|
if snippet is not None: |
|
where += ":\n"+snippet |
|
return where |
|
|
|
class YAMLError(Exception): |
|
pass |
|
|
|
class MarkedYAMLError(YAMLError): |
|
|
|
def __init__(self, context=None, context_mark=None, |
|
problem=None, problem_mark=None, note=None): |
|
self.context = context |
|
self.context_mark = context_mark |
|
self.problem = problem |
|
self.problem_mark = problem_mark |
|
self.note = note |
|
|
|
def __str__(self): |
|
lines = [] |
|
if self.context is not None: |
|
lines.append(self.context) |
|
if self.context_mark is not None \ |
|
and (self.problem is None or self.problem_mark is None |
|
or self.context_mark.name != self.problem_mark.name |
|
or self.context_mark.line != self.problem_mark.line |
|
or self.context_mark.column != self.problem_mark.column): |
|
lines.append(str(self.context_mark)) |
|
if self.problem is not None: |
|
lines.append(self.problem) |
|
if self.problem_mark is not None: |
|
lines.append(str(self.problem_mark)) |
|
if self.note is not None: |
|
lines.append(self.note) |
|
return '\n'.join(lines) |
|
|
|
|