Mirrors/yamllint
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use bytes for buff.find

Browse Source
This commit is contained in:
Spencer Young
2020-03-24 20:25:04 -07:00
parent 28ea079812
commit 92f2df948c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
yamllint/parser.py
View File

@@ -50,9 +50,9 @@ class Comment(object):
self.comment_before = comment_before self.comment_before = comment_before
def __str__(self): def __str__(self):
end = self.buffer.find('\n', self.pointer) end = self.buffer.find(b'\n', self.pointer)
if end == -1: if end == -1:
end = self.buffer.find('\0', self.pointer) end = self.buffer.find(b'\0', self.pointer)
if end != -1: if end != -1:
return self.buffer[self.pointer:end] return self.buffer[self.pointer:end]
return self.buffer[self.pointer:] return self.buffer[self.pointer:]
@@ -75,14 +75,14 @@ class Comment(object):
def line_generator(buffer): def line_generator(buffer):
line_no = 1 line_no = 1
cur = 0 cur = 0
next = buffer.find('\n') next = buffer.find(b'\n')
while next != -1: while next != -1:
if next > 0 and buffer[next - 1] == '\r': if next > 0 and buffer[next - 1] == '\r':
yield Line(line_no, buffer, start=cur, end=next - 1) yield Line(line_no, buffer, start=cur, end=next - 1)
else: else:
yield Line(line_no, buffer, start=cur, end=next) yield Line(line_no, buffer, start=cur, end=next)
cur = next + 1 cur = next + 1
next = buffer.find('\n', cur) next = buffer.find(b'\n', cur)
line_no += 1 line_no += 1
yield Line(line_no, buffer, start=cur, end=len(buffer)) yield Line(line_no, buffer, start=cur, end=len(buffer))