forked from platformio/platformio-core
45e75f7473
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
119 lines
4.0 KiB
Python
119 lines
4.0 KiB
Python
# Copyright (c) 2014-present PlatformIO <contact@platformio.org>
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from email.utils import parsedate_tz
|
|
from math import ceil
|
|
from os.path import getsize, join
|
|
from time import mktime
|
|
|
|
import click
|
|
import requests
|
|
|
|
from platformio import app, util
|
|
from platformio.exception import (FDSHASumMismatch, FDSizeMismatch,
|
|
FDUnrecognizedStatusCode)
|
|
|
|
|
|
class FileDownloader(object):
|
|
|
|
CHUNK_SIZE = 1024
|
|
|
|
def __init__(self, url, dest_dir=None):
|
|
# make connection
|
|
self._request = requests.get(
|
|
url, stream=True, headers=util.get_request_defheaders())
|
|
if self._request.status_code != 200:
|
|
raise FDUnrecognizedStatusCode(self._request.status_code, url)
|
|
|
|
disposition = self._request.headers.get("content-disposition")
|
|
if disposition and "filename=" in disposition:
|
|
self._fname = disposition[disposition.index("filename=") +
|
|
9:].replace('"', "").replace("'", "")
|
|
self._fname = self._fname.encode("utf8")
|
|
else:
|
|
self._fname = url.split("/")[-1]
|
|
|
|
self._progressbar = None
|
|
self._destination = self._fname
|
|
if dest_dir:
|
|
self.set_destination(join(dest_dir, self._fname))
|
|
|
|
def set_destination(self, destination):
|
|
self._destination = destination
|
|
|
|
def get_filepath(self):
|
|
return self._destination
|
|
|
|
def get_lmtime(self):
|
|
return self._request.headers.get("last-modified")
|
|
|
|
def get_size(self):
|
|
if "content-length" not in self._request.headers:
|
|
return -1
|
|
return int(self._request.headers['content-length'])
|
|
|
|
def start(self):
|
|
itercontent = self._request.iter_content(chunk_size=self.CHUNK_SIZE)
|
|
f = open(self._destination, "wb")
|
|
|
|
if app.is_disabled_progressbar() or self.get_size() == -1:
|
|
click.echo("Downloading...")
|
|
for chunk in itercontent:
|
|
if chunk:
|
|
f.write(chunk)
|
|
else:
|
|
chunks = int(ceil(self.get_size() / float(self.CHUNK_SIZE)))
|
|
with click.progressbar(length=chunks, label="Downloading") as pb:
|
|
for _ in pb:
|
|
f.write(next(itercontent))
|
|
f.close()
|
|
self._request.close()
|
|
|
|
if self.get_lmtime():
|
|
self._preserve_filemtime(self.get_lmtime())
|
|
|
|
def verify(self, sha1=None):
|
|
_dlsize = getsize(self._destination)
|
|
if self.get_size() != -1 and _dlsize != self.get_size():
|
|
raise FDSizeMismatch(_dlsize, self._fname, self.get_size())
|
|
|
|
if not sha1:
|
|
return
|
|
|
|
dlsha1 = None
|
|
try:
|
|
result = util.exec_command(["sha1sum", self._destination])
|
|
dlsha1 = result['out']
|
|
except (OSError, ValueError):
|
|
try:
|
|
result = util.exec_command(
|
|
["shasum", "-a", "1", self._destination])
|
|
dlsha1 = result['out']
|
|
except (OSError, ValueError):
|
|
pass
|
|
|
|
if dlsha1:
|
|
dlsha1 = dlsha1[1:41] if dlsha1.startswith("\\") else dlsha1[:40]
|
|
if sha1 != dlsha1:
|
|
raise FDSHASumMismatch(dlsha1, self._fname, sha1)
|
|
|
|
def _preserve_filemtime(self, lmdate):
|
|
timedata = parsedate_tz(lmdate)
|
|
lmtime = mktime(timedata[:9])
|
|
util.change_filemtime(self._destination, lmtime)
|
|
|
|
def __del__(self):
|
|
if self._request:
|
|
self._request.close()
|