Yaltik/hy
2
0
Fork 0
Code Pull Requests Releases Activity

Fix #684

Browse Source
This commit is contained in:
Ryan Gonzalez 2014-11-13 17:49:17 -06:00
parent a6d9a963b5
commit c9362d39c1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
hy/compiler.py
View File

@ -1268,7 +1268,8 @@ class HyASTCompiler(object):
def compile_decorate_expression(self, expr):
expr.pop(0) # with-decorator
fn = self.compile(expr.pop(-1))
if not fn.stmts or not isinstance(fn.stmts[-1], ast.FunctionDef):
if not fn.stmts or not (isinstance(fn.stmts[-1], ast.FunctionDef) or
isinstance(fn.stmts[-1], ast.ClassDef)):
raise HyTypeError(expr, "Decorated a non-function")
decorators, ret = self._compile_collect(expr)
fn.stmts[-1].decorator_list = decorators