#  Copyright (c) 2007, Enthought, Inc.
#  License: BSD Style.

# instance_editor_selection.py -- Example of an instance editor with
#                                 instance selection

#--[Imports]--------------------------------------------------------------

from traits.api    \
    import HasStrictTraits, Int, Instance, List, Regex, Str
from traitsui.api \
    import View, Item, InstanceEditor

#--[Code]-----------------------------------------------------------------


class Person(HasStrictTraits):
    name = Str()
    age = Int()
    phone = Regex(value='000-0000',
                  regex='\d\d\d[-]\d\d\d\d')

    traits_view = View('name', 'age', 'phone')

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')
]


class Team(HasStrictTraits):

    name = Str()
    captain = Instance(Person)
    roster = List(Person)

    traits_view = View(Item('name'),
                       Item('_'),
                       Item('captain',
                            label='Team Captain',
                            editor=InstanceEditor(name='roster',
                                                  editable=True),
                            style='custom',
                            ),
                       buttons=['OK'])

#--[Example*]-------------------------------------------------------------

if __name__ == '__main__':
    Team(name='Vultures',
         captain=people[0],
         roster=people).configure_traits()
