Initial extract.py

1 files changed, 39 insertions, 0 deletions

diff --git a/extract-data-from-abs-guide/extract.py b/extract-data-from-abs-guide/extract.py
new file mode 100644
index 0000000..baeb6e2
--- /dev/null
+++ b/extract-data-from-abs-guide/extract.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+import glob
+
+def chunks(lst, n):
+    """Yield successive n-sized chunks from lst."""
+    for i in range(0, len(lst), n):
+        yield lst[i:i + n]
+
+inputs = glob.glob('files/*.txt')
+
+TEMPLATE_CC = '# TEMPLATE: '
+
+for input in inputs:
+    print(input)
+    with open(input, 'r') as f:
+        lines = f.readlines()
+
+    if not len(lines):
+        continue
+
+    template = None
+
+    if lines[0].startswith(TEMPLATE_CC):
+        template = lines.pop(0).split(TEMPLATE_CC)[1]
+
+    lines = list(filter(bool, [line.strip().replace('\n', '') for line in lines]))
+
+    if template:
+        value_matrix = list(chunks(lines, 2))
+
+        for idx in range(len(value_matrix)):
+            value_matrix[idx].append(
+                    template.strip().format(value_matrix[idx][0])
+                    )
+    else:
+        value_matrix = chunks(lines, 3)
+
+    print(list(value_matrix))