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

fix: Typing, make mypy happy

pull/904/head
Raphaël Vinot 2024-03-27 13:44:35 +01:00
parent 78c0089413
commit 0eaea85204
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lookyloo/lookyloo.py
View File

@ -761,7 +761,7 @@ class Lookyloo():
result.append(self.takedown_details(rendered_hostnode)) result.append(self.takedown_details(rendered_hostnode))
return result return result
def send_mail(self, capture_uuid: str, /, email: str='', comment: str | None=None) -> True | dict[str, Any]: def send_mail(self, capture_uuid: str, /, email: str='', comment: str | None=None) -> bool | dict[str, Any]:
'''Send an email notification regarding a specific capture''' '''Send an email notification regarding a specific capture'''
if not get_config('generic', 'enable_mail_notification'): if not get_config('generic', 'enable_mail_notification'):
return {"error": "Unable to send mail: mail notification disabled"} return {"error": "Unable to send mail: mail notification disabled"}

7
website/web/genericapi.py
View File

@ -380,16 +380,17 @@ class CaptureCookies(Resource): # type: ignore[misc]
def get(self, capture_uuid: str) -> dict[str, Any]: def get(self, capture_uuid: str) -> dict[str, Any]:
return json.loads(lookyloo.get_cookies(capture_uuid).read()) return json.loads(lookyloo.get_cookies(capture_uuid).read())
@api.route('/json/<string:capture_uuid>/report') @api.route('/json/<string:capture_uuid>/report')
@api.doc(description='Reports the url by sending an email to the investigation team', @api.doc(description='Reports the url by sending an email to the investigation team',
params={'capture_uuid': 'The UUID of the capture'}) params={'capture_uuid': 'The UUID of the capture'})
class CaptureReport(Resource): # type: ignore[misc] class CaptureReport(Resource): # type: ignore[misc]
@api.param('email', 'Email of the reporter, used by the analyst to get in touch.') # type: ignore[misc] @api.param('email', 'Email of the reporter, used by the analyst to get in touch.') # type: ignore[misc]
@api.param('comment', 'Description of the URL, will be given to the analyst.') # type: ignore[misc] @api.param('comment', 'Description of the URL, will be given to the analyst.') # type: ignore[misc]
def post(self, capture_uuid) -> True | dict[str, Any]: def post(self, capture_uuid: str) -> bool | dict[str, Any]:
parameters: dict[str, Any] = request.get_json(force=True) parameters: dict[str, Any] = request.get_json(force=True)
return lookyloo.send_mail(capture_uuid, parameters.get('email'), parameters.get('comment')) return lookyloo.send_mail(capture_uuid, parameters.get('email', ''), parameters.get('comment'))
# Just text
auto_report_model = api.model('AutoReportModel', { auto_report_model = api.model('AutoReportModel', {
'email': fields.String(description="Email of the reporter, used by the analyst to get in touch.", example=''), 'email': fields.String(description="Email of the reporter, used by the analyst to get in touch.", example=''),