diff options
| -rw-r--r-- | TurtleArt/tapalette.py | 8 | ||||
| -rw-r--r-- | pysamples/uturn.py | 8 |
2 files changed, 13 insertions, 3 deletions
diff --git a/TurtleArt/tapalette.py b/TurtleArt/tapalette.py index 43a5d25..e34df21 100644 --- a/TurtleArt/tapalette.py +++ b/TurtleArt/tapalette.py @@ -165,6 +165,14 @@ def make_palette(palette_name, colors=None, help_string=None): return palette +def palette_name_to_index(palette_name): + ''' Find the index associated with palette_name. ''' + if palette_name in palette_names: + return palette_names.index(palette_name) + else: + return None + + class Block(): """ a class for defining new block primitives """ diff --git a/pysamples/uturn.py b/pysamples/uturn.py index 40d0a00..cd1b18e 100644 --- a/pysamples/uturn.py +++ b/pysamples/uturn.py @@ -24,19 +24,21 @@ def myblock(lc, arg): ''' Add a uturn block to the 'turtle' palette ''' - from TurtleArt.tapalette import make_palette + from TurtleArt.tapalette import make_palette, palette_name_to_index from TurtleArt.talogo import primitive_dictionary from gettext import gettext as _ - palette = make_palette('turtle') + palette = make_palette('turtle') palette.add_block('uturn', style='basic-style-extended-vertical', label=_('uturn'), prim_name='uturn', help_string=_('make a uturn')) + lc.tw.lc.def_prim('uturn', 0, lambda self: primitive_dictionary['set']( 'heading', lc.tw.canvas.seth, lc.tw.canvas.heading + 180)) - lc.tw.show_toolbar_palette(0, regenerate=True) + lc.tw.show_toolbar_palette(palette_name_to_index('turtle'), + regenerate=True) |