adafruit-mirror/circup
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add --py flag to specify installation of Python source modules, otherwise use .mpy modules as default. Add a friendly confirmation message upon installation.

Browse source
This commit is contained in:
Nicholas H.Tollervey 2019-10-07 14:40:57 +01:00
parent 0b47132aa6
commit c7173defe3
No known key found for this signature in database
GPG key ID: FD2A04F69841B6FA
1 changed files with 41 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
circup.py
View file

@ -724,7 +724,8 @@ def show(): # pragma: no cover
@main.command()
@click.argument("name")
def install(name): # pragma: no cover
@click.option("--py", is_flag=True)
def install(name, py): # pragma: no cover
"""
Install a named module onto the device. This is a very naive / simple
hacky proof of concept.
@ -745,17 +746,45 @@ def install(name): # pragma: no cover
raise IOError("Could not find a connected Adafruit device.")
library_path = os.path.join(device_path, "lib")
metadata = mod_names[name]
source_path = metadata["path"]
if os.path.isdir(source_path):
target = os.path.basename(os.path.dirname(source_path))
target_path = os.path.join(library_path, target)
# Copy the directory.
shutil.copytree(source_path, target_path)
if py:
# Use Python source for module.
source_path = metadata["path"] # Path to Python source version.
if os.path.isdir(source_path):
target = os.path.basename(os.path.dirname(source_path))
target_path = os.path.join(library_path, target)
# Copy the directory.
shutil.copytree(source_path, target_path)
else:
target = os.path.basename(source_path)
target_path = os.path.join(library_path, target)
# Copy file.
shutil.copyfile(source_path, target_path)
else:
target = os.path.basename(source_path)
target_path = os.path.join(library_path, target)
# Copy file.
shutil.copyfile(source_path, target_path)
print("OK")
# Use pre-compiled mpy modules.
module_name = os.path.basename(metadata["path"]).replace(".py", ".mpy")
if not module_name:
# Must be a directory based module.
module_name = os.path.basename(os.path.dirname(metadata["path"]))
major_version = CPY_VERSION.split(".")[0]
bundle_platform = "{}mpy".format(major_version)
bundle_path = ""
for path, subdirs, files in os.walk(
BUNDLE_DIR.format(bundle_platform)
):
if os.path.basename(path) == "lib":
bundle_path = os.path.join(path, module_name)
if bundle_path:
if os.path.isdir(bundle_path):
target_path = os.path.join(library_path, module_name)
# Copy the directory.
shutil.copytree(bundle_path, target_path)
else:
target = os.path.basename(bundle_path)
target_path = os.path.join(library_path, target)
# Copy file.
shutil.copyfile(bundle_path, target_path)
else:
raise IOError("Cannot find compiled version of module.")
click.echo("Installed '{}'.".format(name))
else:
click.echo("Unknown module named, '{}'.".format(name))