From a2ac0c3035264c217d47c54b2957b2534e2f23b7 Mon Sep 17 00:00:00 2001
From: Steven Kryskalla <skryskalla@gmail.com>
Date: Thu, 23 Feb 2012 12:20:44 -0800
Subject: [PATCH] add warning code W901 to mccabe

---
 flake8/mccabe.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/flake8/mccabe.py b/flake8/mccabe.py
index 9f731e2..c41334c 100644
--- a/flake8/mccabe.py
+++ b/flake8/mccabe.py
@@ -13,6 +13,8 @@ import optparse
 import sys
 from collections import defaultdict
 
+WARNING_CODE = "W901"
+
 
 class ASTVisitor:
 
@@ -63,8 +65,10 @@ class PathNode:
 
 
 class PathGraph:
-    def __init__(self, name):
+    def __init__(self, name, entity, lineno):
         self.name = name
+        self.entity = entity
+        self.lineno = lineno
         self.nodes = defaultdict(list)
 
     def connect(self, n1, n2):
@@ -122,7 +126,7 @@ class PathGraphingAstVisitor(ASTVisitor):
             self.graph.connect(pathnode, bottom)
             self.tail = bottom
         else:
-            self.graph = PathGraph(name)
+            self.graph = PathGraph(name, entity, node.lineno)
             pathnode = PathNode(name)
             self.tail = pathnode
             self.default(node)
@@ -163,7 +167,7 @@ class PathGraphingAstVisitor(ASTVisitor):
 
         if self.graph is None:
             # global loop
-            self.graph = PathGraph(name)
+            self.graph = PathGraph(name, name, node.lineno)
             pathnode = PathNode(name)
             self.tail = pathnode
             self.default(node)
@@ -239,8 +243,13 @@ def get_code_complexity(code, min=7, filename='stdin'):
             # ?
             continue
         if graph.complexity() >= min:
-            msg = '%s:%s is too complex (%d)' % (filename,
-                    graph.name, graph.complexity())
+            msg = '%s:%d:1 %s %r is too complex (%d)' % (
+                filename,
+                graph.lineno,
+                WARNING_CODE,
+                graph.entity,
+                graph.complexity(),
+            )
             complex.append(msg)
 
     if len(complex) == 0: