Allow 'Your organisation only' distribution for attributes

pull/55/head
Richard van den Berg 2017-02-21 11:07:57 +01:00
parent 3493b26bd0
commit ef69016e4a
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']))
if kwargs.get('comment'):
self.comment = kwargs['comment']
if kwargs.get('distribution'):
if kwargs.get('distribution') is not None:
self.distribution = int(kwargs['distribution'])
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))