From 907ac1e93565ae9bd7320482bcedd37fa90a7319 Mon Sep 17 00:00:00 2001 From: Jakub Onderka Date: Fri, 24 Dec 2021 16:31:10 +0100 Subject: [PATCH] fix: [ods_enrich] Try to fix reading bytesio --- misp_modules/modules/expansion/ods_enrich.py | 2 +- tests/test_expansions.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/misp_modules/modules/expansion/ods_enrich.py b/misp_modules/modules/expansion/ods_enrich.py index 8b66688..69aca77 100644 --- a/misp_modules/modules/expansion/ods_enrich.py +++ b/misp_modules/modules/expansion/ods_enrich.py @@ -36,7 +36,7 @@ def handler(q=False): num_sheets = len(doc.sheets) try: for i in range(0, num_sheets): - ods = pandas_ods_reader.read_ods(ods_file, i, headers=False) + ods = pandas_ods_reader.algo.read_data(pandas_ods_reader.parsers.ods, ods_file, i, headers=False) ods_content = ods_content + "\n" + ods.to_string(max_rows=None) return {'results': [{'types': ['freetext'], 'values': ods_content, 'comment': ".ods-to-text from file " + filename}, {'types': ['text'], 'values': ods_content, 'comment': ".ods-to-text from file " + filename}]} diff --git a/tests/test_expansions.py b/tests/test_expansions.py index 7af78b7..841a39a 100644 --- a/tests/test_expansions.py +++ b/tests/test_expansions.py @@ -304,7 +304,7 @@ class TestExpansions(unittest.TestCase): encoded = b64encode(f.read()).decode() query = {"module": "ods_enrich", "attachment": filename, "data": encoded} response = self.misp_modules_post(query) - self.assertEqual(self.get_values(response), '\n column_0\n0 ods test') + self.assertEqual(self.get_values(response), '\n column.0\n0 ods test') def test_odt(self): filename = 'test.odt'