aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/runtime_py3/__res.pyx
blob: 97190d9f293b2e65d16243e442f5d58a4e015dab (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"