MISP
/
misp-modules
mirror of https://github.com/MISP/misp-modules
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: [ods_enrich] Try to fix reading bytesio

pull/539/head
Jakub Onderka 2021-12-24 16:31:10 +01:00
parent 2842b27c50
commit 907ac1e935
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
misp_modules/modules/expansion/ods_enrich.py
View File

@ -36,7 +36,7 @@ def handler(q=False):
num_sheets = len(doc.sheets) num_sheets = len(doc.sheets)
try: try:
for i in range(0, num_sheets): 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) 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}, 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}]} {'types': ['text'], 'values': ods_content, 'comment': ".ods-to-text from file " + filename}]}

2
tests/test_expansions.py
View File

@ -304,7 +304,7 @@ class TestExpansions(unittest.TestCase):
encoded = b64encode(f.read()).decode() encoded = b64encode(f.read()).decode()
query = {"module": "ods_enrich", "attachment": filename, "data": encoded} query = {"module": "ods_enrich", "attachment": filename, "data": encoded}
response = self.misp_modules_post(query) 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): def test_odt(self):
filename = 'test.odt' filename = 'test.odt'