Mercurial > cpython-lru_cache-weakref
annotate Lib/imghdr.py @ 4566:6403c94de50d legacy-trunk
Added safeguard against failure in __del__.
| author | Guido van Rossum <guido@python.org> |
|---|---|
| date | Wed, 27 Nov 1996 19:39:58 +0000 |
| parents | 6d9d5e7e588c |
| children | 4459172f40bb |
| rev | line source |
|---|---|
| 876 | 1 # Recognizing image files based on their first few bytes. |
| 2 | |
| 3 | |
| 4 #-------------------------# | |
| 5 # Recognize sound headers # | |
| 6 #-------------------------# | |
| 7 | |
|
3821
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
8 def what(filename, h=None): |
|
4108
6d9d5e7e588c
imghdr.what(): check for not h'. This lets
Guido van Rossum <guido@python.org>
parents:
3821
diff
changeset
|
9 if h is None: |
|
3821
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
10 f = open(filename, 'r') |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
11 h = f.read(32) |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
12 else: |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
13 f = None |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
14 try: |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
15 for tf in tests: |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
16 res = tf(h, f) |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
17 if res: |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
18 return res |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
19 finally: |
|
240f834aee8b
Added optional second arg to what(), giving the data read from the file
Guido van Rossum <guido@python.org>
parents:
1316
diff
changeset
|
20 if f: f.close() |
| 876 | 21 return None |
| 22 | |
| 23 | |
| 24 #---------------------------------# | |
| 25 # Subroutines per image file type # | |
| 26 #---------------------------------# | |
| 27 | |
| 28 tests = [] | |
| 29 | |
| 30 def test_rgb(h, f): | |
| 31 # SGI image library | |
| 32 if h[:2] == '\001\332': | |
| 33 return 'rgb' | |
| 34 | |
| 35 tests.append(test_rgb) | |
| 36 | |
| 37 def test_gif(h, f): | |
| 38 # GIF ('87 and '89 variants) | |
| 39 if h[:6] in ('GIF87a', 'GIF89a'): | |
| 40 return 'gif' | |
| 41 | |
| 42 tests.append(test_gif) | |
| 43 | |
|
1078
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
44 def test_pbm(h, f): |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
45 # PBM (portable bitmap) |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
46 if len(h) >= 3 and \ |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
47 h[0] == 'P' and h[1] in '14' and h[2] in ' \t\n\r': |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
48 return 'pbm' |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
49 |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
50 tests.append(test_pbm) |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
51 |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
52 def test_pgm(h, f): |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
53 # PGM (portable graymap) |
|
880
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
54 if len(h) >= 3 and \ |
|
1078
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
55 h[0] == 'P' and h[1] in '25' and h[2] in ' \t\n\r': |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
56 return 'pgm' |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
57 |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
58 tests.append(test_pgm) |
| 876 | 59 |
|
1078
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
60 def test_ppm(h, f): |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
61 # PPM (portable pixmap) |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
62 if len(h) >= 3 and \ |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
63 h[0] == 'P' and h[1] in '36' and h[2] in ' \t\n\r': |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
64 return 'ppm' |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
65 |
|
245003c006a6
Misc changes and new modules. whrandom is "objectified". SOCKET.py
Guido van Rossum <guido@python.org>
parents:
880
diff
changeset
|
66 tests.append(test_ppm) |
| 876 | 67 |
| 68 def test_tiff(h, f): | |
| 69 # TIFF (can be in Motorola or Intel byte order) | |
| 70 if h[:2] in ('MM', 'II'): | |
| 71 return 'tiff' | |
| 72 | |
| 73 tests.append(test_tiff) | |
| 74 | |
| 75 def test_rast(h, f): | |
| 76 # Sun raster file | |
| 77 if h[:4] == '\x59\xA6\x6A\x95': | |
| 78 return 'rast' | |
| 79 | |
| 80 tests.append(test_rast) | |
| 81 | |
|
880
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
82 def test_xbm(h, f): |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
83 # X bitmap (X10 or X11) |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
84 s = '#define ' |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
85 if h[:len(s)] == s: |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
86 return 'xbm' |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
87 |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
88 tests.append(test_xbm) |
|
36b934326fed
Fix pnm test for short files and add X bitmap as recognized type
Guido van Rossum <guido@python.org>
parents:
876
diff
changeset
|
89 |
|
1316
2f3f293ef5f1
* calendar.py: all libC functionality now moved to built-in time module
Guido van Rossum <guido@python.org>
parents:
1078
diff
changeset
|
90 def test_jpeg(h, f): |
|
2f3f293ef5f1
* calendar.py: all libC functionality now moved to built-in time module
Guido van Rossum <guido@python.org>
parents:
1078
diff
changeset
|
91 # JPEG data in JFIF format |
|
2f3f293ef5f1
* calendar.py: all libC functionality now moved to built-in time module
Guido van Rossum <guido@python.org>
parents:
1078
diff
changeset
|
92 if h[6:10] == 'JFIF': |
|
2f3f293ef5f1
* calendar.py: all libC functionality now moved to built-in time module
Guido van Rossum <guido@python.org>
parents:
1078
diff
changeset
|
93 return 'jpeg' |
|
2f3f293ef5f1
* calendar.py: all libC functionality now moved to built-in time module
Guido van Rossum <guido@python.org>
parents:
1078
diff
changeset
|
94 |
|
2f3f293ef5f1
* calendar.py: all libC functionality now moved to built-in time module
Guido van Rossum <guido@python.org>
parents:
1078
diff
changeset
|
95 tests.append(test_jpeg) |
| 876 | 96 |
| 97 #--------------------# | |
| 98 # Small test program # | |
| 99 #--------------------# | |
| 100 | |
| 101 def test(): | |
| 102 import sys | |
| 103 recursive = 0 | |
| 104 if sys.argv[1:] and sys.argv[1] == '-r': | |
| 105 del sys.argv[1:2] | |
| 106 recursive = 1 | |
| 107 try: | |
| 108 if sys.argv[1:]: | |
| 109 testall(sys.argv[1:], recursive, 1) | |
| 110 else: | |
| 111 testall(['.'], recursive, 1) | |
| 112 except KeyboardInterrupt: | |
| 113 sys.stderr.write('\n[Interrupted]\n') | |
| 114 sys.exit(1) | |
| 115 | |
| 116 def testall(list, recursive, toplevel): | |
| 117 import sys | |
| 118 import os | |
| 119 for filename in list: | |
| 120 if os.path.isdir(filename): | |
| 121 print filename + '/:', | |
| 122 if recursive or toplevel: | |
| 123 print 'recursing down:' | |
| 124 import glob | |
| 125 names = glob.glob(os.path.join(filename, '*')) | |
| 126 testall(names, recursive, 0) | |
| 127 else: | |
| 128 print '*** directory (use -r) ***' | |
| 129 else: | |
| 130 print filename + ':', | |
| 131 sys.stdout.flush() | |
| 132 try: | |
| 133 print what(filename) | |
| 134 except IOError: | |
| 135 print '*** not found ***' |