Typing: Image.getchannel

This commit is contained in:
Eric Soroos 2018-01-05 12:36:52 +00:00
parent 4b47177634
commit e220e00386

View File

@ -2128,14 +2128,18 @@ class Image(object):
"""
self.load()
channel_int = 0
if isStringType(channel):
try:
channel = self.getbands().index(channel)
channel_int = self.getbands().index(channel)
except ValueError:
raise ValueError(
'The image has no channel "{}"'.format(channel))
else:
channel_int = channel # type: ignore
return self._new(self.im.getband(channel))
return self._new(self.im.getband(channel_int))
def tell(self):
# type: () -> int