From 4818d450e4ac02d80749b2ddc3474a632e2eecdc Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Sun, 2 Mar 2025 17:57:56 +0000 Subject: rename DataReader class to more appropriate name FileHandler --- gdpr_obfuscator/__init__.py | 4 ++-- gdpr_obfuscator/read.py | 2 +- test/test_read.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py index 694c512..6c0e47e 100644 --- a/gdpr_obfuscator/__init__.py +++ b/gdpr_obfuscator/__init__.py @@ -1,4 +1,4 @@ -from .read import DataReader +from .read import FileHandler from .obfuscate import obfuscate_data from .utils import Utilities from typing import List @@ -6,7 +6,7 @@ from typing import List class Obfuscator: def __init__(self): - self.reader = DataReader() + self.reader = FileHandler() self.utils = Utilities() def process_s3(self, input: str) -> bytes: diff --git a/gdpr_obfuscator/read.py b/gdpr_obfuscator/read.py index 27f2a3d..a3761ae 100644 --- a/gdpr_obfuscator/read.py +++ b/gdpr_obfuscator/read.py @@ -5,7 +5,7 @@ from typing import List, Dict from .utils import Utilities -class DataReader: +class FileHandler: """ A class to read CSV data from a local file, S3 object, or string. Near the project completion, support for JSON/Parquet files will be added. diff --git a/test/test_read.py b/test/test_read.py index 81ba053..4cd376f 100644 --- a/test/test_read.py +++ b/test/test_read.py @@ -1,9 +1,9 @@ import boto3 from moto import mock_aws -from gdpr_obfuscator.read import DataReader +from gdpr_obfuscator.read import FileHandler import pytest -reader = DataReader() +reader = FileHandler() def test_empty_csv_should_return_no_content(): -- cgit v1.2.3