From b35b069e2b2952e9258c7d9c410486175cdc0eef Mon Sep 17 00:00:00 2001 From: Peter LaFosse Date: Tue, 27 Oct 2020 13:53:32 -0400 Subject: automatically coerce Variable.source_type to enumeration --- python/function.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'python/function.py') diff --git a/python/function.py b/python/function.py index 3b2a08bd..e9a0a5e9 100644 --- a/python/function.py +++ b/python/function.py @@ -799,9 +799,7 @@ class Variable(object): self._type = var_type def __repr__(self): - if self._type is None: - return "" % self.name - return "" % (self._type.get_string_before_name(), self.name, self._type.get_string_after_name()) + return "" % (self.type.get_string_before_name(), self.name, self.type.get_string_after_name()) def __str__(self): return self.name @@ -831,6 +829,9 @@ class Variable(object): @property def source_type(self): """:class:`~enums.VariableSourceType`""" + if not isinstance(self._source_type, VariableSourceType): + self._source_type = VariableSourceType(self._source_type) + return self._source_type @source_type.setter @@ -894,7 +895,7 @@ class Variable(object): def to_BNVariable(self): v = core.BNVariable() - v.type = self._source_type + v.type = self.source_type v.index = self._index v.storage = self._storage return v -- cgit v1.3.1