1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
from typing import Union, Dict
import re
supported_data = Union[int, float, str]
class FastConfigParser:
"""
This class is to replace the much slower configparser provided by Python itself.
It's probably nowhere near as robust and supports only a fraction of the functionality of the real deal.
In it's current state it supports reading config headers and the key value pairs beneath it.
It also supports the contains syntax (So if the config has a header [Foo], "Foo" in config will be true) as well
as the getItem syntax config["foo"] returns a dict with the key value pairs in the header.
"""
header_regex = re.compile(r"\[(\w+?)\]\n(.*?)(?:(?=\n\[(?:\w+?)\])|\Z)", re.S)
key_value_regex = re.compile(r"([^=\n !]+)[ \t]*=[ \t]*(.*?)(?:(?=\s+(?:^[^=\n\t !<>\[]+)[ \t]*=[ \t]*[^=])|(?=\n\[)|\Z)", flags = re.S|re.M)
def __init__(self, data: str) -> None:
header_result = self.header_regex.findall(data)
self._parsed_data = {} # type: Dict[str, Dict[str, supported_data]]
for header, content in header_result:
extracted_key_value_pairs = {}
for key, value in self.key_value_regex.findall(content.rstrip()):
# Multiline are stored with a tab, so we need to remove that again
extracted_key_value_pairs[key] = value.replace("\n\t", "\n")
self._parsed_data[header] = extracted_key_value_pairs
def __contains__(self, key: str) -> bool:
return key in self._parsed_data
def __getitem__(self, key: str) -> Dict[str, supported_data]:
return self._parsed_data[key]
def __iter__(self):
return iter(self._parsed_data)
|
