You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
154 lines
3.3 KiB
154 lines
3.3 KiB
"""Recognize image file formats based on their first few bytes.""" |
|
|
|
__all__ = ["what"] |
|
|
|
#-------------------------# |
|
# Recognize image headers # |
|
#-------------------------# |
|
|
|
def what(file, h=None): |
|
if h is None: |
|
if type(file) == type(''): |
|
f = open(file, 'rb') |
|
h = f.read(32) |
|
else: |
|
location = file.tell() |
|
h = file.read(32) |
|
file.seek(location) |
|
f = None |
|
else: |
|
f = None |
|
try: |
|
for tf in tests: |
|
res = tf(h, f) |
|
if res: |
|
return res |
|
finally: |
|
if f: f.close() |
|
return None |
|
|
|
|
|
#---------------------------------# |
|
# Subroutines per image file type # |
|
#---------------------------------# |
|
|
|
tests = [] |
|
|
|
def test_rgb(h, f): |
|
"""SGI image library""" |
|
if h[:2] == '\001\332': |
|
return 'rgb' |
|
|
|
tests.append(test_rgb) |
|
|
|
def test_gif(h, f): |
|
"""GIF ('87 and '89 variants)""" |
|
if h[:6] in ('GIF87a', 'GIF89a'): |
|
return 'gif' |
|
|
|
tests.append(test_gif) |
|
|
|
def test_pbm(h, f): |
|
"""PBM (portable bitmap)""" |
|
if len(h) >= 3 and \ |
|
h[0] == 'P' and h[1] in '14' and h[2] in ' \t\n\r': |
|
return 'pbm' |
|
|
|
tests.append(test_pbm) |
|
|
|
def test_pgm(h, f): |
|
"""PGM (portable graymap)""" |
|
if len(h) >= 3 and \ |
|
h[0] == 'P' and h[1] in '25' and h[2] in ' \t\n\r': |
|
return 'pgm' |
|
|
|
tests.append(test_pgm) |
|
|
|
def test_ppm(h, f): |
|
"""PPM (portable pixmap)""" |
|
if len(h) >= 3 and \ |
|
h[0] == 'P' and h[1] in '36' and h[2] in ' \t\n\r': |
|
return 'ppm' |
|
|
|
tests.append(test_ppm) |
|
|
|
def test_tiff(h, f): |
|
"""TIFF (can be in Motorola or Intel byte order)""" |
|
if h[:2] in ('MM', 'II'): |
|
return 'tiff' |
|
|
|
tests.append(test_tiff) |
|
|
|
def test_rast(h, f): |
|
"""Sun raster file""" |
|
if h[:4] == '\x59\xA6\x6A\x95': |
|
return 'rast' |
|
|
|
tests.append(test_rast) |
|
|
|
def test_xbm(h, f): |
|
"""X bitmap (X10 or X11)""" |
|
s = '#define ' |
|
if h[:len(s)] == s: |
|
return 'xbm' |
|
|
|
tests.append(test_xbm) |
|
|
|
def test_jpeg(h, f): |
|
"""JPEG data in JFIF format""" |
|
if h[6:10] == 'JFIF': |
|
return 'jpeg' |
|
|
|
tests.append(test_jpeg) |
|
|
|
def test_bmp(h, f): |
|
if h[:2] == 'BM': |
|
return 'bmp' |
|
|
|
tests.append(test_bmp) |
|
|
|
def test_png(h, f): |
|
if h[:8] == "\211PNG\r\n\032\n": |
|
return 'png' |
|
|
|
tests.append(test_png) |
|
|
|
#--------------------# |
|
# Small test program # |
|
#--------------------# |
|
|
|
def test(): |
|
import sys |
|
recursive = 0 |
|
if sys.argv[1:] and sys.argv[1] == '-r': |
|
del sys.argv[1:2] |
|
recursive = 1 |
|
try: |
|
if sys.argv[1:]: |
|
testall(sys.argv[1:], recursive, 1) |
|
else: |
|
testall(['.'], recursive, 1) |
|
except KeyboardInterrupt: |
|
sys.stderr.write('\n[Interrupted]\n') |
|
sys.exit(1) |
|
|
|
def testall(list, recursive, toplevel): |
|
import sys |
|
import os |
|
for filename in list: |
|
if os.path.isdir(filename): |
|
print filename + '/:', |
|
if recursive or toplevel: |
|
print 'recursing down:' |
|
import glob |
|
names = glob.glob(os.path.join(filename, '*')) |
|
testall(names, recursive, 0) |
|
else: |
|
print '*** directory (use -r) ***' |
|
else: |
|
print filename + ':', |
|
sys.stdout.flush() |
|
try: |
|
print what(filename) |
|
except IOError: |
|
print '*** not found ***'
|
|
|