Simplify download function.
This commit is contained in:
parent
993c7ad650
commit
9bcb677012
2 changed files with 18 additions and 11 deletions
|
@ -1,20 +1,27 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import atexit
|
||||
import errno
|
||||
import shutil
|
||||
import tarfile
|
||||
import tempfile
|
||||
import urllib2
|
||||
import os
|
||||
|
||||
|
||||
def download(text, url, filename):
|
||||
directory = tempfile.mkdtemp(prefix='atom-shell-')
|
||||
def tempdir(prefix=''):
|
||||
directory = tempfile.mkdtemp(prefix=prefix)
|
||||
atexit.register(shutil.rmtree, directory)
|
||||
return directory
|
||||
|
||||
web_file = urllib2.urlopen(url)
|
||||
file_size = int(web_file.info().getheaders("Content-Length")[0])
|
||||
downloaded_size = 0
|
||||
block_size = 128
|
||||
|
||||
with open(os.path.join(directory, filename), 'w+') as local_file:
|
||||
def download(text, url, path):
|
||||
with open(path, 'w') as local_file:
|
||||
web_file = urllib2.urlopen(url)
|
||||
file_size = int(web_file.info().getheaders("Content-Length")[0])
|
||||
downloaded_size = 0
|
||||
block_size = 128
|
||||
|
||||
while True:
|
||||
buf = web_file.read(block_size)
|
||||
if not buf:
|
||||
|
@ -29,8 +36,6 @@ def download(text, url, filename):
|
|||
|
||||
print
|
||||
|
||||
return directory
|
||||
|
||||
|
||||
def extract_tarball(tarball_path, member, path):
|
||||
with tarfile.open(tarball_path) as tarball:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue