aboutsummaryrefslogtreecommitdiffstats
path: root/obfuscator/csv_reader.py
blob: 4ae8a063c8def5b0ae88b33e6966c6f17f9a3e59 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import csv
from io import StringIO
from typing import List, Dict
import logging

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
if not logger.handlers:
    handler = logging.StreamHandler()
    formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
    handler.setFormatter(formatter)
    logger.addHandler(handler)

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
git.ajschof.me — hosted by ajschofield — powered by cgit