From 30bbec771e2e80e436197de07b5a0179eb277b02 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Mon, 5 Feb 2024 16:49:25 +0100 Subject: [PATCH] use ext instead of mimetypes --- deefuzzer/tools/utils.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/deefuzzer/tools/utils.py b/deefuzzer/tools/utils.py index cb896ad..7406710 100644 --- a/deefuzzer/tools/utils.py +++ b/deefuzzer/tools/utils.py @@ -85,22 +85,22 @@ def replace_all(option, repl): def get_conf_dict(file): - mime_type = mimetypes.guess_type(file)[0] + filename, ext = os.path.splitext(path) # Do the type check first, so we don't load huge files that won't be used - if 'xml' in mime_type: - tree = ElementTree.parse(file) - root = tree.getroot() - xmldict = etree_to_dict(root) - return xmldict + if 'xml' in ext: + confile = open(file, 'r') + data = confile.read() + confile.close() + return xmltodict(data, 'utf-8') - elif 'yaml' in mime_type or 'yml' in mime_type: + elif 'yaml' in ext or 'yml' in ext: import yaml confile = open(file, 'r') conf = yaml.safe_load(confile) return conf - elif 'json' in mime_type: + elif 'json' in ext: import json confile = open(file, 'r') data = confile.read() -- 2.47.3