From 6083543b7a259fa7ac967eb3e0d787a1dda2d7f4 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Sun, 2 Mar 2025 18:29:28 +0000 Subject: improve type validation in Utilities methods --- gdpr_obfuscator/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'gdpr_obfuscator') 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) -- cgit v1.2.3