(2*) lambda x: 2*x (*2) lambda x: x*2 (*) lambda x, y: x*ySo, can we do the same in Python?
Grammar
The first form,(2*)
, is unambiguous. There is no place in Python where an operator can be legally followed by a close-paren. That works for every binary operator, including boolean and comparison operators. But the grammar is a bit tricky. Without lookahead, how do you make sure that a '('
followed by an expr
followed by a binary operator followed by a ')'
doesn't start parsing as just a parenthesized expression? I couldn't find a way to make this ambiguous without manually hacking up ast.c
. (If you're willing to do that, it's not hard, but you shouldn't be willing to do that.)The second form,
(*2)
looks a lot simpler--no lookahead problems. But consider (-2)
. That's already legal Python! So, does that mean you can't section the +
and -
operators?The third form,
(*)
is the simplest. But it's really tempting to want to be able to do the same with unary operators. Why shouldn't I be able to pass (~)
or (not)
around as a function instead of having to use operator.not_
and operator.neg
? And of course that brings us right back to the problem with +
and -
being ambiguous. (Plus, it makes the compile step a little harder. But that's not a huge deal.)I solved these problems using a horrible hack: sectioned operators are enclosed in parens and colons. This looks hideous, but it did let me get things building so I can play with the idea. Now there's no lookahead needed—a colon inside parens isn't valid for anything else (unless you want to be compatible with my bare lambda hack…). And to resolve the
+
/-
issue, only the binary operators can be sectioned, which also means (: -3* :)
is a SyntaxError
instead of meaning lambda x: -3 * x
. Ick. But, again, it's good enough to play with it.The key grammar change looks like this:
atom: ('(' [yield_expr|testlist_comp] ')' | '(' ':' sectionable_unop ':' ')' | '(' ':' sectionable_binop ':' ')' | '(' ':' expr sectionable_binop ':' ')' | '(' ':' sectionable_binop expr ':' ')' | '[' [testlist_comp] ']' | '{' [dictorsetmaker] '}' | NAME | NUMBER | STRING+ | '...' | 'None' | 'True' | 'False')
What about precedence?
Ignored. It only matters if you want to be able to section expressions made up of multiple operators, like(2+3*)
. Which I don't think you do. For non-trivial cases, there are no readability gains for operator sectioning, and having to think about precedence actually might be a readability cost. If you still don't want to use lambda
, do what you'd do in Haskell and compose (2+)
with (3*)
.AST
For the AST, each of those four productions creates a different node type. Except that you _also_ need separate node types for normal binary operators, comparison operators, and boolean operators, because they have different enums for their operators. So I ended up with 10 new types:UnOpSect
, BinOpSect
, BinOpSectRight
, and BinOpSectLeft
, CmpOpSect
, etc. There's probably a better way to do this.Symbol table
How do you deal with an anonymous argument in the symbol table for the function we're going to generate? You don't want to have to create a whole args structure just to insert a name just so you can refer to it in the compiler. Plus, whatever name you pick could collide with a name in the parent scope, hiding it from alambda
or a comprehension that you define inside the expr
. (Why would you ever do that? Who knows, but it's legal.)This problem must have already been solved. After all, generator expressions have created hidden functions that don't collide any names in the outer scope since they were first created, and in 3.x all comprehensions do that. It's a little tricky to actually get at these hidden functions, but here's one way to do it:
>>> def f(): (i for i in []) >>> f.__code__.co_consts (None, <code object <genexpr> at 0x10bc57a50, file "<stdin>", line 1>, 'z.<locals>.<genexpr>') >>> f.__code__.co_consts[1].co_varnames ('.0', 'i')So, the parameter is named
.0
which isn't legal in a def
or lambda
and can't be referenced. Clever. And once you dig into symtable.c
, you can see that this is handled in a function named symtable_implicit_arg
. So:VISIT(st, expr, e->v.BinOpSectLeft.right); if (!symtable_enter_block(st, binopsect, FunctionBlock, (void *)e, e->lineno, e->col_offset)) VISIT_QUIT(st, 0); if (!symtable_implicit_arg(st, 0)) VISIT_QUIT(st, 0); if (!symtable_exit_block(st, (void *)e)) VISIT_QUIT(st, 0);
Compiler
The compilation works similar tolambda
. Other than sprintf
'ing up a nice name instead of just <lambda>
, and the fact that everything is simpler when there's exactly one argument with no defaults and no keywords, everything is the same except the body, which looks like this:ADDOP_I_IN_SCOPE(c, LOAD_FAST, 0); VISIT_IN_SCOPE(c, expr, e->v.BinOpSectLeft.right); ADDOP_IN_SCOPE(c, binop(c, e->v.BinOpSectLeft.op)); ADDOP_IN_SCOPE(c, RETURN_VALUE); co = assemble(c, 1);I did have to create that
ADDOP_I_IN_SCOPE
macro, but that's trivial.Does it work?
>>> (: *2 :) <function <2> at 0x10bc9f048> >>> (: *2 :).__code__.co_varnames ('.0',) >>> (: *2 :)(23) 46As you can see, I screwed up the name a bit.
More importantly, I screwed up nonlocal references in the symtable. I think I need to visit the argument? Anyway, what happens is this:
>>> a = 23 >>> (: *a :)(23) Traceback (most recent call last): File "<stdin>", line 1, in <module> File "<stdin>", line 1, in <a> SystemError: no locals when loading 'a'But that's much better than the segfault I expected. :)
Is it useful?
Really, most of the obvious use cases for this are already handled by bound methods, likespam.__add__
instead of (spam+)
, and the operator module, like operator.add
instead of (+)
. Is that perfect? No:spam.__add__
isn't as flexible as(spam+)
, because the latter will automatically handle calling its argument's__radd__
when appropriate.- Often, you want to section with literals. Especially with integers. But
0.__add__
is ambiguous between a method on an integer literal or a float literal followed by garbage, and therefore aSyntaxError
, so you need0 .__add__
or(0).__add__
. - For right-sectioning,
spam.__radd__
to mean(+spam)
isn't so bad, butspam.__gt__
to mean(<spam)
is a bit less readable.
(<0)
is all that useful. Most examples I look at, what I really want is something like lambda x: x.attr < 0
. In Haskell I'd probably write that by the rough equivalent of composing operator.attrgetter('attr')
with (<0)
. But, even if you pretend that attribution is an operator (even though it isn't) and add sectioning syntax for it, and you use the @
operator for compose (as was proposed and rejected at least twice during the PEP 465 process and at least once since…), the best you can get is (<0) @ (.attr)
which still doesn't look nearly as readable to me in Python as the lambda
.And, without a compelling use case, I'm not sure it's worth spending more time debugging this, or trying to think of a clever way to make it work without the colons and without lookahead, or coming up with a disambiguating rule for
+
/-
. (It's obviously never going to make it into core…)Anything else worth learning here?
When I was having problems getting the symbol table set up (which I still didn't get right…), I realized there's another way to tackle this: Just stop at the AST, which is the easy part. The result, when run normally, is that any operator-sectioning expression resolves to an empty tuple, which doesn't seem all that useful… but you've got an AST node that you can transform with, say, MacroPy. And converting the meaningless AST node into a validlambda
node in Python is a lot easier to building the symbol table and bytecodes in C. Plus, you don't have to rebuild Python every time you make a change.I don't think this is an argument for adding do-nothing AST structures to the core, of course… but as a strategy for hacking on Python, I may start with that next time around.
View comments