intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 36 insertions, 0 deletions

diff --git a/library/python/runtime_py3/__res.pyx b/library/python/runtime_py3/__res.pyx
new file mode 100644
index 0000000000..97190d9f29
--- /dev/null
+++ b/library/python/runtime_py3/__res.pyx
@@ -0,0 +1,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"