Any simpler way to do this

Virgil Dupras hardcoded.software at gmail.com
Fri Dec 7 04:04:02 EST 2007


On Dec 7, 9:37 am, Lars Johansen <l... at larsjohansen.org> wrote:
> I have a function that looks like this:
>
> def Chooser(color):
>
>         if color == "RED":
>                 x = term.RED
>         elif color == "BLUE":
>                 x = term.BLUE
>         elif color == "GREEN":
>                 x = term.GREEN
>         elif color == "YELLOW":
>                 x = term.YELLOW
>         elif color == "CYAN":
>                 x = term.CYAN
>         elif color == "MAGENTA":
>                 x = term.MAGENTA
>         return x
>
> Wouldn there been easier if I could just skip all the "*if's" and just
> "return term.color", however this gives a syntax error, are there any
> clever way to do this ?
> --
> Lars Johansen <l... at larsjohansen.org>

"return getattr(term, color)" should do the trick.



More information about the Python-list mailing list