1 files changed, 58 insertions, 0 deletions
diff --git a/src/python/xlsx.cpp b/src/python/xlsx.cpp
new file mode 100644
index 0000000..56cf47d
--- /dev/null
+++ b/src/python/xlsx.cpp
@@ -0,0 +1,58 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/*
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+#include "xlsx.hpp"
+#include "global.hpp"
+
+#ifdef __ORCUS_PYTHON_XLSX
+#include "document.hpp"
+#include "orcus/orcus_xlsx.hpp"
+#include "orcus/spreadsheet/document.hpp"
+#include "orcus/spreadsheet/factory.hpp"
+#endif
+
+namespace orcus { namespace python {
+
+#ifdef __ORCUS_PYTHON_XLSX
+
+PyObject* xlsx_read(PyObject* /*module*/, PyObject* args, PyObject* kwargs)
+{
+    stream_with_formulas data = read_stream_and_formula_params_from_args(args, kwargs);
+    if (!data.stream)
+        return nullptr;
+
+    try
+    {
+        spreadsheet::range_size_t ss{1048576, 16384};
+        std::unique_ptr<spreadsheet::document> doc = std::make_unique<spreadsheet::document>(ss);
+        spreadsheet::import_factory fact(*doc);
+        fact.set_recalc_formula_cells(data.recalc_formula_cells);
+        fact.set_formula_error_policy(data.error_policy);
+        orcus_xlsx app(&fact);
+
+        return import_from_stream_into_document(data.stream.get(), app, std::move(doc));
+    }
+    catch (const std::exception& e)
+    {
+        set_python_exception(PyExc_RuntimeError, e);
+        return nullptr;
+    }
+}
+
+#else
+
+PyObject* xlsx_read(PyObject*, PyObject*, PyObject*)
+{
+    PyErr_SetString(PyExc_RuntimeError, "The xlsx module is not enabled.");
+    return nullptr;
+}
+
+#endif
+
+}}
+
+/* vim:set shiftwidth=4 softtabstop=4 expandtab: */