Add post-install command to get Java dependencies

Author: ruivieira

setup.py

@@ -1,10 +1,30 @@
 import os
 from setuptools import setup
+from setuptools.command.install import install
+import site
 
 here = os.path.abspath(os.path.dirname(__file__))
 with open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
     long_description = f.read()
 
+TRUSTY_VERSION = "1.12.0.Final"
+
+
+class PostInstall(install):
+    """Fetch TrustyAI explainability JARs from Maven Central"""
+
+    def run(self):
+        install.run(self)
+        _ROOT = os.path.join(site.getsitepackages()[0], "trustyai", "dep")
+        print(f"Installing Maven dependencies into {_ROOT}")
+        os.system(f"mvn org.apache.maven.plugins:maven-dependency-plugin:2.10:get "
+                  f"-DremoteRepositories=https://repository.sonatype.org/content/repositories/central  "
+                  f"-Dartifact=org.kie.kogito:explainability-core:{TRUSTY_VERSION} -Dmaven.repo.local={_ROOT} -q")
+        _TESTS_FILE = os.path.join("org", "kie", "kogito", "explainability-core", TRUSTY_VERSION,
+                                   f"explainability-core-{TRUSTY_VERSION}-tests.jar")
+        os.system(f"wget -O {os.path.join(_ROOT, _TESTS_FILE)} https://repo1.maven.org/maven2/{_TESTS_FILE}")
+
+
 setup(
     name="trustyai",
     version="0.0.5",
@@ -29,4 +49,5 @@
     packages=['trustyai', 'trustyai.model', 'trustyai.utils', 'trustyai.local'],
     include_package_data=False,
     install_requires=['Jpype1'],
+    cmdclass={"install": PostInstall},
 )

tests/common.py

@@ -14,10 +14,10 @@
 if not INITIALISED:
     trustyai.init(
         path=trustyai.CORE_DEPS + [
-            "./dep/org/optaplanner/optaplanner-core/8.12.0.Final/optaplanner-core-8.12.0.Final.jar",
-            "./dep/org/apache/commons/commons-math3/3.6.1/commons-math3-3.6.1.jar",
-            "./dep/org/kie/kie-api/7.59.0.Final/kie-api-7.59.0.Final.jar",
-            "./dep/io/micrometer/micrometer-core/1.7.4/micrometer-core-1.7.4.jar",
+            f"{trustyai.DEFAULT_DEP_PATH}/org/optaplanner/optaplanner-core/8.12.0.Final/optaplanner-core-8.12.0.Final.jar",
+            f"{trustyai.DEFAULT_DEP_PATH}/org/apache/commons/commons-math3/3.6.1/commons-math3-3.6.1.jar",
+            f"{trustyai.DEFAULT_DEP_PATH}/org/kie/kie-api/7.59.0.Final/kie-api-7.59.0.Final.jar",
+            f"{trustyai.DEFAULT_DEP_PATH}/io/micrometer/micrometer-core/1.7.4/micrometer-core-1.7.4.jar",
         ]
     )
 

trustyai/__init__.py

@@ -5,12 +5,16 @@
 import jpype
 import jpype.imports
 from jpype import _jcustomizer, _jclass
+import site
+import os
 
 TRUSTY_VERSION = "1.12.0.Final"
+DEFAULT_DEP_PATH = os.path.join(site.getsitepackages()[0], "trustyai", "dep")
+
 CORE_DEPS = [
-    f"./dep/org/kie/kogito/explainability-core/{TRUSTY_VERSION}/*",
-    "./dep/org/slf4j/slf4j-api/1.7.30/slf4j-api-1.7.30.jar",
-    "./dep/org/apache/commons/commons-lang3/3.12.0/commons-lang3-3.12.0.jar",
+    f"{DEFAULT_DEP_PATH}/org/kie/kogito/explainability-core/{TRUSTY_VERSION}/*",
+    f"{DEFAULT_DEP_PATH}/org/slf4j/slf4j-api/1.7.30/slf4j-api-1.7.30.jar",
+    f"{DEFAULT_DEP_PATH}/org/apache/commons/commons-lang3/3.12.0/commons-lang3-3.12.0.jar",
 ]