yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove temp model archive after extraction

Browse Source 
Without this change every time a model is downloaded
it leaks a file in the temp directory.
This commit is contained in:
Marshall Scorcio
2020-04-25 22:06:17 -07:00
parent 0366a4c83d
commit ba9999ac22
1 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
spleeter/model/provider/github.py
View File

@@ -16,6 +16,7 @@
import hashlib import hashlib
import tarfile import tarfile
import os
from tempfile import NamedTemporaryFile from tempfile import NamedTemporaryFile
@@ -96,6 +97,7 @@ class GithubModelProvider(ModelProvider):
with requests.get(url, stream=True) as response: with requests.get(url, stream=True) as response:
response.raise_for_status() response.raise_for_status()
archive = NamedTemporaryFile(delete=False) archive = NamedTemporaryFile(delete=False)
try:
with archive as stream: with archive as stream:
# Note: check for chunk size parameters ? # Note: check for chunk size parameters ?
for chunk in response.iter_content(chunk_size=8192): for chunk in response.iter_content(chunk_size=8192):
@@ -107,4 +109,6 @@ class GithubModelProvider(ModelProvider):
get_logger().info('Extracting downloaded %s archive', name) get_logger().info('Extracting downloaded %s archive', name)
with tarfile.open(name=archive.name) as tar: with tarfile.open(name=archive.name) as tar:
tar.extractall(path=path) tar.extractall(path=path)
finally:
os.unlink(archive.name)
get_logger().info('%s model file(s) extracted', name) get_logger().info('%s model file(s) extracted', name)