forked from platformio/platformio-core
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
|
# Copyright (C) Ivan Kravets <me@ikravets.com>
|
||
|
|
# See LICENSE for details.
|
||
|
|
|
||
|
|
from os.path import join
|
||
|
|
|
||
|
|
from click import echo, secho, style
|
||
|
|
|
||
|
|
from platformio.exception import (PackageInstalled, UnknownPackage,
|
||
|
|
UnknownPlatform)
|
||
|
|
from platformio.pkgmanager import PackageManager
|
||
|
|
from platformio.util import exec_command, get_source_dir
|
||
|
|
|
||
|
|
|
||
|
|
class PlatformFactory(object):
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def newPlatform(name):
|
||
|
|
clsname = "%sPlatform" % name.title()
|
||
|
|
try:
|
||
|
|
mod = __import__("platformio.platforms." + name.lower(),
|
||
|
|
None, None, [clsname])
|
||
|
|
except ImportError:
|
||
|
|
raise UnknownPlatform(name)
|
||
|
|
|
||
|
|
obj = getattr(mod, clsname)()
|
||
|
|
assert isinstance(obj, BasePlatform)
|
||
|
|
return obj
|
||
|
|
|
||
|
|
|
||
|
|
class BasePlatform(object):
|
||
|
|
|
||
|
|
PACKAGES = {}
|
||
|
|
|
||
|
|
def get_name(self):
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
def install(self, with_packages, without_packages):
|
||
|
|
requirements = []
|
||
|
|
pm = PackageManager(self.get_name())
|
||
|
|
|
||
|
|
upkgs = set(with_packages + without_packages)
|
||
|
|
ppkgs = set(self.PACKAGES.keys())
|
||
|
|
if not upkgs.issubset(ppkgs):
|
||
|
|
raise UnknownPackage(", ".join(upkgs - ppkgs))
|
||
|
|
|
||
|
|
for name, opts in self.PACKAGES.iteritems():
|
||
|
|
if name in without_packages:
|
||
|
|
continue
|
||
|
|
elif name in with_packages or opts["default"]:
|
||
|
|
requirements.append((name, opts["path"]))
|
||
|
|
|
||
|
|
for (name, path) in requirements:
|
||
|
|
echo("Installing %s package:" % style(name, fg="cyan"))
|
||
|
|
try:
|
||
|
|
pm.install(name, path)
|
||
|
|
except PackageInstalled:
|
||
|
|
secho("Already installed", fg="yellow")
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
def after_run(self, result): # pylint: disable=R0201
|
||
|
|
return result
|
||
|
|
|
||
|
|
def run(self, variables, targets):
|
||
|
|
assert isinstance(variables, list)
|
||
|
|
assert isinstance(targets, list)
|
||
|
|
|
||
|
|
if "clean" in targets:
|
||
|
|
targets.remove("clean")
|
||
|
|
targets.append("-c")
|
||
|
|
|
||
|
|
result = exec_command([
|
||
|
|
"scons",
|
||
|
|
"-Q",
|
||
|
|
"-f", join(get_source_dir(), "builder", "main.py")
|
||
|
|
] + variables + targets)
|
||
|
|
|
||
|
|
return self.after_run(result)
|