wacom: de-duplicate an opcode check

we don't need the same-ish check twice, we can just pop our non-list into a
list an go from there.

Signed-off-by: Peter Hutterer <peter.hutterer@who-t.net>
pull/149/head
Peter Hutterer 2019-07-31 15:54:45 +10:00 committed by Benjamin Tissoires
parent c84f3f8c63
commit 6e27d70517
1 changed files with 5 additions and 6 deletions

View File

@ -501,12 +501,11 @@ class WacomProtocolLowLevelComm(GObject.Object):
# logger.debug(f'received {data.opcode:02x} / {data.length:02x} / {b2hex(bytes(data))}')
if isinstance(expected_opcode, list):
if data.opcode not in expected_opcode:
raise WacomException(f'unexpected opcode: {data.opcode:02x}')
else:
if data.opcode != expected_opcode:
raise WacomException(f'unexpected opcode: {data.opcode:02x}')
if not isinstance(expected_opcode, list):
expected_opcode = [expected_opcode]
if data.opcode not in expected_opcode:
raise WacomException(f'unexpected opcode: {data.opcode:02x}')
return data