58 lines
1.5 KiB
Cython
58 lines
1.5 KiB
Cython
from collections.abc import MutableMapping
|
|
|
|
from av.error cimport err_check
|
|
|
|
|
|
cdef class _Dictionary:
|
|
def __cinit__(self, *args, **kwargs):
|
|
for arg in args:
|
|
self.update(arg)
|
|
if kwargs:
|
|
self.update(kwargs)
|
|
|
|
def __dealloc__(self):
|
|
if self.ptr != NULL:
|
|
lib.av_dict_free(&self.ptr)
|
|
|
|
def __getitem__(self, str key):
|
|
cdef lib.AVDictionaryEntry *element = lib.av_dict_get(self.ptr, key, NULL, 0)
|
|
if element != NULL:
|
|
return element.value
|
|
else:
|
|
raise KeyError(key)
|
|
|
|
def __setitem__(self, str key, str value):
|
|
err_check(lib.av_dict_set(&self.ptr, key, value, 0))
|
|
|
|
def __delitem__(self, str key):
|
|
err_check(lib.av_dict_set(&self.ptr, key, NULL, 0))
|
|
|
|
def __len__(self):
|
|
return err_check(lib.av_dict_count(self.ptr))
|
|
|
|
def __iter__(self):
|
|
cdef lib.AVDictionaryEntry *element = NULL
|
|
while True:
|
|
element = lib.av_dict_get(self.ptr, "", element, lib.AV_DICT_IGNORE_SUFFIX)
|
|
if element == NULL:
|
|
break
|
|
yield element.key
|
|
|
|
def __repr__(self):
|
|
return f"av.Dictionary({dict(self)!r})"
|
|
|
|
cpdef _Dictionary copy(self):
|
|
cdef _Dictionary other = Dictionary()
|
|
lib.av_dict_copy(&other.ptr, self.ptr, 0)
|
|
return other
|
|
|
|
|
|
class Dictionary(_Dictionary, MutableMapping):
|
|
pass
|
|
|
|
|
|
cdef _Dictionary wrap_dictionary(lib.AVDictionary *input_):
|
|
cdef _Dictionary output = Dictionary()
|
|
output.ptr = input_
|
|
return output
|