1 files changed, 117 insertions, 0 deletions
diff --git a/tipboard/parser.py b/tipboard/parser.py
new file mode 100644
index 0000000..05d23c3
--- /dev/null
+++ b/tipboard/parser.py
@@ -0,0 +1,117 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+import glob
+import os
+import yaml
+
+from tipboard import settings
+
+
+class WrongSumOfRows(Exception):
+    pass
+
+
+def _get_tiles_dict(col):
+    return col.values().pop()
+
+
+def _get_cols(rows):
+    #TODO: validation col_1_of_4
+    cols = []
+    for col in rows.values()[0]:
+        cols.append(col)
+    return cols
+
+
+def _get_rows(layout):
+    """Validates and returns number of rows."""
+    rows_data = []
+    rows_class = []
+    for row in layout:
+        rows_data.append(row)
+        rows_class.append(row.keys().pop())
+    rows_count = 0
+    sum_of_rows = []
+    for row_class in rows_class:
+        splited_class = row_class.split('_')  # ex: row_1_of_2
+        row = splited_class[1]
+        of_rows = int(splited_class[3])
+        if rows_count == 0:
+            rows_count = int(of_rows)
+            sum_of_rows.append(int(row))
+        elif not rows_count == of_rows:
+            raise WrongSumOfRows('The sum of the lines is incorrect.')
+        else:
+            sum_of_rows.append(int(row))
+    if not sum(sum_of_rows) == rows_count:
+        raise WrongSumOfRows('The sum of the lines is incorrect.')
+    return rows_data
+
+
+def _find_tiles_names(layout):
+    name_list, key_list = [], []
+    for row in _get_rows(layout):
+        for col in _get_cols(row):
+            for tile_dict in _get_tiles_dict(col):
+                name = tile_dict['tile_template']
+                key = tile_dict['tile_id']
+                if key not in key_list:
+                    key_list.append(key)
+                    if name not in name_list:
+                        name_list.append(name)
+    return name_list, key_list
+
+
+def get_config_files_names():
+    """
+    Return all configs files' names (without '.yaml' ext.) from user space
+    (~/.tipboard/)
+    """
+    configs_names = []
+    configs_dir = os.path.join(settings._user_config_dir, '*.yaml')
+    for config_path in glob.glob(configs_dir):
+        filename = os.path.basename(config_path)
+        head, ext = os.path.splitext(filename)
+        configs_names.append(head)
+    return configs_names
+
+
+def config_file_name2path(config_name):
+    """
+    Return file path to *config_name* (eg. 'layout_config')
+    """
+    path = os.path.join(
+        settings._user_config_dir, '.'.join([config_name, 'yaml'])
+    )
+    return path
+
+
+def get_tiles_configs():
+    """
+    Return dict with both tiles' keys and ids from all available configs
+    """
+    tiles_configs = {
+        'tiles_keys': set(),
+        'tiles_names': set()
+    }
+    configs_names = get_config_files_names()
+    for config_name in configs_names:
+        parsed_config = process_layout_config(config_name)
+        tiles_configs['tiles_keys'].update(set(parsed_config['tiles_keys']))
+        tiles_configs['tiles_names'].update(set(parsed_config['tiles_names']))
+    return tiles_configs
+
+
+def process_layout_config(layout_name):
+    config_path = config_file_name2path(layout_name)
+    with open(config_path, 'r') as layout_config:
+        config = yaml.load(layout_config)
+    layout = config['layout']
+    config['tiles_names'], config['tiles_keys'] = _find_tiles_names(layout)
+    return config