Source code for visions.utils.monkeypatches.imghdr_patch

# Monkeypatch bug in imagehdr
from imghdr import tests


[docs]def test_jpeg1(h, f): """JPEG data in JFIF format""" if b"JFIF" in h[:23]: return "jpeg"
JPEG_MARK = ( b"\xff\xd8\xff\xdb\x00C\x00\x08\x06\x06" b"\x07\x06\x05\x08\x07\x07\x07\t\t\x08\n\x0c\x14\r\x0c\x0b\x0b\x0c\x19\x12\x13\x0f" )
[docs]def test_jpeg2(h, f): """JPEG with small header""" if len(h) >= 32 and 67 == h[5] and h[:32] == JPEG_MARK: return "jpeg"
[docs]def test_jpeg3(h, f): """JPEG data in JFIF or Exif format""" if h[6:10] in (b"JFIF", b"Exif") or h[:2] == b"\xff\xd8": return "jpeg"
tests.append(test_jpeg1) tests.append(test_jpeg2) tests.append(test_jpeg3)