Partially added docstrings. Not working

burakozturkdot · Jan 10, 2017 · ace05e1 · ace05e1
1 parent 93fd913
commit ace05e1
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 1 deletion.
diff --git a/pythonlua/context.py b/pythonlua/context.py
@@ -12,6 +12,7 @@ def __init__(self, values=None):
             "locals": SymbolsStack(),
             "globals": SymbolsStack(),  # Not working yet
             "loop_label_name": "",
+            "docstring": False,
         }
 
         self.ctx_stack = [values]

diff --git a/pythonlua/nodevisitor.py b/pythonlua/nodevisitor.py
@@ -476,11 +476,13 @@ def visit_Starred(self, node):
         self.emit(line)
 
     def visit_Str(self, node):
-        """Visit str"""        
+        """Visit str"""
         value = node.s
         if value.startswith(NodeVisitor.LUACODE):
             value = value[len(NodeVisitor.LUACODE):]
             self.emit(value)
+        elif self.context.last()["docstring"]:
+            self.emit('--[[ {} ]]'.format(node.s))
         else:
             self.emit('"{}"'.format(node.s))
 

diff --git a/tests/comments.py b/tests/comments.py
@@ -0,0 +1,5 @@
+"""Documentation comments test"""
+
+name = "John Parrish"
+
+print(name)
diff --git a/tests/comments.py.expected b/tests/comments.py.expected
@@ -0,0 +1 @@
+John Parrish