this should work now

Author: tomvanmele

src/compas_rhino/install.py

@@ -19,17 +19,21 @@
 ]
 
 
-def install(version=None, packages=None):
+def install(version=None, packages=None, clean=False):
     """Install COMPAS for Rhino.
 
     Parameters
     ----------
-    version : {'5.0', '6.0', '7.0'}, optional
+    version : {'5.0', '6.0', '7.0', '8.0'}, optional
         The version number of Rhino.
         Default is ``'6.0'``.
     packages : list of str, optional
         List of packages to install or None to use default package list.
-        Default is ``['compas', 'compas_rhino', 'compas_ghpython']``.
+        Default is the result of ``installable_rhino_packages``,
+        which collects all installable packages in the current environment.
+    clean : bool, optional
+        If ``True``, this will clean up the entire scripts folder and remove
+        also existing symlinks that are not importable in the current environment.
 
     Examples
     --------
@@ -44,7 +48,7 @@ def install(version=None, packages=None):
 
     """
 
-    if version not in ('5.0', '6.0', '7.0'):
+    if version not in ('5.0', '6.0', '7.0', '8.0'):
         version = '6.0'
 
     # We install COMPAS packages in the scripts folder
@@ -70,25 +74,29 @@ def install(version=None, packages=None):
         try:
             importlib.import_module(name)
         except ImportError:
-            symlink_path = os.path.join(scripts_path, name)
-            symlinks_to_uninstall.append(dict(name=name, link=symlink_path))
+            path = os.path.join(scripts_path, name)
+            symlinks_to_uninstall.append(dict(name=name, link=path))
             packages.remove(name)
 
-    # check all the compas packages/folders in the scripts directory
-    # if one of them is not importable from the current env
-    # it should be listed for un-installation
-    # otherwise it should be added to the list of packages to install
-    # if it wasn't already in there
+    # Also remove all broken symlinks from the scripts folder
+    # because ... they're broken!
+    # If it is an actual folder or a file, leave it alone
+    # because probably someone put it there on purpose.
     for name in os.listdir(scripts_path):
-        if name.startswith('compas') and not name.endswith('.py'):
-            try:
-                importlib.import_module(name)
-            except ImportError:
-                symlink_path = os.path.join(scripts_path, name)
-                symlinks_to_uninstall.append(dict(name=name, link=symlink_path))
+        path = os.path.join(scripts_path, name)
+        if os.path.islink(path):
+            if not os.path.exists(path):
+                symlinks_to_uninstall.append(dict(name=name, link=path))
             else:
-                if name not in packages:
-                    packages.append(name)
+                if clean:
+                    try:
+                        importlib.import_module(name)
+                    except ImportError:
+                        path = os.path.join(scripts_path, name)
+                        symlinks_to_uninstall.append(dict(name=name, link=path))
+                    else:
+                        if name not in packages:
+                            packages.append(name)
 
     # add all of the packages in the list of installable packages
     # to the list of symlinks to uninstall
@@ -314,9 +322,10 @@ def _filter_installable_packages(version, packages):
 
     parser = argparse.ArgumentParser()
 
-    parser.add_argument('-v', '--version', choices=['5.0', '6.0', '7.0'], default='6.0', help="The version of Rhino to install the packages in.")
+    parser.add_argument('-v', '--version', choices=['5.0', '6.0', '7.0', '8.0'], default='6.0', help="The version of Rhino to install the packages in.")
     parser.add_argument('-p', '--packages', nargs='+', help="The packages to install.")
+    parser.add_argument('--clean', dest='clean', default=False, action='store_true')
 
     args = parser.parse_args()
 
-    install(version=args.version, packages=args.packages)
+    install(version=args.version, packages=args.packages, clean=args.clean)

src/compas_rhino/install_plugin.py

@@ -17,7 +17,7 @@ def install_plugin(plugin, version=None):
     ----------
     plugin : str
         The path to the plugin folder.
-    version : {'5.0', '6.0', '7.0'}, optional
+    version : {'5.0', '6.0', '7.0', '8.0'}, optional
         The version of Rhino for which the plugin should be installed.
         Default is ``'6.0'``.
 
