<reponame>Mikuana/rumydata
"""
rumydata row rules

These rules are applied to header rows and are generally not intended to be
used directly. These rules ensure that the headers are named as expected, is
not missing any names, or containing any extras.
"""
from collections import namedtuple
from typing import List

from rumydata._base import _BaseRule

# this named tuple is here to allow for setting the default argument without
# needing to import the Layout class, which results in a circular import
_default_thing = namedtuple('DefaultDict', ['layout', 'header_mode', 'empty_cols_ok'])


class Rule(_BaseRule):
    """ Header Rule """

    _default_args = (_default_thing({}, 'exact', False),)

    def __init__(self, columns):
        super().__init__()
        self.header_mode = columns.header_mode
        self.definition = columns.layout
        self.empty_cols_ok = columns.empty_cols_ok

    def _prepare(self, data: List[str]) -> tuple:
        if self.empty_cols_ok:
            return [x for x in data if x != ''],
        else:
            return data,


class NoExtra(Rule):
    """ No extra header elements Rule """

    def _evaluator(self):
        modes = {
            'exact': lambda x: all([y in self.definition for y in x]),
            'startswith': lambda x: all([any([y.startswith(z) for z in self.definition]) for y in x]),
            'contains': lambda x: all([any([z in y for z in self.definition]) for y in x])
        }
        return modes[self.header_mode]

    def _explain(self):
        return 'Header row must not have unexpected columns'


class NoMissing(Rule):
    """ No missing header elements Rule """

    def _evaluator(self):
        modes = {
            'exact': lambda x: all([y in x for y in self.definition]),
            'startswith': lambda x: all([any([z.startswith(y) for z in x]) for y in self.definition]),
            'contains': lambda x: all([any([y in z for z in x]) for y in self.definition])
        }
        return modes[self.header_mode]

    def _explain(self) -> str:
        return 'Header row must not be missing any expected columns'


class NoDuplicate(Rule):
    """ No duplicate header elements Rule """

    def _evaluator(self):
        def starts_with(x):
            ixs = [[y.startswith(z) for z in list(self.definition.keys())].index(True) for y in x]
            return len(ixs) == len(set(ixs))

        def contains(x):
            ixs = [[z in y for z in list(self.definition.keys())].index(True) for y in x]
            return len(ixs) == len(set(ixs))

        modes = {
            'exact': lambda x: len(x) == len(set(x)),
            'startswith': starts_with,
            'contains': contains
        }
        return modes[self.header_mode]

    def _explain(self):
        return 'Header row must not contain duplicate values'


class ColumnOrder(Rule):
    """ Fixed header element order Rule """

    def _evaluator(self):
        def starts_with(x):
            ixs = [[y.startswith(z) for z in list(self.definition.keys())].index(True) for y in x]
            return ixs == sorted(ixs)

        def contains(x):
            ixs = [[z in y for z in list(self.definition.keys())].index(True) for y in x]
            return ixs == sorted(ixs)

        modes = {
            'exact': lambda x: x == list(self.definition),
            'startswith': starts_with,
            'contains': contains
        }
        return modes[self.header_mode]

    def _explain(self):
        return 'Header row must explicitly match order of definition'

"""
rumydata row rules

These rules are applied to header rows and are generally not intended to be
used directly. These rules ensure that the headers are named as expected, is
not missing any names, or containing any extras.
"""
from collections import namedtuple
from typing import List

from rumydata._base import _BaseRule

# this named tuple is here to allow for setting the default argument without
# needing to import the Layout class, which results in a circular import
_default_thing = namedtuple('DefaultDict', ['layout', 'header_mode', 'empty_cols_ok'])


class Rule(_BaseRule):
    """ Header Rule """

    _default_args = (_default_thing({}, 'exact', False),)

    def __init__(self, columns):
        super().__init__()
        self.header_mode = columns.header_mode
        self.definition = columns.layout
        self.empty_cols_ok = columns.empty_cols_ok

    def _prepare(self, data: List[str]) -> tuple:
        if self.empty_cols_ok:
            return [x for x in data if x != ''],
        else:
            return data,


class NoExtra(Rule):
    """ No extra header elements Rule """

    def _evaluator(self):
        modes = {
            'exact': lambda x: all([y in self.definition for y in x]),
            'startswith': lambda x: all([any([y.startswith(z) for z in self.definition]) for y in x]),
            'contains': lambda x: all([any([z in y for z in self.definition]) for y in x])
        }
        return modes[self.header_mode]

    def _explain(self):
        return 'Header row must not have unexpected columns'


class NoMissing(Rule):
    """ No missing header elements Rule """

    def _evaluator(self):
        modes = {
            'exact': lambda x: all([y in x for y in self.definition]),
            'startswith': lambda x: all([any([z.startswith(y) for z in x]) for y in self.definition]),
            'contains': lambda x: all([any([y in z for z in x]) for y in self.definition])
        }
        return modes[self.header_mode]

    def _explain(self) -> str:
        return 'Header row must not be missing any expected columns'


class NoDuplicate(Rule):
    """ No duplicate header elements Rule """

    def _evaluator(self):
        def starts_with(x):
            ixs = [[y.startswith(z) for z in list(self.definition.keys())].index(True) for y in x]
            return len(ixs) == len(set(ixs))

        def contains(x):
            ixs = [[z in y for z in list(self.definition.keys())].index(True) for y in x]
            return len(ixs) == len(set(ixs))

        modes = {
            'exact': lambda x: len(x) == len(set(x)),
            'startswith': starts_with,
            'contains': contains
        }
        return modes[self.header_mode]

    def _explain(self):
        return 'Header row must not contain duplicate values'


class ColumnOrder(Rule):
    """ Fixed header element order Rule """

    def _evaluator(self):
        def starts_with(x):
            ixs = [[y.startswith(z) for z in list(self.definition.keys())].index(True) for y in x]
            return ixs == sorted(ixs)

        def contains(x):
            ixs = [[z in y for z in list(self.definition.keys())].index(True) for y in x]
            return ixs == sorted(ixs)

        modes = {
            'exact': lambda x: x == list(self.definition),
            'startswith': starts_with,
            'contains': contains
        }
        return modes[self.header_mode]

    def _explain(self):
        return 'Header row must explicitly match order of definition'
rumydata row rules  These rules are applied to header rows and are generally not intended to be used directly. These rules ensure that the headers are named as expected, is not missing any names, or containing any extras. # this named tuple is here to allow for setting the default argument without # needing to import the Layout class, which results in a circular import Header Rule No extra header elements Rule No missing header elements Rule No duplicate header elements Rule Fixed header element order Rule