safer error class

stix2.0
= 2018-05-22 10:03:06 -04:00
parent e484b7c25f
commit 3fb1840289
1 changed files with 4 additions and 3 deletions

View File

@ -40,8 +40,9 @@ class DataSourceError(Exception):
def __str__(self): def __str__(self):
if self.root_exception: if self.root_exception:
self.message = "{} \"{}\"".format(self.message, self.root_exception) return "{} \"{}\"".format(self.message, self.root_exception)
return self.message else:
return self.message
class DataStoreMixin(object): class DataStoreMixin(object):
@ -149,7 +150,7 @@ class DataStoreMixin(object):
obj (STIX object OR dict OR str): The STIX object (or its ID) whose obj (STIX object OR dict OR str): The STIX object (or its ID) whose
relationships will be looked up. relationships will be looked up.
relationship_type (str): Only retrieve Relationships of this type. relationship_type (str): Only retrieve Relationships of this type.
If None, all relationships will be returned, regardless of type. If None, all relationships will be returned, regardless of type.
source_only (bool): Only retrieve Relationships for which this source_only (bool): Only retrieve Relationships for which this
object is the source_ref. Default: False. object is the source_ref. Default: False.
target_only (bool): Only retrieve Relationships for which this target_only (bool): Only retrieve Relationships for which this