@@ -58,14 +58,29 @@ def install_plugin(plugin, version=None):
         python -m compas_rhino.install_plugin -v 7.0 ui/Rhino/XXX
 
     """
-    if version not in ('5.0', '6.0', '7.0'):
+    if version not in ('5.0', '6.0', '7.0', '8.0'):
         version = '6.0'
 
     if not os.path.isdir(plugin):
         raise Exception('Cannot find the plugin: {}'.format(plugin))
 
     plugin_dir = os.path.abspath(plugin)
 
+    # clean up the plugin directory
+
+    # # Also remove all broken symlinks
+    # # because ... they're broken!
+    # broken = []
+    # for name in os.listdir(plugin_dir):
+    #     path = os.path.join(plugin_dir, name)
+    #     if os.path.islink(path):
+    #         if not os.path.exists(path):
+    #             broken.append(path)
+    # if broken:
+    #     pass
+
+    # proceed with the installation
+
     plugin_path, plugin_name = os.path.split(plugin_dir)
     if not plugin_path:
         plugin_path = os.getcwd()
@@ -98,15 +113,13 @@ def install_plugin(plugin, version=None):
     source = plugin_dir
     destination = os.path.join(python_plugins_path, plugin_fullname)
 
-    print('Installing PlugIn {} to Rhino PythonPlugIns.'.format(plugin_name))
+    print('\nInstalling PlugIn {} to Rhino PythonPlugIns.'.format(plugin_name))
 
     remove_symlinks([destination])
     create_symlinks([(source, destination)])
 
-    print()
-    print('PlugIn {} Installed.'.format(plugin_name))
-    print()
-    print('Restart Rhino and open the Python editor at least once to make it available.')
+    print('\nPlugIn {} Installed.'.format(plugin_name))
+    print('\nRestart Rhino and open the Python editor at least once to make it available.')
 
 
 # ==============================================================================
@@ -118,10 +131,10 @@ def install_plugin(plugin, version=None):
     import argparse
 
     parser = argparse.ArgumentParser(
-        description='COMPAS Rhino PLugin Installation command-line utility.')
+        description='COMPAS Rhino Plugin Installation command-line utility.')
 
     parser.add_argument('plugin', help="The path to the plugin directory.")
-    parser.add_argument('-v', '--version', choices=['5.0', '6.0', '7.0'], default='6.0', help="The version of Rhino.")
+    parser.add_argument('-v', '--version', choices=['5.0', '6.0', '7.0', '8.0'], default='6.0', help="The version of Rhino.")
 
     args = parser.parse_args()
 

src/compas_rhino/uninstall.py

@@ -23,7 +23,7 @@ def uninstall(version=None, packages=None):
 
     Parameters
     ----------
-    version : {'5.0', '6.0', '7.0'}, optional
+    version : {'5.0', '6.0', '7.0', '8.0'}, optional
         The version number of Rhino.
         Default is ``'6.0'``.
     packages : list of str, optional
@@ -42,7 +42,7 @@ def uninstall(version=None, packages=None):
         python -m compas_rhino.uninstall -v 6.0
 
     """
-    if version not in ('5.0', '6.0', '7.0'):
+    if version not in ('5.0', '6.0', '7.0', '8.0'):
         version = '6.0'
 
     # We install COMPAS packages in the scripts folder
@@ -52,18 +52,17 @@ def uninstall(version=None, packages=None):
     # This is for old installs
     ipylib_path = compas_rhino._get_ironpython_lib_path(version)
 
-    # If no specific packages are provided for uninstall
-    # everything should be removed,
-    # and not just the default packages
-    # or the packages that the bootstrapper is aware of.
-    if not packages:
-        packages = []
-        for name in os.listdir(scripts_path):
-            if name.startswith('compas') and not name.endswith('.py'):
-                packages.append(name)
-
     packages = _filter_installed_packages(version, packages)
 
+    # Also remove all broken symlinks
+    # because ... they're broken!
+    for name in os.listdir(scripts_path):
+        path = os.path.join(scripts_path, name)
+        if os.path.islink(path):
+            if not os.path.exists(path):
+                if name not in packages:
+                    packages.append(name)
+
     print('Uninstalling COMPAS packages from Rhino {0} scripts folder: \n{1}'.format(version, scripts_path))
 
     results = []
@@ -208,7 +207,7 @@ def after_rhino_uninstall(uninstalled_packages):
 
     parser = argparse.ArgumentParser()
 
-    parser.add_argument('-v', '--version', choices=['5.0', '6.0', '7.0'], default='6.0', help="The version of Rhino to install the packages in.")
+    parser.add_argument('-v', '--version', choices=['5.0', '6.0', '7.0', '8.0'], default='6.0', help="The version of Rhino to install the packages in.")
     parser.add_argument('-p', '--packages', nargs='+', help="The packages to uninstall.")
 
     args = parser.parse_args()