add format string support (#365)

Format strings require support for two new nodes, FormattedValue, respectively JoinedStr.

1 files changed, 12 insertions, 0 deletions

diff --git a/astroid/rebuilder.py b/astroid/rebuilder.py
index d80033fe..91774cfd 100644
--- a/astroid/rebuilder.py
+++ b/astroid/rebuilder.py
@@ -869,6 +869,18 @@ class TreeRebuilder3(TreeRebuilder):
     def visit_asyncwith(self, node, parent):
         return self._visit_with(nodes.AsyncWith, node, parent)
 
+    def visit_joinedstr(self, node, parent):
+        newnode = nodes.JoinedStr(node.lineno, node.col_offset, parent)
+        newnode.postinit([self.visit(child, newnode)
+                          for child in node.values])
+        return newnode
+
+    def visit_formattedvalue(self, node, parent):
+        newnode = nodes.FormattedValue(node.lineno, node.col_offset, parent)
+        newnode.postinit(self.visit(node.value, newnode),
+                         node.conversion,
+                         _visit_or_none(node, 'format_spec', self, newnode))
+        return newnode
 
 if sys.version_info >= (3, 0):
     TreeRebuilder = TreeRebuilder3