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