CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: Make mypy happy

pull/186/head
Raphaël Vinot 2021-03-30 16:43:57 +02:00
parent 9f88edd74d
commit a8711a18f1
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lookyloo/modules.py
View File

@ -17,7 +17,7 @@ import vt # type: ignore
from vt.error import APIError # type: ignore from vt.error import APIError # type: ignore
from pysanejs import SaneJS from pysanejs import SaneJS
from pyeupi import PyEUPI from pyeupi import PyEUPI
from pymisp import PyMISP, MISPEvent from pymisp import PyMISP, MISPEvent, MISPAttribute
class MISP(): class MISP():
@ -65,7 +65,7 @@ class MISP():
def get_existing_event(self, permaurl: str) -> Optional[str]: def get_existing_event(self, permaurl: str) -> Optional[str]:
attributes = self.client.search('attributes', value=permaurl, limit=1, page=1, pythonify=True) attributes = self.client.search('attributes', value=permaurl, limit=1, page=1, pythonify=True)
if not attributes: if not attributes or not isinstance(attributes[0], MISPAttribute):
return None return None
url = f'{self.client.root_url}/events/{attributes[0].event_id}' url = f'{self.client.root_url}/events/{attributes[0].event_id}'
return url return url

4
website/web/__init__.py
View File

@ -886,6 +886,8 @@ def web_misp_push_view(tree_uuid: str):
if error: if error:
return redirect(url_for('tree', tree_uuid=tree_uuid)) return redirect(url_for('tree', tree_uuid=tree_uuid))
# After this point, event is a MISPEvent
if request.method == 'POST': if request.method == 'POST':
# event is a MISPEvent at this point # event is a MISPEvent at this point
# Submit the event # Submit the event
@ -900,7 +902,7 @@ def web_misp_push_view(tree_uuid: str):
return redirect(url_for('tree', tree_uuid=tree_uuid)) return redirect(url_for('tree', tree_uuid=tree_uuid))
else: else:
# the 1st attribute in the event is the link to lookyloo # the 1st attribute in the event is the link to lookyloo
existing_misp_url = lookyloo.misp.get_existing_event(event.attributes[0].value) existing_misp_url = lookyloo.misp.get_existing_event(event.attributes[0].value) # type: ignore
fav_tags = lookyloo.misp.get_fav_tags() fav_tags = lookyloo.misp.get_fav_tags()