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/tree/rebuilder.py b/astroid/tree/rebuilder.py
index bb963f38..c45467f8 100644
--- a/astroid/tree/rebuilder.py
+++ b/astroid/tree/rebuilder.py
@@ -873,6 +873,18 @@ class TreeRebuilder3(TreeRebuilder):
     def visit_asyncwith(self, node, parent):
         return self.visit_with(node, parent, constructor=nodes.AsyncWith)
 
+    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