93 lines
3.8 KiB
C++
93 lines
3.8 KiB
C++
#include <pybind11/functional.h>
|
||
#include <pybind11/pybind11.h>
|
||
#include <pybind11/stl.h>
|
||
|
||
#include "sdk.h"
|
||
#include "util.h"
|
||
|
||
namespace py = pybind11;
|
||
|
||
int WxEnableRecvMsgPy(const std::function<int(WxMessage_t)> &onMsg) { return WxEnableRecvMsg(onMsg); }
|
||
|
||
py::object WxExecDbQueryPy(std::wstring db, std::wstring sql)
|
||
{
|
||
py::list results;
|
||
SqlRetVector_t cResults = WxExecDbQuery(db, sql);
|
||
for (auto vv = cResults.begin(); vv != cResults.end(); vv++) {
|
||
py::dict row;
|
||
for (auto v = vv->begin(); v != vv->end(); v++) {
|
||
switch (v->type) {
|
||
case 1: { // SQLITE_INTEGER
|
||
row[py::cast(v->column)] = stoi(v->content);
|
||
break;
|
||
}
|
||
case 2: { // SQLITE_FLOAT
|
||
row[py::cast(v->column)] = stof(v->content);
|
||
break;
|
||
}
|
||
case 3: { // SQLITE_TEXT
|
||
row[py::cast(v->column)] = String2Wstring(v->content);
|
||
break;
|
||
}
|
||
case 4: { // SQLITE_BLOB
|
||
row[py::cast(v->column)] = py::bytes(v->content.c_str(), v->content.size());
|
||
break;
|
||
}
|
||
default: {
|
||
row[py::cast(v->column)] = "";
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
results.append(row);
|
||
}
|
||
return results;
|
||
}
|
||
|
||
PYBIND11_MODULE(wcferry, m)
|
||
{
|
||
m.doc() = "SDK python API";
|
||
|
||
py::class_<WxMessage>(m, "WxMessage")
|
||
.def_readonly("id", &WxMessage::id)
|
||
.def_readonly("self", &WxMessage::self)
|
||
.def_readonly("type", &WxMessage::type)
|
||
.def_readonly("source", &WxMessage::source)
|
||
.def_readonly("xml", &WxMessage::xml)
|
||
.def_readonly("wxId", &WxMessage::wxId)
|
||
.def_readonly("roomId", &WxMessage::roomId)
|
||
.def_readonly("content", &WxMessage::content);
|
||
|
||
py::class_<WxContact>(m, "WxContact")
|
||
.def_readonly("wxId", &WxContact::wxId)
|
||
.def_readonly("wxCode", &WxContact::wxCode)
|
||
.def_readonly("wxName", &WxContact::wxName)
|
||
.def_readonly("wxCountry", &WxContact::wxCountry)
|
||
.def_readonly("wxProvince", &WxContact::wxProvince)
|
||
.def_readonly("wxCity", &WxContact::wxCity)
|
||
.def_readonly("wxGender", &WxContact::wxGender);
|
||
|
||
py::class_<WxDbTable>(m, "WxDbTable").def_readonly("table", &WxDbTable::table).def_readonly("sql", &WxDbTable::sql);
|
||
|
||
m.def("WxInitSDK", &WxInitSDK, "Initiate SDK. Return 0 on success，else on failure.");
|
||
m.def("WxEnableRecvMsg", &WxEnableRecvMsgPy, "Enable message receiving and provide a callback", py::arg("onMsg"));
|
||
m.def("WxDisableRecvMsg", &WxDisableRecvMsg, "Disable message receiving.");
|
||
m.def("WxSendTextMsg", &WxSendTextMsg, "Send text message.", py::arg("wxid"), py::arg("msg"),
|
||
py::arg("atWxids") = L"");
|
||
m.def("WxSendImageMsg", &WxSendImageMsg, "Send image message.", py::arg("wxid"), py::arg("path"));
|
||
m.def("WxGetSelfWxid", &WxGetSelfWxid, "Get Self Wxid.");
|
||
m.def("WxGetContacts", &WxGetContacts, py::return_value_policy::reference, "Get contact list.");
|
||
m.def("WxGetMsgTypes", &WxGetMsgTypes, py::return_value_policy::reference, "Get message types.");
|
||
m.def("WxGetDbNames", &WxGetDbNames, py::return_value_policy::reference, "Get DB names.");
|
||
m.def("WxGetDbTables", &WxGetDbTables, py::return_value_policy::reference, "Get DB tables.", py::arg("db"));
|
||
m.def("WxExecDbQuery", &WxExecDbQueryPy, py::return_value_policy::reference, "Get DB tables.", py::arg("db"),
|
||
py::arg("sql"));
|
||
m.def("WxAcceptNewFriend", &WxAcceptNewFriend, "Accept new friend application.", py::arg("v3"), py::arg("v4"));
|
||
|
||
#ifdef VERSION_INFO
|
||
m.attr("__version__") = VERSION_INFO;
|
||
#else
|
||
m.attr("__version__") = "dev";
|
||
#endif
|
||
}
|