From 7f0ecddfb2505d89e72850b935ec4e3b24bf3131 Mon Sep 17 00:00:00 2001
From: Tim 'mithro' Ansell <me@mith.ro>
Date: Mon, 6 Apr 2020 17:45:55 -0700
Subject: [PATCH] Use shutil.unpack_archive.

---
 litex_setup.py | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/litex_setup.py b/litex_setup.py
index 61a556a8b..fa33fb992 100755
--- a/litex_setup.py
+++ b/litex_setup.py
@@ -3,6 +3,7 @@
 import os
 import sys
 import subprocess
+import shutil
 from collections import OrderedDict
 
 import urllib.request
@@ -59,14 +60,7 @@ def sifive_riscv_download():
         print("Using existing file", fn)
 
     print("Extracting", fn)
-    if fn.endswith(".tar.gz"):
-        import tarfile
-        with tarfile.open(fn) as t:
-            t.extractall()
-    elif fn.endswith(".zip"):
-        import zipfile
-        with zipfile.open(fn) as z:
-            z.extractall()
+    shutil.unpack_archive(fn)
 
     if "--user" in sys.argv[1:] and not is_windows:
         print("Linking compiler into ~/.local/bin")