ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: usr / lib64 / python3.6

"""Recognize image file formats based on their first few bytes."""

from os import PathLike

__all__ = ["what"]

#-------------------------#
# Recognize image headers #
#-------------------------#

def what(file, h=None):
    f = None
    try:
        if h is None:
            if isinstance(file, (str, PathLike)):
                f = open(file, 'rb')
                h = f.read(32)
            else:
                location = file.tell()
                h = file.read(32)
                file.seek(location)
        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_jpeg(h, f):
    """JPEG data in JFIF or Exif format"""
    if h[6:10] in (b'JFIF', b'Exif'):
        return 'jpeg'

tests.append(test_jpeg)

def test_png(h, f):
    if h.startswith(b'\211PNG\r\n\032\n'):
        return 'png'

tests.append(test_png)

def test_gif(h, f):
    """GIF ('87 and '89 variants)"""
    if h[:6] in (b'GIF87a', b'GIF89a'):
        return 'gif'

tests.append(test_gif)

def test_tiff(h, f):
    """TIFF (can be in Motorola or Intel byte order)"""
    if h[:2] in (b'MM', b'II'):
        return 'tiff'

tests.append(test_tiff)

def test_rgb(h, f):
    """SGI image library"""
    if h.startswith(b'\001\332'):
        return 'rgb'

tests.append(test_rgb)

def test_pbm(h, f):
    """PBM (portable bitmap)"""
    if len(h) >= 3 and \
        h[0] == ord(b'P') and h[1] in b'14' and h[2] in b' \t\n\r':
        return 'pbm'

tests.append(test_pbm)

def test_pgm(h, f):
    """PGM (portable graymap)"""
    if len(h) >= 3 and \
        h[0] == ord(b'P') and h[1] in b'25' and h[2] in b' \t\n\r':
        return 'pgm'

tests.append(test_pgm)

def test_ppm(h, f):
    """PPM (portable pixmap)"""
    if len(h) >= 3 and \
        h[0] == ord(b'P') and h[1] in b'36' and h[2] in b' \t\n\r':
        return 'ppm'

tests.append(test_ppm)

def test_rast(h, f):
    """Sun raster file"""
    if h.startswith(b'\x59\xA6\x6A\x95'):
        return 'rast'

tests.append(test_rast)

def test_xbm(h, f):
    """X bitmap (X10 or X11)"""
    if h.startswith(b'#define '):
        return 'xbm'

tests.append(test_xbm)

def test_bmp(h, f):
    if h.startswith(b'BM'):
        return 'bmp'

tests.append(test_bmp)

def test_webp(h, f):
    if h.startswith(b'RIFF') and h[8:12] == b'WEBP':
        return 'webp'

tests.append(test_webp)

def test_exr(h, f):
    if h.startswith(b'\x76\x2f\x31\x01'):
        return 'exr'

tests.append(test_exr)

#--------------------#
# 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 + '/:', end=' ')
            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 + ':', end=' ')
            sys.stdout.flush()
            try:
                print(what(filename))
            except OSError:
                print('*** not found ***')

if __name__ == '__main__':
    test()

