diff --git a/user_scanner/core/formatter.py b/user_scanner/core/formatter.py index 87e883b..0abcdee 100644 --- a/user_scanner/core/formatter.py +++ b/user_scanner/core/formatter.py @@ -1,10 +1,10 @@ import json -from user_scanner.core.result import Result from typing import List -INDENT = " " -CSV_HEADER = "username,category,site_name,status,url,reason" +from user_scanner.core.result import CSV_TEMPLATE, Result +INDENT = " " +CSV_HEADER = CSV_TEMPLATE.replace("{", "").replace("}", "") def indentate(msg: str, indent: int): @@ -17,6 +17,7 @@ def indentate(msg: str, indent: int): def into_json(results: List[Result]) -> str: return json.dumps([r.to_dict() for r in results], indent=4) + def get_json_data(results: List[Result]) -> list: """Returns a list of dictionaries ready for JSON serialization.""" return [r.to_dict() for r in results]