Merge pull request #84 from RichieB2B/ncsc-nl/fix-distribution

Allow 'Your organisation only' distribution for attributes
pull/55/head
Raphaël Vinot 2017-02-21 11:25:44 +01:00 committed by GitHub
commit 9f781a65e5
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))