aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/runtime_py3/__res.pyx
blob: 306e780b2b25a1e1acd04085bd2f5df83929479b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from _codecs import utf_8_decode, utf_8_encode

from libcpp cimport bool 
 
from util.generic.string cimport TString, TStringBuf
 
 
cdef extern from "library/cpp/resource/resource.h" namespace "NResource":
    cdef size_t Count() except + 
    cdef TStringBuf KeyByIndex(size_t idx) except + 
    cdef bool FindExact(const TStringBuf key, TString* result) nogil except +
 
 
def count(): 
    return Count() 
 
 
def key_by_index(idx): 
    cdef TStringBuf ret = KeyByIndex(idx) 
 
    return ret.Data()[:ret.Size()] 
 
 
def find(s): 
    cdef TString res
 
    if isinstance(s, str):
        s = utf_8_encode(s)[0]

    if FindExact(TStringBuf(s, len(s)), &res): 
        return res.c_str()[:res.length()] 
 
    return None 


include "importer.pxi"