rename CSVReader to DataReader & update references

3 files changed, 4 insertions, 4 deletions

diff --git a/cli.py b/cli.py
index 7ffccd8..5100e2b 100644
--- a/cli.py
+++ b/cli.py
@@ -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")