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

from traits.api \
    import *

from traitsui.api \
    import *

from traitsui.menu \
    import *

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


class Person(Handler):

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

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

    #-------------------------------------------------------------------------
    #  Handles the 'Annoy' button being clicked:
    #-------------------------------------------------------------------------

    def _annoy_clicked(self, info):
        self.edit_traits(view=View(title='Annoying',
                                   kind='modal',
                                   buttons=['OK']))

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

if __name__ == '__main__':
    AnnoyButton = Action(name='Annoy',
                         tooltip='Click me to be annoyed',
                         enabled_when='age >= 40')

    person = Person(name='Bill', age=42, phone='555-1212')

    fields = Group('name', 'age', 'phone', 'notes~')

    person.notes = ("Should have 6 standard 'live' buttons: Undo, Redo, "
                    "Revert, OK, Cancel, Help")
    person.configure_traits(view=View(fields,
                                      kind='livemodal',
                                      buttons=LiveButtons))

    person.notes = ("Should have 5 standard 'modal' buttons: Apply, Revert, "
                    "OK, Cancel, Help")
    person.configure_traits(view=View(fields,
                                      buttons=ModalButtons))

    person.notes = "Should have 2 standard buttons: OK, Cancel"
    person.configure_traits(
        view=View(fields,
                  buttons=[OKButton, CancelButton]))

    person.notes = "Should have 1 standard button: OK (enabled when age >= 40)"
    person.configure_traits(
        view=View(fields,
                  buttons=[Action(name='OK',
                                  enabled_when='age >= 40')]))

    person.notes = "Should have 1 standard button: OK (visible when age >= 40)"
    person.configure_traits(
        view=View(fields,
                  buttons=[Action(name='OK',
                                  visible_when='age >= 40')]))

    person.notes = ("Should have 2 standard buttons: OK, Help (defined when "
                    "age >= 50)")
    person.configure_traits(
        view=View(fields,
                  buttons=['OK', Action(name='Help',
                                        defined_when='age >= 50')]))

    person.notes = (
        "Should have 1 user and 5 standard buttons: Annoy (enabled "
        "when age >= 40), Apply, Revert, OK, Cancel, Help")
    person.configure_traits(
        view=View(fields,
                  buttons=[AnnoyButton] + ModalButtons))
