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
|
from itertools import chain
from bzrlib.errors import BinaryFile
from bzrlib.iterablefile import IterableFile
from bzrlib.osutils import file_iterator
def text_file(input):
"""Produce a file iterator that is guaranteed to be text, without seeking.
BinaryFile is raised if the file contains a NUL in the first 1024 bytes.
"""
first_chunk = input.read(1024)
if '\x00' in first_chunk:
raise BinaryFile()
return IterableFile(chain((first_chunk,), file_iterator(input)))
def check_text_lines(lines):
"""Raise BinaryFile if the supplied lines contain NULs.
Only the first 1024 characters are checked.
"""
f = IterableFile(lines)
if '\x00' in f.read(1024):
raise BinaryFile()
def check_text_path(path):
"""Check whether the supplied path is a text, not binary file.
Raise BinaryFile if a NUL occurs in the first 1024 bytes.
"""
text_file(open(path, 'rb'))
|