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
------------
value : int
value: int
The raw integer colour value.
"""
__slots__ = ('value',)
def __init__(self, value):
if not isinstance(value, int):
raise TypeError('Expected int parameter, received %s instead.' % value.__class__.__name__)
self.value = value
def _get_byte(self, byte):

View File

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