diff options
| -rw-r--r-- | cli.py | 2 | ||||
| -rw-r--r-- | obfuscator/csv_reader.py | 2 |
2 files changed, 2 insertions, 2 deletions
@@ -1,5 +1,5 @@ import argparse -from obfuscator import csv_reader +from obfuscator.csv_reader import read def main(): parser = argparse.ArgumentParser(description="gdpr-obfuscator") diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py index 8ee40e9..8a9be9b 100644 --- a/obfuscator/csv_reader.py +++ b/obfuscator/csv_reader.py @@ -11,7 +11,7 @@ if not logger.handlers: handler.setFormatter(formatter) logger.addHandler(handler) -def csv_reader(content: str) -> List[Dict[str, str]]: +def read(content: str) -> List[Dict[str, str]]: f = StringIO(content) reader = csv.DictReader(f) logger.info("Finished reading CSV!") |