forked from platformio/platformio-core
111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
# Copyright 2014-2016 Ivan Kravets <me@ikravets.com>
|
|
#
|
|
# 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 sys
|
|
|
|
import click
|
|
import requests
|
|
|
|
from platformio import __version__, exception, util
|
|
|
|
|
|
@click.command("upgrade",
|
|
short_help="Upgrade PlatformIO to the latest version")
|
|
def cli():
|
|
last = get_latest_version()
|
|
if __version__ == last:
|
|
return click.secho(
|
|
"You're up-to-date!\nPlatformIO %s is currently the "
|
|
"newest version available." % __version__, fg="green"
|
|
)
|
|
else:
|
|
click.secho("Please wait while upgrading PlatformIO ...",
|
|
fg="yellow")
|
|
|
|
to_develop = False
|
|
try:
|
|
from pkg_resources import parse_version
|
|
to_develop = parse_version(last) < parse_version(__version__)
|
|
except ImportError:
|
|
pass
|
|
|
|
cmds = (
|
|
["pip", "install", "--upgrade",
|
|
"https://github.com/platformio/platformio/archive/develop.zip"
|
|
if to_develop else "platformio"],
|
|
["platformio", "--version"]
|
|
)
|
|
|
|
cmd = None
|
|
r = None
|
|
try:
|
|
for cmd in cmds:
|
|
cmd = [os.path.normpath(sys.executable), "-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("http://docs.platformio.org/en/latest/history.html",
|
|
fg="cyan")
|
|
except Exception as e: # pylint: disable=W0703
|
|
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()
|
|
else:
|
|
raise exception.UpgradeError(
|
|
"\n".join([str(cmd), r['out'], r['err']]))
|
|
finally:
|
|
if r:
|
|
r.close()
|
|
|
|
|
|
def get_latest_version():
|
|
try:
|
|
pkgdata = requests.get(
|
|
"https://pypi.python.org/pypi/platformio/json",
|
|
headers=util.get_request_defheaders()
|
|
).json()
|
|
return pkgdata['info']['version']
|
|
except:
|
|
raise exception.GetLatestVersionError()
|