wip: python imports

src/main/kotlin/app/extractors/PythonExtractor.kt

@@ -14,6 +14,8 @@ class PythonExtractor : ExtractorInterface {
         val evaluator by lazy {
             ExtractorInterface.getLibraryClassifier(LANGUAGE_NAME)
         }
+        val MULTI_IMPORT_TO_LIB =
+                ExtractorInterface.getMultipleImportsToLibraryMap(LANGUAGE_NAME)
     }
 
     override fun extract(files: List<DiffFile>): List<CommitStats> {
@@ -35,7 +37,12 @@ class PythonExtractor : ExtractorInterface {
             }
         }
 
-        return imports.toList()
+        var libraries = imports.map { MULTI_IMPORT_TO_LIB.getOrDefault(it, it) }
+            .filter { !it.endsWith("pb")}.toMutableList()
+        if (libraries.size < imports.size) {
+            libraries.add("protobuf")
+        }
+        return libraries
 
     }
 

src/main/resources/data/imports/python.txt

@@ -0,0 +1,2 @@
+cv2:opencv
+cv:opencv

src/test/kotlin/test/tests/extractors/ExtractorTest.kt

@@ -178,4 +178,14 @@ class ExtractorTest : Spek({
             assertExtractsImport(lib, line2, CppExtractor())
         }
     }
+
+    given("import cv2 or cv") {
+        it("imports opencv") {
+            val lib = "opencv"
+            val line1 = "import cv2"
+            assertExtractsImport(lib, line1, PythonExtractor())
+            val line2 = "import cv"
+            assertExtractsImport(lib, line2, PythonExtractor())
+        }
+    }
 })