implement INI and CSV file parsing

yasha/parsers.py

@@ -60,11 +60,46 @@ def parse_svd(file):
         "peripherals": svd.peripherals,
     }
 
+
+def parse_ini(file):
+    from configparser import ConfigParser
+    from io import TextIOWrapper
+    cfg = ConfigParser()
+    # yasha opens files in binary mode, configparser expects files in text mode
+    content = file.read().decode(ENCODING)
+    cfg.read_string(content)
+    result = dict(cfg)
+    for section, data in result.items():
+        result[section] = dict(data)
+    return result
+
+
+def parse_csv(file):
+    from csv import reader, DictReader, Sniffer
+    from io import TextIOWrapper
+    from os.path import basename, splitext
+    assert file.name.endswith('.csv')
+    name = splitext(basename(file.name))[0]  # get the filename without the extension
+    content = TextIOWrapper(file, encoding='utf-8', errors='replace')
+    sample = content.read(1024)
+    content.seek(0)
+    csv = list()
+    if Sniffer().has_header(sample):
+        for row in DictReader(content):
+            csv.append(dict(row))
+    else:
+        for row in reader(content):
+            csv.append(row)
+    return {name: csv}
+
+
 PARSERS = {
     '.json': parse_json,
     '.yaml': parse_yaml,
     '.yml': parse_yaml,
     '.toml': parse_toml,
     '.xml': parse_xml,
     '.svd': parse_svd,
+    '.ini': parse_ini,
+    '.csv': parse_csv
 }