# (C) Copyright 2004-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!

# -------------------------------------------------------------------------
#  Imports:
# -------------------------------------------------------------------------

from traits.api import HasStrictTraits, Str, Int, Regex, List, Instance

from traitsui.api import View, Group, Item, TableEditor, EnumEditor

from traitsui.table_column import ObjectColumn

from traitsui.table_filter import (
    TableFilter,
    RuleTableFilter,
    RuleFilterTemplate,
    MenuFilterTemplate,
    EvalFilterTemplate,
)

# -------------------------------------------------------------------------
#  'Person' class:
# -------------------------------------------------------------------------


class Person(HasStrictTraits):

    # -------------------------------------------------------------------------
    #  Trait definitions:
    # -------------------------------------------------------------------------

    name = Str()
    age = Int()
    phone = Regex(value='000-0000', regex=r'\d\d\d[-]\d\d\d\d')
    state = Str()

    # -------------------------------------------------------------------------
    #  Traits view definition:
    # -------------------------------------------------------------------------

    traits_view = View(
        'name',
        'age',
        'phone',
        'state',
        title='Create new person',
        width=0.18,
        buttons=['OK', 'Cancel'],
    )


# -------------------------------------------------------------------------
#  Sample data:
# -------------------------------------------------------------------------

people = [
    Person(name='Dave', age=39, phone='555-1212'),
    Person(name='Mike', age=28, phone='555-3526'),
    Person(name='Joe', age=34, phone='555-6943'),
    Person(name='Tom', age=22, phone='555-7586'),
    Person(name='Dick', age=63, phone='555-3895'),
    Person(name='Harry', age=46, phone='555-3285'),
    Person(name='Sally', age=43, phone='555-8797'),
    Person(name='Fields', age=31, phone='555-3547'),
]

# -------------------------------------------------------------------------
#  'AgeFilter' class:
# -------------------------------------------------------------------------


class AgeFilter(TableFilter):

    # -------------------------------------------------------------------------
    #  Trait definitions:
    # -------------------------------------------------------------------------

    name = "Age filter"
    _name = "Age filter"

    age = Int(0)

    # -------------------------------------------------------------------------
    #  Traits view definitions:
    # -------------------------------------------------------------------------

    # filter_view = Group( 'age{Age >=}' )

    # -------------------------------------------------------------------------
    #  Returns whether an object passes the filter or not:
    # -------------------------------------------------------------------------

    def filter(self, person):
        """Returns whether an object passes the filter or not."""
        return person.age >= self.age

    # -------------------------------------------------------------------------
    #  Returns a user readable description of what the filter does:
    # -------------------------------------------------------------------------

    def description(self):
        """Returns a user readable description of what the filter does."""
        return 'Age >= %d' % self.age

    def _age_changed(self, old, new):
        self.name = self.description()
        print('AgeFilter _age_changed', self.name)


# -------------------------------------------------------------------------
#  'NameFilter' class:
# -------------------------------------------------------------------------


class NameFilter(TableFilter):

    # -------------------------------------------------------------------------
    #  Trait definitions:
    # -------------------------------------------------------------------------

    mname = Str()

    # -------------------------------------------------------------------------
    #  Traits view definitions:
    # -------------------------------------------------------------------------

    filter_view = Group('mname{Name contains}')

    # -------------------------------------------------------------------------
    #  Returns whether an object passes the filter or not:
    # -------------------------------------------------------------------------

    def filter(self, person):
        """Returns whether an object passes the filter or not."""
        return person.name.lower().find(self.mname.lower()) >= 0

    # -------------------------------------------------------------------------
    #  Returns a user readable description of what the filter does:
    # -------------------------------------------------------------------------

    def description(self):
        """Returns a user readable description of what the filter does."""
        return "Name contains '%s'" % self.mname


# -------------------------------------------------------------------------
#  Table editor definition:
# -------------------------------------------------------------------------

filters = [
    AgeFilter(age=30),
    NameFilter(mname='d'),
    EvalFilterTemplate,
    MenuFilterTemplate,
    RuleFilterTemplate,
]


def evaluate_value(v):
    print('evaluate_value', v)
    return str(v)


# -------------------------------------------------------------------------
#  'TableTest' class:
# -------------------------------------------------------------------------


class TableTest(HasStrictTraits):

    # -------------------------------------------------------------------------
    #  Trait definitions:
    # -------------------------------------------------------------------------

    # people = Instance( Person )
    people = List(Person)

    # -------------------------------------------------------------------------
    #  Traits view definitions:
    # -------------------------------------------------------------------------

    _valid_states = List(["AL", "AR", "AZ", "AK"])

    _state_editor = EnumEditor(
        name="_valid_states",
        evaluate=evaluate_value,
        object='table_editor_object',
    )

    table_editor = TableEditor(
        columns=[
            ObjectColumn(name='name'),
            ObjectColumn(name='age'),
            ObjectColumn(name='phone'),
            ObjectColumn(name='state', editor=_state_editor),
        ],
        editable=True,
        deletable=True,
        sortable=True,
        sort_model=True,
        show_lines=True,
        orientation='vertical',
        show_column_labels=True,
        edit_view=View(
            ['name', 'age', 'phone', 'state', '|[]'], resizable=True
        ),
        filter=None,
        filters=filters,
        row_factory=Person,
    )

    traits_view = View(
        [Item('people', id='people', editor=table_editor), '|[]<>'],
        title='Table Editor Test',
        id='traitsui.tests.table_editor_test',
        dock='horizontal',
        width=0.4,
        height=0.3,
        resizable=True,
        kind='live',
    )


# -------------------------------------------------------------------------
#  Run the tests:
# -------------------------------------------------------------------------

if __name__ == '__main__':
    tt = TableTest(people=people)
    tt.configure_traits()
    for p in tt.people:
        p.print_traits()
        print('--------------')