-
Notifications
You must be signed in to change notification settings - Fork 3
/
PythonSyntax.jl
122 lines (109 loc) · 4.09 KB
/
PythonSyntax.jl
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
module PythonSyntax
using PyCall
using Match
@pyimport ast
include("meta.jl")
include("symbol.jl")
include("ops.jl")
include("function.jl")
include("class.jl")
include("expr.jl")
include("control.jl")
export pyparse, @pysyntax_str, @pymodule_str
function transpile(s::Symbol, t::Vector)
Expr(s, transpile.(t)...)
end
function transpile(t::PyObject)
Tup = ast.pymember(:Tuple)
@match pytypeof(t) begin
ast.Module || ast.Interactive || ast.Expression ||
ast.Suite => transpile(t[:body])
ast.FunctionDef => transpilefn(t, jlident(t[:name]))
ast.AsyncFunctionDef => transpilefn(t, jlident(t[:name])) # FIXME: async
ast.ClassDef => transpilecls(t)
ast.Return => Expr(:return, transpile(t[:value]))
ast.Delete => error("Delete not yet supported.")
ast.Assign => transpileassign(t)
ast.AugAssign => transpileaugassign(t)
ast.For => transpilefor(t)
ast.AsyncFor => transpilefor(t) # FIXME: async
ast.While => transpilewhile(t)
ast.If || ast.IfExp => quote
if $(transpile(t[:test]))
$(transpile(t[:body]))
else
$(transpile(t[:orelse]))
end
end
ast.With => transpilewith(t)
ast.AsyncWith => transpilewith(t) # FIXME: async
ast.Raise => transpileraise(t)
ast.Try => transpiletry(t)
ast.Assert => if t[:msg] == nothing
quote @assert $(transpile(t[:test])) end
else
quote @assert $(transpile(t[:test])) $(transpile(t[:msg])) end
end
# FIXME: add support for Import
ast.Import || ast.ImportFrom => error("Import is not yet supported.")
ast.Global || ast.Nonlocal => quote
global $(Symbol.(t[:names])...)
end
ast.Expr => transpile(t[:value])
ast.Pass => :( nothing )
ast.Break => :( break )
ast.Continue => :( continue )
# Expressions
ast.BoolOp => Expr(jlboolop(t[:op]), transpile.(t[:values])...)
ast.BinOp => Expr(:call, jlop(t[:op]), transpile(t[:left]),
transpile(t[:right]))
ast.UnaryOp => Expr(:call, jlop(t[:op]), transpile(t[:operand]))
ast.Lambda => transpilefn(t)
ast.Dict => :(Dict(zip($(transpile(:vect, t[:keys])),
$(transpile(:vect, t[:values])))))
ast.Set => :(Set($(transpile(:vect, t[:elts]))))
ast.ListComp => transpilecomp(t)
ast.SetComp => :(Set($(transpilecomp(t))))
ast.DictComp => Expr(:call, :Dict,
Expr(:generator,
Expr(:(=>), transpile(t[:key]), transpile(t[:value])),
transpilegen.(t[:generators])...))
ast.GeneratorExp => Expr(:generator,
transpile(t[:elt]),
transpilegen.(t[:generators])...)
ast.Await || ast.Yield || ast.YieldFrom =>
error("Generators are not yet supported.")
ast.Compare => transpilecmp(t)
# FIXME: keyword arguments
ast.Call => transpilecall(t)
ast.Num => t[:n]
ast.Str => t[:s]
ast.Bytes => t[:s]
ast.NameConstant => t[:value]
ast.Ellipsis => error("Ellipsis not yet supported.")
ast.Attribute => Expr(:., transpile(t[:value]),
Expr(:quote, jlident(t[:attr])))
ast.Subscript => Expr(:ref, transpile(t[:value]),
transpileslice(t[:slice])...)
ast.Starred => Expr(:(...), transpile(t[:value]))
ast.Name => jlident(t[:id])
ast.List => transpile(:vect, t[:elts])
Tup => transpile(:tuple, t[:elts])
end
end
transpile(t::Vector) = Expr(:block, transpile.(t)...)
pyparse(s::AbstractString) = simplify(transpile(ast.parse(s)))
macro pysyntax_str(str)
pyparse(str) |> esc
end
macro pymodule_str(str)
lines = split(str, '\n')
name = jlident(lines[1])
expr = pyparse(join(lines[2:end], '\n'))
quote
const $name = Module($(Expr(:quote, name)), true)
eval($name, :( importall Base.Operators ))
eval($name, $(Expr(:quote, expr)))
end |> esc
end
end # module Python Syntax