Skip to content

Some modifications so PyDev doesn't complain: #9

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
58 changes: 31 additions & 27 deletions codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,8 +145,8 @@ def visit_Assert(self, node):
self.write('assert ')
self.visit(node.test)
if node.msg is not None:
self.write(', ')
self.visit(node.msg)
self.write(', ')
self.visit(node.msg)

def visit_Assign(self, node):
self.newline(node)
Expand Down Expand Up @@ -414,19 +414,19 @@ def visit_Tuple(self, node):
self.visit(item)
self.write(idx and ')' or ',)')

def sequence_visit(left, right):
def visit(self, node):
self.write(left)
for idx, item in enumerate(node.elts):
if idx:
self.write(', ')
self.visit(item)
self.write(right)
return visit
def __sequence_visit(self, left, right, node):
self.write(left)
for idx, item in enumerate(node.elts):
if idx:
self.write(', ')
self.visit(item)
self.write(right)

def visit_List(self, node):
self.__sequence_visit('[', ']', node)

visit_List = sequence_visit('[', ']')
visit_Set = sequence_visit('{', '}')
del sequence_visit
def visit_Set(self, node):
self.__sequence_visit('{', '}', node)

def visit_Dict(self, node):
self.write('{')
Expand All @@ -439,9 +439,11 @@ def visit_Dict(self, node):
self.write('}')

def visit_BinOp(self, node):
self.write('(')
self.visit(node.left)
self.write(' %s ' % BINOP_SYMBOLS[type(node.op)])
self.visit(node.right)
self.write(')')

def visit_BoolOp(self, node):
self.write('(')
Expand Down Expand Up @@ -504,19 +506,21 @@ def visit_Lambda(self, node):
def visit_Ellipsis(self, node):
self.write('Ellipsis')

def generator_visit(left, right):
def visit(self, node):
self.write(left)
self.visit(node.elt)
for comprehension in node.generators:
self.visit(comprehension)
self.write(right)
return visit

visit_ListComp = generator_visit('[', ']')
visit_GeneratorExp = generator_visit('(', ')')
visit_SetComp = generator_visit('{', '}')
del generator_visit
def __generator_visit(self, left, right, node):
self.write(left)
self.visit(node.elt)
for comprehension in node.generators:
self.visit(comprehension)
self.write(right)

def visit_ListComp(self, node):
self.__generator_visit('[', ']', node)

def visit_GeneratorExp(self, node):
self.__generator_visit('(', ')', node)

def visit_SetComp(self, node):
self.__generator_visit('{', '}', node)

def visit_DictComp(self, node):
self.write('{')
Expand Down