openmedialibrary_platform_w.../Lib/site-packages/PIL/features.py

87 lines
1.9 KiB
Python
Raw Permalink Normal View History

2019-01-20 10:35:31 +00:00
from . import Image
2016-01-31 14:44:46 +00:00
modules = {
"pil": "PIL._imaging",
2019-01-20 10:35:31 +00:00
"tkinter": "PIL._tkinter_finder",
2016-01-31 14:44:46 +00:00
"freetype2": "PIL._imagingft",
"littlecms2": "PIL._imagingcms",
"webp": "PIL._webp",
}
def check_module(feature):
2019-01-20 10:35:31 +00:00
if not (feature in modules):
2016-01-31 14:44:46 +00:00
raise ValueError("Unknown module %s" % feature)
module = modules[feature]
try:
2019-01-20 10:35:31 +00:00
__import__(module)
2016-01-31 14:44:46 +00:00
return True
2019-01-20 10:35:31 +00:00
except ImportError:
return False
2016-01-31 14:44:46 +00:00
def get_supported_modules():
2019-01-20 10:35:31 +00:00
return [f for f in modules if check_module(f)]
2016-01-31 14:44:46 +00:00
codecs = {
"jpg": "jpeg",
"jpg_2000": "jpeg2k",
"zlib": "zip",
"libtiff": "libtiff"
}
def check_codec(feature):
if feature not in codecs:
raise ValueError("Unknown codec %s" % feature)
codec = codecs[feature]
return codec + "_encoder" in dir(Image.core)
def get_supported_codecs():
2019-01-20 10:35:31 +00:00
return [f for f in codecs if check_codec(f)]
features = {
"webp_anim": ("PIL._webp", 'HAVE_WEBPANIM'),
"webp_mux": ("PIL._webp", 'HAVE_WEBPMUX'),
"transp_webp": ("PIL._webp", "HAVE_TRANSPARENCY"),
"raqm": ("PIL._imagingft", "HAVE_RAQM"),
"libjpeg_turbo": ("PIL._imaging", "HAVE_LIBJPEGTURBO"),
}
def check_feature(feature):
if feature not in features:
raise ValueError("Unknown feature %s" % feature)
module, flag = features[feature]
try:
imported_module = __import__(module, fromlist=['PIL'])
return getattr(imported_module, flag)
except ImportError:
return None
def get_supported_features():
return [f for f in features if check_feature(f)]
def check(feature):
return (feature in modules and check_module(feature) or
feature in codecs and check_codec(feature) or
feature in features and check_feature(feature))
def get_supported():
ret = get_supported_modules()
ret.extend(get_supported_features())
ret.extend(get_supported_codecs())
return ret