Skip to content

Commit

Permalink
Update copy_marlin_variant_to_framework.py
Browse files Browse the repository at this point in the history
  • Loading branch information
thinkyhead committed Apr 29, 2021
1 parent 070412a commit a5dc42c
Showing 1 changed file with 14 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -1,7 +1,19 @@
import os,shutil
from SCons.Script import DefaultEnvironment
from platformio import util
from platformio.managers.package import PackageManager
try:
# PIO < 4.4
from platformio.managers.package import PackageManager
except ImportError:
# PIO >= 4.4
from platformio.package.meta import PackageSpec as PackageManager

def parse_pkg_uri(spec):
if PackageManager.__name__ == 'PackageSpec':
return PackageManager(spec).name
else:
name, _, _ = PackageManager.parse_pkg_uri(spec)
return name

def copytree(src, dst, symlinks=False, ignore=None):
for item in os.listdir(src):
Expand All @@ -26,7 +38,7 @@ def copytree(src, dst, symlinks=False, ignore=None):
if len(platform_packages) == 0:
platform_name = framewords[platform.__class__.__name__]
else:
platform_name, _, _ = PackageManager.parse_pkg_uri(platform_packages[0])
platform_name = parse_pkg_uri(platform_packages[0])

FRAMEWORK_DIR = platform.get_package_dir(platform_name)
assert os.path.isdir(FRAMEWORK_DIR)
Expand Down

0 comments on commit a5dc42c

Please sign in to comment.