Added cyclomatic complexities for python subunits

Author: isc-cge

cls/TestCoverage/Data/CodeSubUnit.cls

@@ -9,6 +9,9 @@ Property Mask As TestCoverage.DataType.Bitstring;
 /// Cyclomatic complexity of the code subunit
 Property Complexity As %Integer [ InitialExpression = 1 ];
 
+/// 1 if it's a python class method, 0 if not
+Property IsPythonMethod As %Boolean;
+
 Method UpdateComplexity() As %Status
 {
 	Quit $$$OK
@@ -26,6 +29,9 @@ Storage Default
 <Value name="3">
 <Value>Complexity</Value>
 </Value>
+<Value name="4">
+<Value>IsPythonMethod</Value>
+</Value>
 </Data>
 <DataLocation>{%%PARENT}("SubUnits")</DataLocation>
 <DefaultData>CodeSubUnitDefaultData</DefaultData>
@@ -36,4 +42,3 @@ Storage Default
 }
 
 }
-

cls/TestCoverage/Data/CodeUnit.cls

@@ -163,11 +163,6 @@ ClassMethod GetCurrentByName(pInternalName As %String, pSourceNamespace As %Stri
 					Set $Bit(tMethodMask,j) = 1
 				}
 				Set tMethodSignature = $list(pDocumentText, tStartLine)
-				Set tSubUnit = ##class(TestCoverage.Data.CodeSubUnit.Method).%New()
-				Set tSubUnit.Name = tMethod
-				Set tSubUnit.DisplaySignature = tMethodSignature
-				Set tSubUnit.Mask = tMethodMask
-				Do pCodeUnit.SubUnits.Insert(tSubUnit)
 			}
 		}
 		Else {
@@ -194,6 +189,8 @@ ClassMethod GetCurrentByName(pInternalName As %String, pSourceNamespace As %Stri
 						Set tSubUnit.Name = tMethod
 						Set tSubUnit.DisplaySignature = tMethodSignature
 						Set tSubUnit.Mask = tMethodMask
+						set NormalizedSignature =  $zconvert($zstrip(tMethodSignature, "*W"), "l")
+						set tSubUnit.IsPythonMethod = (NormalizedSignature [ "[language=python]") 
 						Do pCodeUnit.SubUnits.Insert(tSubUnit)
 						Set tMethod = ""
 						Set tMethodSignature = ""
@@ -506,15 +503,28 @@ Method UpdateComplexity() As %Status
 		If (..Type '= "CLS") {
 			Quit
 		}
-		
+
+		// python methods
+		If (##class(TestCoverage.Manager).HasPython(..Name)) {
+			do ##class(TestCoverage.Data.CodeUnit).GetCurrentByName(..Name _ ".PY", , .pPyCodeUnit, )
+			set tDocumentText = pPyCodeUnit.Lines.Serialize()
+			set tMethodComplexities = ..GetPythonComplexities(tDocumentText)
+		}
+
 		Set tKey = ""
 		For {
 			Set tSubUnit = ..SubUnits.GetNext(.tKey)
 			If (tKey = "") {
 				Quit
 			}
-			$$$ThrowOnError(tSubUnit.UpdateComplexity())
+			If (tSubUnit.IsPythonMethod) {
+				set tSubUnit.Complexity = tMethodComplexities."__getitem__"(tSubUnit.Name)
+				$$$ThrowOnError(tSubUnit.%Save(0))
+			} Else {
+				$$$ThrowOnError(tSubUnit.UpdateComplexity())
+			}
 		}
+
 
 		$$$ThrowOnError(..%Save())
 	} Catch e {
@@ -523,6 +533,20 @@ Method UpdateComplexity() As %Status
 	Quit tSC
 }
 
+ClassMethod GetPythonComplexities(pDocumentText) [ Language = python ]
+{
+    from radon.complexity import cc_visit
+    import iris
+    source_lines = iris.cls('%SYS.Python').ToList(pDocumentText)
+    source_code = "\n".join(source_lines) 
+    visitor = cc_visit(source_code)
+    class_info = visitor[0]
+    method_complexities = {}
+    for method in class_info.methods:
+        method_complexities[method.name] = method.complexity 
+    return method_complexities
+}
+
 Method GetMethodOffset(pAbsoluteLine As %Integer, Output pMethod As %String, Output pOffset As %Integer)
 {
 }

cls/TestCoverage/Utils.cls

@@ -122,7 +122,6 @@ ClassMethod Snapshot(pIntRoutines As %List, Output pRelevantRoutines As %List =
 					$$$ThrowOnError(##class(TestCoverage.Data.CodeUnit).GetCurrentByName(tName_".PY",,.tPyCodeUnit))
 
 					// update the executable lines for the .cls file's python
-					set ^IRIS.TEMPCG($i(^IRIS.TEMPCG)) = "Calling update executable lines on " _ tName 
 					$$$ThrowOnError(tCodeUnit.UpdatePyExecutableLines(tName, .tPyCodeUnit))
 
 				} ElseIf '$BitCount(tCodeUnit.ExecutableLines,1) {

requirements.txt

@@ -0,0 +1 @@
+radon==6.*