Support importing libraries that have .tgz as extension

schakrava · Oct 12, 2017 · a32d663 · a32d663
1 parent 0a0cf07
commit a32d663
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...porter/src/main/groovy/com/linkedin/python/importer/distribution/SourceDistPackage.groovy b/...porter/src/main/groovy/com/linkedin/python/importer/distribution/SourceDistPackage.groovy
@@ -151,7 +151,7 @@ class SourceDistPackage {
     }
 
     private String getRequiresTextFile() {
-        if (packageFile.absolutePath.contains('.tar.')) {
+        if (packageFile.absolutePath.contains('.tar.') || packageFile.absolutePath.endsWith('.tgz')) {
             return explodeTarForRequiresText()
         } else {
             return explodeZipForRequiresText()
@@ -167,7 +167,6 @@ class SourceDistPackage {
         return ''
     }
 
-
     private String explodeTarForRequiresText() {
         TarArchiveInputStream tarIn = explodeArtifact()
 
@@ -191,7 +190,7 @@ class SourceDistPackage {
         BufferedInputStream inputStream = new BufferedInputStream(fin)
         InputStream compressorInputStream
 
-        if (packageFile.absolutePath.endsWith('.gz')) {
+        if (packageFile.absolutePath.endsWith('.gz') || packageFile.absolutePath.endsWith('.tgz')) {
             compressorInputStream = new GzipCompressorInputStream(inputStream)
         } else if (packageFile.absolutePath.endsWith('.bz2')) {
             compressorInputStream = new BZip2CompressorInputStream(inputStream)