Check that the type in Colour and Permissions are int.

This commit is contained in:
Rapptz 2017-02-21 08:19:32 -05:00
parent b7488d7c6e
commit 702d596af9
2 changed files with 7 additions and 1 deletions

View File

@ -46,13 +46,16 @@ class Colour:
Attributes Attributes
------------ ------------
value : int value: int
The raw integer colour value. The raw integer colour value.
""" """
__slots__ = ('value',) __slots__ = ('value',)
def __init__(self, value): def __init__(self, value):
if not isinstance(value, int):
raise TypeError('Expected int parameter, received %s instead.' % value.__class__.__name__)
self.value = value self.value = value
def _get_byte(self, byte): def _get_byte(self, byte):

View File

@ -69,6 +69,9 @@ class Permissions:
__slots__ = ('value',) __slots__ = ('value',)
def __init__(self, permissions=0): def __init__(self, permissions=0):
if not isinstance(permissions, int):
raise TypeError('Expected int parameter, received %s instead.' % permissions.__class__.__name__)
self.value = permissions self.value = permissions
def __eq__(self, other): def __eq__(self, other):