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

@ -53,6 +53,9 @@ class Colour:
__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):