obitools/obitools3
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 66 Wiki Activity

Cython API: column elements: added possibility to check if an element

Browse Source 
exists from its index, and a dict-like get() method
This commit is contained in:
Celine Mercier
2017-08-20 17:44:05 +02:00
parent c6d5436a58
commit 663a1a1091
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
python/obitools3/dms/column/column.pyx
View File

@ -371,8 +371,18 @@ cdef class Column_line :
self._column.set_item(self._index, elt_id, value) self._column.set_item(self._index, elt_id, value)
def __contains__(self, object element_name): def get(self, object elt_id, object default=None): # TODO returns default if None???
return (tobytes(element_name) in self._column.elements_names) if elt_id in self:
return self._column.get_item(self._index, elt_id)
else:
return default
def __contains__(self, object elt_id):
if type(elt_id) == int:
return elt_id < self._column.nb_elements_per_line
else:
return (tobytes(elt_id) in self._column.elements_names)
def __repr__(self) : def __repr__(self) :