summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRichard Ipsum <richard.ipsum@codethink.co.uk>2014-10-09 16:13:13 +0100
committerRichard Ipsum <richard.ipsum@codethink.co.uk>2014-11-04 12:40:54 +0000
commitbdde963ea7685feab9740dd21cc6e2ace3d0248e (patch)
tree8deab2d4a5c9639de3a78e2b6b1eba717408ac53
parent88a3d0787058fdb8820547d55c63bc68eee2db51 (diff)
downloadimport-bdde963ea7685feab9740dd21cc6e2ace3d0248e.tar.gz
Fix get_compresion
-rwxr-xr-ximport/pip.to_lorry8
1 files changed, 3 insertions, 5 deletions
diff --git a/import/pip.to_lorry b/import/pip.to_lorry
index 7dd1678..214f840 100755
--- a/import/pip.to_lorry
+++ b/import/pip.to_lorry
@@ -67,11 +67,9 @@ def get_compression(url):
'tar.bz2': bzip, 'tbz2': bzip,
'tar.lzma': lzma, 'tar.xz': lzma, 'tlz': lzma, 'txz': lzma}
- ext = '.'.join(url.split('.')[-2:])
- if ext in m: return m[ext]
-
- ext = url.split('.')[-1:]
- if ext in m: return m[ext]
+ for x in [1, 2]:
+ ext = '.'.join(url.split('.')[-x:])
+ if ext in m: return m[ext]
return None