diff options
| -rw-r--r-- | gdpr_obfuscator/utils.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/gdpr_obfuscator/utils.py b/gdpr_obfuscator/utils.py index 87dd216..1012358 100644 --- a/gdpr_obfuscator/utils.py +++ b/gdpr_obfuscator/utils.py @@ -1,12 +1,12 @@ import csv import io -from typing import List, Dict +from typing import List, Dict, Tuple import json class Utilities: @staticmethod - def process_json_input(json_input: str): + def process_json_input(json_input: str) -> Tuple[str, List[str]]: data = json.loads(json_input) if not data.get("file_path") or not data.get("pii_fields"): @@ -17,7 +17,7 @@ class Utilities: return data["file_path"], data["pii_fields"] @staticmethod - def get_s3_path(uri): + def get_s3_path(uri) -> Tuple[str, str]: parts = uri.replace("s3://", "").split("/") bucket = parts.pop(0) key = "/".join(parts) |