File: ListStrAdapter_demo.py

package info (click to toggle)
python-traitsui 8.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 18,232 kB
  • sloc: python: 58,982; makefile: 113
file content (100 lines) | stat: -rw-r--r-- 2,974 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# (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!

"""
Display an editable List of Strings.

The ListStrEditor uses a custom ListStrAdapter to define colors and tooltips.
"""

from traitsui.list_str_adapter import ListStrAdapter
from traits.api import HasTraits, List, Str
from traitsui.api import View, Item, ListStrEditor


# -- The adapter ----------------------------------------------------------

class HeadlinesListAdapter(ListStrAdapter):
    """ Custom adapter for string lists being edited in the ListStrEditor.

    - Style ALL CAPS titles in red.
    - Describe the capitalization in the tooltip.
    """

    def get_text_color(self, object, trait, row):
        try:
            row_data = getattr(object, trait)[row]
        except IndexError:
            return ""

        if row_data.isupper():
            return "red"
        return super().get_text_color(object, trait, row)

    def get_tooltip(self, object, trait, row):
        try:
            row_data = getattr(object, trait)[row]
        except IndexError:
            # if the ListStr is editable, IndexError indicates the empty next slots
            return ""

        if row_data.isupper():
            return "An uppercase headline"
        elif row_data.istitle():
            return "A headline in title case"
        elif row_data.islower():
            return "A fully lowercase headline"
        else:
            return self.tooltip


# -- The class providing the view -------------------------------------

class HeadlinesListDemo(HasTraits):

    headlines = List(Str)

    view = View(
        Item(
            "headlines",
            show_label=False,
            editor=ListStrEditor(
                title="List of headlines (hover for description)",
                adapter=HeadlinesListAdapter(
                    # the default tooltip for generic rows
                    tooltip="A headline"
                ),
                auto_add=True
            ),
            # (QT only) the tooltip shown in the area outside of rows
            # or when the row's tooltip is empty
            tooltip="List of headlines"
        ),
        title='List of headlines',
        width=320,
        height=370,
        resizable=True,
    )


# -- Set up the Demo ------------------------------------------------------

demo = HeadlinesListDemo(
    headlines=[
        "Shark bites man",
        "MAN BITES SHARK",
        "The London Stock Exchange Collapses",
        "she sells sea shells on the sea shore"
    ]
)

# Run the demo (in invoked from the command line):
if __name__ == "__main__":
    demo.configure_traits()