| 64 | } |
| 65 | |
| 66 | PyObject* Find(PyObject* self[[maybe_unused]], PyObject *const* args, Py_ssize_t nargs) noexcept { |
| 67 | if (nargs != 1) { |
| 68 | PyErr_Format(PyExc_TypeError, "__res.find takes 1 positional arguments but %z were given", nargs); |
| 69 | return nullptr; |
| 70 | } |
| 71 | |
| 72 | TStringBuf key; |
| 73 | if (PyUnicode_Check(args[0])) { |
| 74 | Py_ssize_t sz; |
| 75 | const char* data = PyUnicode_AsUTF8AndSize(args[0], &sz); |
| 76 | if (sz < 0) { |
| 77 | return nullptr; |
| 78 | } |
| 79 | key = {data, static_cast<size_t>(sz)}; |
| 80 | } else { |
| 81 | char* data = nullptr; |
| 82 | Py_ssize_t sz; |
| 83 | if (PyBytes_AsStringAndSize(args[0], &data, &sz) != 0) { |
| 84 | return nullptr; |
| 85 | } |
| 86 | key = {data, static_cast<size_t>(sz)}; |
| 87 | } |
| 88 | |
| 89 | return CallWithErrorTranslation([&]{ |
| 90 | TString res; |
| 91 | if (!NResource::FindExact(key, &res)) { |
| 92 | Py_RETURN_NONE; |
| 93 | } |
| 94 | return PyBytes_FromStringAndSize(res.data(), res.size()); |
| 95 | }); |
| 96 | } |
| 97 | |
| 98 | PyObject* Has(PyObject* self[[maybe_unused]], PyObject *const* args, Py_ssize_t nargs) noexcept { |
| 99 | if (nargs != 1) { |
no test coverage detected