mirror of https://github.com/MISP/misp-modules
fix: Fixed pep8 in the new module and related libraries
parent
7722e2cb93
commit
70b3079aa3
|
@ -86,4 +86,3 @@ def parse_pymisp_response(payload):
|
|||
event_attrs = payload.get("Attribute", [])
|
||||
objects = payload.get("Object")
|
||||
return _parse_data(event_attrs, objects)
|
||||
|
||||
|
|
|
@ -52,8 +52,7 @@ class MispAttribute(object):
|
|||
self.misp_type = misp_type
|
||||
|
||||
def __eq__(self, other):
|
||||
return (isinstance(other, self.__class__) and self.value == other.value and
|
||||
self.type == other.type)
|
||||
return (isinstance(other, self.__class__) and self.value == other.value and self.type == other.type)
|
||||
|
||||
def __repr__(self):
|
||||
return 'MispAttribute("{type}", "{category}", "{value}")'.format(type=self.type, category=self.category, value=self.value)
|
||||
|
|
|
@ -74,7 +74,8 @@ def import_misp_graph(
|
|||
if attr.type in vt_graph_api.Node.SUPPORTED_NODE_TYPES),
|
||||
1)
|
||||
|
||||
max_quotas_per_search = max(int(max_api_quotas / total_expandable_attrs), 1)
|
||||
max_quotas_per_search = max(
|
||||
int(max_api_quotas / total_expandable_attrs), 1)
|
||||
|
||||
previous_node_id = ""
|
||||
for attr in attributes_to_add:
|
||||
|
@ -85,8 +86,7 @@ def import_misp_graph(
|
|||
# If use_vt_to_connect_the_grap is True the nodes will be connected using
|
||||
# VT API.
|
||||
if use_vt_to_connect_the_graph:
|
||||
if (attr.type not in vt_graph_api.Node.SUPPORTED_NODE_TYPES and
|
||||
previous_node_id):
|
||||
if (attr.type not in vt_graph_api.Node.SUPPORTED_NODE_TYPES and previous_node_id):
|
||||
graph.add_link(previous_node_id, attr.value, "manual")
|
||||
else:
|
||||
graph.connect_with_graph(
|
||||
|
|
|
@ -5,8 +5,6 @@ response payload giving by MISP API directly.
|
|||
"""
|
||||
|
||||
|
||||
import json
|
||||
from lib.vt_graph_parser import errors
|
||||
from lib.vt_graph_parser.helpers.parsers import parse_pymisp_response
|
||||
from lib.vt_graph_parser.importers.base import import_misp_graph
|
||||
|
||||
|
|
|
@ -80,7 +80,6 @@ def handler(q=False):
|
|||
if group_viewers:
|
||||
group_viewers = group_viewers.split(',')
|
||||
|
||||
|
||||
graphs = from_pymisp_response(
|
||||
request, api_key, fetch_information=fetch_information,
|
||||
private=private, fetch_vt_enterprise=fetch_vt_enterprise,
|
||||
|
@ -94,7 +93,8 @@ def handler(q=False):
|
|||
links.append(graph.get_ui_link())
|
||||
|
||||
# This file will contains one VirusTotal graph link for each exported event
|
||||
file_data = str(base64.b64encode(bytes('\n'.join(links), 'utf-8')), 'utf-8')
|
||||
file_data = str(base64.b64encode(
|
||||
bytes('\n'.join(links), 'utf-8')), 'utf-8')
|
||||
return {'response': [], 'data': file_data}
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue