Method Initialize

src/binding/python/model/common/python_config.cc:59–220

void ZVecPyConfig::Initialize(pybind11::module_ &m) {
  m.def("Initialize", [](py::args args, py::kwargs kwargs) -> py::none {
    py::dict config_dict;
    // parse args
    for (auto &arg : args) {
      if (py::isinstance<py::dict>(arg)) {
        for (auto item : arg.cast<py::dict>()) {
          config_dict[item.first] = item.second;
        }
      } else {
        throw py::type_error("Positional argument must be a dict if provided");
      }
    }

    // parser kwargs
    if (kwargs) {
      for (auto item : kwargs) {
        config_dict[item.first] = item.second;
      }
    }

    if (config_dict.empty()) {
      return py::none();
    }

    GlobalConfig::ConfigData data;
    // config memory_limit_mb
    if (has_key(config_dict, "memory_limit_mb")) {
      auto mb = get_if<int64_t>(config_dict, "memory_limit_mb").value();
      if (mb <= 0) throw py::value_error("memory_limit_mb must be positive");
      data.memory_limit_bytes = static_cast<uint64_t>(mb) * 1024 * 1024;
    }

    // config log
    bool has_log_type = has_key(config_dict, "log_type");
    bool has_log_level = has_key(config_dict, "log_level");
    if (has_log_type || has_log_level) {
      std::string log_type = "console";
      std::string log_level_str = "warn";

      if (has_log_type) {
        log_type = config_dict["log_type"].cast<std::string>();
      }
      if (has_log_level) {
        log_level_str = config_dict["log_level"].cast<std::string>();
      }
      auto log_level = str_to_loglevel(log_level_str);
      if (iequals(log_type, "file")) {
        std::string dir = DEFAULT_LOG_DIR;
        std::string basename = DEFAULT_LOG_BASENAME;
        uint32_t file_size = DEFAULT_LOG_FILE_SIZE;
        uint32_t overdue_days = DEFAULT_LOG_OVERDUE_DAYS;

        if (has_key(config_dict, "log_dir")) {
          dir = get_if<std::string>(config_dict, "log_dir").value();
        }
        if (has_key(config_dict, "log_basename")) {
          basename = get_if<std::string>(config_dict, "log_basename").value();