[+] ..
[-] __phello__.foo.py [edit]
[-] zipfile.py [edit]
[-] _dummy_thread.py [edit]
[-] signal.py [edit]
[-] tabnanny.py [edit]
[-] cmd.py [edit]
[-] cProfile.py [edit]
[-] tempfile.py [edit]
[-] argparse.py [edit]
[-] platform.py [edit]
[-] binhex.py [edit]
[-] antigravity.py [edit]
[+] curses
[-] _compat_pickle.py [edit]
[-] statistics.py [edit]
[-] inspect.py [edit]
[-] ipaddress.py [edit]
[-] sunau.py [edit]
[-] imghdr.py [edit]
[-] random.py [edit]
[-] macpath.py [edit]
[-] hmac.py [edit]
[-] base64.py [edit]
[-] cgitb.py [edit]
[-] stringprep.py [edit]
[-] ast.py [edit]
[-] textwrap.py [edit]
[-] enum.py [edit]
[-] _compression.py [edit]
[-] dis.py [edit]
[+] venv
[-] chunk.py [edit]
[-] this.py [edit]
[-] py_compile.py [edit]
[-] fnmatch.py [edit]
[+] xml
[-] opcode.py [edit]
[+] site-packages
[-] sre_compile.py [edit]
[-] _pyio.py [edit]
[-] pty.py [edit]
[-] datetime.py [edit]
[-] macurl2path.py [edit]
[+] lib2to3
[+] wsgiref
[-] heapq.py [edit]
[-] keyword.py [edit]
[-] codecs.py [edit]
[+] ensurepip
[-] pickletools.py [edit]
[-] compileall.py [edit]
[-] nturl2path.py [edit]
[-] warnings.py [edit]
[-] shutil.py [edit]
[-] socket.py [edit]
[-] pdb.py [edit]
[-] zipapp.py [edit]
[-] formatter.py [edit]
[-] _sitebuiltins.py [edit]
[-] _markupbase.py [edit]
[-] types.py [edit]
[-] runpy.py [edit]
[-] uu.py [edit]
[-] typing.py [edit]
[-] pathlib.py [edit]
[-] _threading_local.py [edit]
[-] mailcap.py [edit]
[-] bisect.py [edit]
[-] linecache.py [edit]
[-] colorsys.py [edit]
[-] glob.py [edit]
[-] sndhdr.py [edit]
[-] ssl.py [edit]
[-] _strptime.py [edit]
[-] _sysconfigdata_m_linux_x86_64-linux-gnu.py [edit]
[-] imp.py [edit]
[-] contextlib.py [edit]
[-] copyreg.py [edit]
[-] filecmp.py [edit]
[+] logging
[-] operator.py [edit]
[-] token.py [edit]
[-] gzip.py [edit]
[-] string.py [edit]
[-] genericpath.py [edit]
[-] _osx_support.py [edit]
[-] stat.py [edit]
[-] lzma.py [edit]
[-] __future__.py [edit]
[+] distutils
[-] dummy_threading.py [edit]
[+] asyncio
[-] webbrowser.py [edit]
[-] symtable.py [edit]
[+] lib-dynload
[-] getopt.py [edit]
[-] wave.py [edit]
[-] codeop.py [edit]
[-] _sysconfigdata_dm_linux_x86_64-linux-gnu.py [edit]
[-] abc.py [edit]
[+] config-3.6m-x86_64-linux-gnu
[+] pydoc_data
[-] telnetlib.py [edit]
[+] unittest
[-] fileinput.py [edit]
[-] ftplib.py [edit]
[+] ctypes
[-] _bootlocale.py [edit]
[-] pipes.py [edit]
[-] aifc.py [edit]
[-] pprint.py [edit]
[-] queue.py [edit]
[-] difflib.py [edit]
[+] __pycache__
[-] uuid.py [edit]
[-] reprlib.py [edit]
[-] threading.py [edit]
[+] http
[-] sre_parse.py [edit]
[-] subprocess.py [edit]
[+] importlib
[-] selectors.py [edit]
[-] getpass.py [edit]
[-] trace.py [edit]
[-] struct.py [edit]
[-] tracemalloc.py [edit]
[-] plistlib.py [edit]
[-] asynchat.py [edit]
[-] fractions.py [edit]
[-] shlex.py [edit]
[-] bdb.py [edit]
[-] locale.py [edit]
[-] calendar.py [edit]
[-] poplib.py [edit]
[-] posixpath.py [edit]
[-] imaplib.py [edit]
[+] concurrent
[-] pickle.py [edit]
[-] timeit.py [edit]
[+] json
[-] traceback.py [edit]
[+] multiprocessing
[-] shelve.py [edit]
[-] csv.py [edit]
[-] tarfile.py [edit]
[-] numbers.py [edit]
[-] optparse.py [edit]
[-] cgi.py [edit]
[-] pstats.py [edit]
[-] crypt.py [edit]
[+] collections
[-] site.py [edit]
[-] pkgutil.py [edit]
[-] re.py [edit]
[-] pyclbr.py [edit]
[-] socketserver.py [edit]
[-] _pydecimal.py [edit]
[-] pydoc.py [edit]
[-] sre_constants.py [edit]
[-] xdrlib.py [edit]
[-] nntplib.py [edit]
[-] mimetypes.py [edit]
[+] email
[-] quopri.py [edit]
[-] doctest.py [edit]
[-] profile.py [edit]
[-] sched.py [edit]
[-] hashlib.py [edit]
[+] urllib
[-] secrets.py [edit]
[-] functools.py [edit]
[-] smtplib.py [edit]
[-] copy.py [edit]
[+] dbm
[-] tty.py [edit]
[-] os.py [edit]
[-] _collections_abc.py [edit]
[-] configparser.py [edit]
[-] _weakrefset.py [edit]
[-] netrc.py [edit]
[-] code.py [edit]
[-] asyncore.py [edit]
[-] tokenize.py [edit]
[+] encodings
[-] io.py [edit]
[-] smtpd.py [edit]
[-] sysconfig.py [edit]
[-] decimal.py [edit]
[+] sqlite3
[-] rlcompleter.py [edit]
[-] ntpath.py [edit]
[+] test
[-] weakref.py [edit]
[-] symbol.py [edit]
[-] modulefinder.py [edit]
[+] xmlrpc
[-] bz2.py [edit]
[-] gettext.py [edit]
[-] mailbox.py [edit]
[+] html