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

fix: changes

pull/926/head
Adrian Maraj 2024-07-23 14:22:27 +02:00
parent e4268554b9
commit 65275adb90
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
website/web/genericapi.py
View File

@ -802,7 +802,7 @@ class RecentCaptures(Resource): # type: ignore[misc]
params={'category': 'The category according to which the uuids are to be returned.'}, params={'category': 'The category according to which the uuids are to be returned.'},
required=False) required=False)
class CategoriesCaptures(Resource): # type: ignore[misc] class CategoriesCaptures(Resource): # type: ignore[misc]
def get(self, category: str | None=None) -> list[str] | dict[str, list[str]] | None: def get(self, category: str | None=None) -> list[str] | dict[str, any]:
categories = ['legitimate', 'parking-page', 'default-page', 'insti_usertution', 'captcha', categories = ['legitimate', 'parking-page', 'default-page', 'insti_usertution', 'captcha',
'authentication-form', 'adult-content', 'shop', 'malicious', 'clone', 'phishing', 'unclear'] 'authentication-form', 'adult-content', 'shop', 'malicious', 'clone', 'phishing', 'unclear']
if not category: if not category:
@ -816,8 +816,8 @@ class CategoriesCaptures(Resource): # type: ignore[misc]
all_categorized_uuids[uuid] = {c} all_categorized_uuids[uuid] = {c}
else: else:
all_categorized_uuids[uuid].add(c) all_categorized_uuids[uuid].add(c)
all_categorized_uuids = {uuid: list(categories) for uuid, categories in all_categorized_uuids.items()} all_categorized_uuids_list = {uuid: list(categories) for uuid, categories in all_categorized_uuids.items()}
return all_categorized_uuids return all_categorized_uuids_list
if not category in categories: if not category in categories:
return {'error': f'Invalid category: {category}'} return {'error': f'Invalid category: {category}'}
return list(get_indexing(flask_login.current_user).get_captures_category(category)) return list(get_indexing(flask_login.current_user).get_captures_category(category))