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() self.load()
channel_int = 0
if isStringType(channel): if isStringType(channel):
try: try:
channel = self.getbands().index(channel) channel_int = self.getbands().index(channel)
except ValueError: except ValueError:
raise ValueError( raise ValueError(
'The image has no channel "{}"'.format(channel)) '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): def tell(self):
# type: () -> int # type: () -> int