forked from platformio/platformio-core
158 lines
5.1 KiB
Python
158 lines
5.1 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.
|
|
|
|
import os
|
|
import re
|
|
from zipfile import ZipFile
|
|
|
|
import click
|
|
import requests
|
|
|
|
from platformio import VERSION, __version__, exception, util
|
|
from platformio.managers.core import shutdown_piohome_servers
|
|
|
|
|
|
@click.command(
|
|
"upgrade", short_help="Upgrade PlatformIO to the latest version")
|
|
@click.option("--dev", is_flag=True, help="Use development branch")
|
|
def cli(dev):
|
|
if not dev and __version__ == get_latest_version():
|
|
return click.secho(
|
|
"You're up-to-date!\nPlatformIO %s is currently the "
|
|
"newest version available." % __version__,
|
|
fg="green")
|
|
|
|
click.secho("Please wait while upgrading PlatformIO ...", fg="yellow")
|
|
|
|
# kill all PIO Home servers, they block `pioplus` binary
|
|
shutdown_piohome_servers()
|
|
|
|
to_develop = dev or not all(c.isdigit() for c in __version__ if c != ".")
|
|
cmds = (["pip", "install", "--upgrade",
|
|
get_pip_package(to_develop)], ["platformio", "--version"])
|
|
|
|
cmd = None
|
|
r = None
|
|
try:
|
|
for cmd in cmds:
|
|
cmd = [util.get_pythonexe_path(), "-m"] + cmd
|
|
r = None
|
|
r = util.exec_command(cmd)
|
|
|
|
# try pip with disabled cache
|
|
if r['returncode'] != 0 and cmd[2] == "pip":
|
|
cmd.insert(3, "--no-cache-dir")
|
|
r = util.exec_command(cmd)
|
|
|
|
assert r['returncode'] == 0
|
|
assert "version" in r['out']
|
|
actual_version = r['out'].strip().split("version", 1)[1].strip()
|
|
click.secho(
|
|
"PlatformIO has been successfully upgraded to %s" % actual_version,
|
|
fg="green")
|
|
click.echo("Release notes: ", nl=False)
|
|
click.secho(
|
|
"https://docs.platformio.org/en/latest/history.html", fg="cyan")
|
|
except Exception as e: # pylint: disable=broad-except
|
|
if not r:
|
|
raise exception.UpgradeError("\n".join([str(cmd), str(e)]))
|
|
permission_errors = ("permission denied", "not permitted")
|
|
if (any(m in r['err'].lower() for m in permission_errors)
|
|
and "windows" not in util.get_systype()):
|
|
click.secho(
|
|
"""
|
|
-----------------
|
|
Permission denied
|
|
-----------------
|
|
You need the `sudo` permission to install Python packages. Try
|
|
|
|
> sudo pip install -U platformio
|
|
|
|
WARNING! Don't use `sudo` for the rest PlatformIO commands.
|
|
""",
|
|
fg="yellow",
|
|
err=True)
|
|
raise exception.ReturnErrorCode(1)
|
|
else:
|
|
raise exception.UpgradeError("\n".join(
|
|
[str(cmd), r['out'], r['err']]))
|
|
|
|
return True
|
|
|
|
|
|
def get_pip_package(to_develop):
|
|
if not to_develop:
|
|
return "platformio"
|
|
dl_url = ("https://github.com/platformio/"
|
|
"platformio-core/archive/develop.zip")
|
|
cache_dir = util.get_cache_dir()
|
|
if not os.path.isdir(cache_dir):
|
|
os.makedirs(cache_dir)
|
|
pkg_name = os.path.join(cache_dir, "piocoredevelop.zip")
|
|
try:
|
|
with open(pkg_name, "w") as fp:
|
|
r = util.exec_command(["curl", "-fsSL", dl_url],
|
|
stdout=fp,
|
|
universal_newlines=True)
|
|
assert r['returncode'] == 0
|
|
# check ZIP structure
|
|
with ZipFile(pkg_name) as zp:
|
|
assert zp.testzip() is None
|
|
return pkg_name
|
|
except: # pylint: disable=bare-except
|
|
pass
|
|
return dl_url
|
|
|
|
|
|
def get_latest_version():
|
|
try:
|
|
if not str(VERSION[2]).isdigit():
|
|
try:
|
|
return get_develop_latest_version()
|
|
except: # pylint: disable=bare-except
|
|
pass
|
|
return get_pypi_latest_version()
|
|
except:
|
|
raise exception.GetLatestVersionError()
|
|
|
|
|
|
def get_develop_latest_version():
|
|
version = None
|
|
r = requests.get(
|
|
"https://raw.githubusercontent.com/platformio/platformio"
|
|
"/develop/platformio/__init__.py",
|
|
headers=util.get_request_defheaders())
|
|
r.raise_for_status()
|
|
for line in r.text.split("\n"):
|
|
line = line.strip()
|
|
if not line.startswith("VERSION"):
|
|
continue
|
|
match = re.match(r"VERSION\s*=\s*\(([^\)]+)\)", line)
|
|
if not match:
|
|
continue
|
|
version = match.group(1)
|
|
for c in (" ", "'", '"'):
|
|
version = version.replace(c, "")
|
|
version = ".".join(version.split(","))
|
|
assert version
|
|
return version
|
|
|
|
|
|
def get_pypi_latest_version():
|
|
r = requests.get(
|
|
"https://pypi.org/pypi/platformio/json",
|
|
headers=util.get_request_defheaders())
|
|
r.raise_for_status()
|
|
return r.json()['info']['version']
|