#include "module.h"

#include <ctranslate2/logging.h>

namespace ctranslate2 {

namespace python {

void register_logging(py::module& m) {

py::enum_<ctranslate2::LogLevel>(m, "LogLevel")

.value("Off", ctranslate2::LogLevel::Off)

.value("Critical", ctranslate2::LogLevel::Critical)

.value("Error", ctranslate2::LogLevel::Error)

.value("Warning", ctranslate2::LogLevel::Warning)

.value("Info", ctranslate2::LogLevel::Info)

.value("Debug", ctranslate2::LogLevel::Debug)

.value("Trace", ctranslate2::LogLevel::Trace)

.export_values();

m.def("set_log_level", &ctranslate2::set_log_level);

m.def("get_log_level", &ctranslate2::get_log_level);

}

}

}