fix: Properly match IO in load event

pull/665/head
Raphaël Vinot 2020-11-25 13:34:13 +01:00
parent f254e15bd4
commit 4c2ee4fd2f
1 changed files with 2 additions and 2 deletions

View File

@ -6,7 +6,7 @@ import json
import os import os
import base64 import base64
import sys import sys
from io import BytesIO, RawIOBase from io import BytesIO, BufferedIOBase, TextIOBase
from zipfile import ZipFile from zipfile import ZipFile
import uuid import uuid
from collections import defaultdict from collections import defaultdict
@ -1150,7 +1150,7 @@ class MISPEvent(AbstractMISP):
def load(self, json_event: Union[IO, str, bytes, dict], validate: bool = False, metadata_only: bool = False): def load(self, json_event: Union[IO, str, bytes, dict], validate: bool = False, metadata_only: bool = False):
"""Load a JSON dump from a pseudo file or a JSON string""" """Load a JSON dump from a pseudo file or a JSON string"""
if isinstance(json_event, RawIOBase): if isinstance(json_event, (BufferedIOBase, TextIOBase)):
json_event = json_event.read() # type: ignore json_event = json_event.read() # type: ignore
if isinstance(json_event, (str, bytes)): if isinstance(json_event, (str, bytes)):