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

Cython API: Added a function to get a column from its index in the view

Browse Source
This commit is contained in:
Celine Mercier
2017-07-11 11:36:42 +02:00
parent b729b8928f
commit f8895e879d
1 changed files with 26 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
python/obitools3/dms/view/view.pyx
View File

@ -174,19 +174,21 @@ cdef class View(OBIWrapper) :
def keys(self): def keys(self):
cdef int i
cdef Obiview_p pointer = self.pointer() cdef str col_alias
cdef int nb_column = pointer.infos.column_count cdef int i
cdef Alias_column_pair_p column_p = pointer.infos.column_references cdef Obiview_p pointer = self.pointer()
cdef int nb_column = pointer.infos.column_count
cdef Alias_column_pair_p column_p = pointer.infos.column_references
if not self.active() : if not self.active() :
raise OBIObjectClosedInstance() raise OBIObjectClosedInstance()
for i in range(nb_column) : for i in range(nb_column) :
col_alias = bytes2str(pointer.infos.column_references[i].alias) col_alias = bytes2str(column_p[i].alias)
yield col_alias yield col_alias
def get_column(self, def get_column(self,
object column_name): object column_name):
@ -194,8 +196,23 @@ cdef class View(OBIWrapper) :
raise OBIObjectClosedInstance() raise OBIObjectClosedInstance()
return Column.open(self, column_name) return Column.open(self, column_name)
def get_column_with_idx(self,
int column_idx):
cdef Obiview_p pointer = self.pointer()
cdef int nb_column = pointer.infos.column_count
if not self.active() :
raise OBIObjectClosedInstance()
if column_idx > nb_column :
raise IndexError(column_idx, "No column with this index")
return Column.open(self, pointer.infos.column_references[column_idx].alias)
cpdef delete_column(self, cpdef delete_column(self,
object column_name) : object column_name) :