diff options
| -rw-r--r-- | cli.py | 4 | ||||
| -rw-r--r-- | obfuscator/read.py (renamed from obfuscator/csv_reader.py) | 2 | ||||
| -rw-r--r-- | test/test_read.py (renamed from test/test_csv_reader.py) | 2 |
3 files changed, 4 insertions, 4 deletions
@@ -1,5 +1,5 @@ import argparse -from obfuscator.csv_reader import CSVReader +from obfuscator.read import DataReader from obfuscator.obfuscate import obfuscate from obfuscator.logger import get_logger from obfuscator.csv_writer import create_byte_stream @@ -32,7 +32,7 @@ def main(): log_level = "DEBUG" if args.verbose else "INFO" logger = get_logger("CLI", log_level) - reader = CSVReader(log_level) + reader = DataReader(log_level) if args.local and not args.s3: logger.debug("Read data from local path") diff --git a/obfuscator/csv_reader.py b/obfuscator/read.py index 1f503d7..b704643 100644 --- a/obfuscator/csv_reader.py +++ b/obfuscator/read.py @@ -7,7 +7,7 @@ from obfuscator.logger import get_logger from obfuscator.utils import Utilities -class CSVReader: +class DataReader: """ 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_csv_reader.py b/test/test_read.py index d6129e7..903ab5d 100644 --- a/test/test_csv_reader.py +++ b/test/test_read.py @@ -1,6 +1,6 @@ import boto3 from moto import mock_aws -from obfuscator.csv_reader import CSVReader +from obfuscator.read import CSVReader import pytest reader = CSVReader(log_level="DEBUG") |