103 lines
3.7 KiB
Cython
103 lines
3.7 KiB
Cython
|
#cython: language_level=3
|
||
|
|
|
||
|
|
from .capi.obidmscolumn cimport obi_close_column,\
|
||
|
|
obi_truncate_and_close_column, \
|
||
|
|
obi_column_get_obiseq_with_elt_name, \
|
||
|
|
obi_column_get_obiseq_with_elt_idx, \
|
||
|
|
obi_column_set_obiseq_with_elt_name, \
|
||
|
|
obi_column_set_obiseq_with_elt_idx
|
||
|
|
from .capi.obierrno cimport obi_errno
|
||
|
|
from .capi.obitypes cimport OBIIdx_NA, const_char_p
|
||
|
|
|
||
|
|
from obitools3.utils cimport str2bytes, bytes2str
|
||
|
|
|
||
|
|
|
||
|
|
cdef class OBIDMS_column_seq(OBIDMS_column):
|
||
|
|
|
||
|
|
cpdef object get_line(self, index_t line_nb):
|
||
|
|
cdef bytes value
|
||
|
|
cdef object result
|
||
|
|
value = <bytes> obi_column_get_obiseq_with_elt_idx(self.pointer, line_nb, 0)
|
||
|
|
if obi_errno > 0 :
|
||
|
|
raise IndexError(line_nb)
|
||
|
|
if value == OBIIdx_NA :
|
||
|
|
result = None
|
||
|
|
else :
|
||
|
|
result = bytes2str(value)
|
||
|
|
return result
|
||
|
|
|
||
|
|
cpdef set_line(self, index_t line_nb, object value):
|
||
|
|
raise Exception("Column is read-only")
|
||
|
|
|
||
|
|
cpdef close(self):
|
||
|
|
if obi_close_column(self.pointer) < 0 :
|
||
|
|
raise Exception("Problem closing a column")
|
||
|
|
|
||
|
|
|
||
|
|
cdef class OBIDMS_column_seq_writable(OBIDMS_column_seq):
|
||
|
|
|
||
|
|
cpdef set_line(self, index_t line_nb, object value):
|
||
|
|
if obi_column_set_obiseq_with_elt_idx(self.pointer, line_nb, 0, str2bytes(value)) < 0:
|
||
|
|
raise Exception("Problem setting a value in a column")
|
||
|
|
|
||
|
|
cpdef close(self):
|
||
|
|
if obi_truncate_and_close_column(self.pointer) < 0 :
|
||
|
|
raise Exception("Problem closing a column")
|
||
|
|
|
||
|
|
|
||
|
|
cdef class OBIDMS_column_seq_multi_elts(OBIDMS_column_seq):
|
||
|
|
|
||
|
|
cpdef object get_item(self, index_t line_nb, str element_name):
|
||
|
|
cdef bytes value
|
||
|
|
cdef object result
|
||
|
|
value = <bytes> obi_column_get_obiseq_with_elt_name(self.pointer, line_nb, str2bytes(element_name))
|
||
|
|
if obi_errno > 0 :
|
||
|
|
raise IndexError(line_nb, element_name)
|
||
|
|
if value == OBIIdx_NA :
|
||
|
|
result = None
|
||
|
|
else :
|
||
|
|
result = bytes2str(value)
|
||
|
|
return result
|
||
|
|
|
||
|
|
cpdef object get_line(self, index_t line_nb) :
|
||
|
|
cdef bytes value
|
||
|
|
cdef object result
|
||
|
|
cdef index_t i
|
||
|
|
cdef bint all_NA
|
||
|
|
result = {}
|
||
|
|
all_NA = True
|
||
|
|
for i in range(self.nb_elements_per_line) :
|
||
|
|
value = <bytes> obi_column_get_obiseq_with_elt_idx(self.pointer, line_nb, i)
|
||
|
|
if obi_errno > 0 :
|
||
|
|
raise IndexError(line_nb)
|
||
|
|
result[self.elements_names[i]] = bytes2str(value)
|
||
|
|
if all_NA and (value != OBIIdx_NA) :
|
||
|
|
all_NA = False
|
||
|
|
if all_NA :
|
||
|
|
result = None
|
||
|
|
return result
|
||
|
|
|
||
|
|
cpdef set_item(self, index_t line_nb, str element_name, str value):
|
||
|
|
raise Exception("Column is read-only")
|
||
|
|
|
||
|
|
cpdef set_line(self, index_t line_nb, object values):
|
||
|
|
raise Exception("Column is read-only")
|
||
|
|
|
||
|
|
|
||
|
|
cdef class OBIDMS_column_seq_multi_elts_writable(OBIDMS_column_seq_multi_elts):
|
||
|
|
|
||
|
|
cpdef set_item(self, index_t line_nb, str element_name, str value):
|
||
|
|
if obi_column_set_obiseq_with_elt_name(self.pointer, line_nb, str2bytes(element_name), str2bytes(value)) < 0:
|
||
|
|
raise Exception("Problem setting a value in a column")
|
||
|
|
|
||
|
|
cpdef set_line(self, index_t line_nb, object values):
|
||
|
|
cdef str value
|
||
|
|
for element_name in values :
|
||
|
|
value = values[element_name]
|
||
|
|
self.set_item(line_nb, element_name, value)
|
||
|
|
|
||
|
|
cpdef close(self):
|
||
|
|
if obi_truncate_and_close_column(self.pointer) < 0 :
|
||
|
|
raise Exception("Problem closing a column")
|
||
|
|
|
||
|
|
|