Merge branch 'master' of github.com:CIRCL/PyMISP

pull/55/head
Raphaël Vinot 2017-02-21 15:45:36 +01:00
commit 1ef63604e1
1 changed files with 1 additions and 1 deletions

View File

@ -151,7 +151,7 @@ class MISPAttribute(object):
self.to_ids = bool(int(type_defaults['to_ids'])) self.to_ids = bool(int(type_defaults['to_ids']))
if kwargs.get('comment'): if kwargs.get('comment'):
self.comment = kwargs['comment'] self.comment = kwargs['comment']
if kwargs.get('distribution'): if kwargs.get('distribution') is not None:
self.distribution = int(kwargs['distribution']) self.distribution = int(kwargs['distribution'])
if self.distribution not in [0, 1, 2, 3, 4, 5]: if self.distribution not in [0, 1, 2, 3, 4, 5]:
raise NewAttributeError('{} is invalid, the distribution has to be in 0, 1, 2, 3, 4, 5'.format(self.distribution)) raise NewAttributeError('{} is invalid, the distribution has to be in 0, 1, 2, 3, 4, 5'.format(self.distribution))