47 lines
1.0 KiB
C++
47 lines
1.0 KiB
C++
|
#include "vdf_parser.hpp"
|
||
|
|
#include <fstream>
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
#include <pybind11/pybind11.h>
|
||
|
|
#include <pybind11/stl_bind.h>
|
||
|
|
|
||
|
|
namespace py = pybind11;
|
||
|
|
|
||
|
|
struct python_object
|
||
|
|
{
|
||
|
|
py::dict dict;
|
||
|
|
std::string name;
|
||
|
|
|
||
|
|
void add_attribute(std::string key, std::string value)
|
||
|
|
{
|
||
|
|
dict[py::cast(std::move(key))] = py::cast(std::move(value));
|
||
|
|
}
|
||
|
|
void add_child(std::unique_ptr< python_object > child)
|
||
|
|
{
|
||
|
|
std::string n = std::move(child->name);
|
||
|
|
dict[py::cast(n)] = std::move(child->dict);
|
||
|
|
}
|
||
|
|
void set_name(std::string n)
|
||
|
|
{
|
||
|
|
name = std::move(n);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
py::dict py_read_file(const char* filename)
|
||
|
|
{
|
||
|
|
std::ifstream input(filename);
|
||
|
|
return tyti::vdf::read<python_object>(input).dict;
|
||
|
|
}
|
||
|
|
|
||
|
|
py::dict py_read(const std::string& filename)
|
||
|
|
{
|
||
|
|
return tyti::vdf::read<python_object>(std::begin(filename), std::end(filename)).dict;
|
||
|
|
}
|
||
|
|
|
||
|
|
PYBIND11_MODULE(vdf, m)
|
||
|
|
{
|
||
|
|
m.doc() = "Read and Write Valve's vdf files.";
|
||
|
|
|
||
|
|
m.def("read", &py_read, "Read vdf from memory");
|
||
|
|
m.def("read_file", &py_read_file, "Read vdf file");
|
||
|
|
}
|