add numba dependency (#4209)

5a3492f8 · Dealga McArdle · GitHub · 1ca7fd69 · 5a3492f8 · 5a3492f8
--- a/dependencies.py
+++ b/dependencies.py
@@ -128,7 +128,7 @@ except ImportError:
    info(freecad_d.message)
    FreeCAD = None

-cython_d = sv_dependencies["cython"] = SvDependency("Cython", "https://www.freecadweb.org/")
+cython_d = sv_dependencies["cython"] = SvDependency("Cython", "https://cython.org/")
 cython_d.pip_installable = True
 try:
    import Cython
@@ -139,6 +139,17 @@ except ImportError:
    info(cython_d.message)
    Cython = None

+numba_d = sv_dependencies["numba"] = SvDependency("Numba", "https://numba.pydata.org/")
+numba_d.pip_installable = True
+try:
+    import numba
+    numba_d.message = "Numba package is available"
+    numba_d.module = numba
+except ImportError:
+    numba_d.message = "Numba package is not available, njit compiled functions will not be available"
+    info(numba_d.message)
+    numba = None
+
 good_names = [d.package for d in sv_dependencies.values() if d.module is not None and d.package is not None]
 if good_names:
    info("Dependencies available: %s.", ", ".join(good_names))

--- a/settings.py
+++ b/settings.py
@@ -162,10 +162,10 @@ class SvSetFreeCadPath(bpy.types.Operator):
                site_packages = p
                break

-        file_path= open(os.path.join(site_packages, "freecad_path.pth"), "w+")
-
+        file_path = open(os.path.join(site_packages, "freecad_path.pth"), "w+")
        file_path.write(self.FreeCAD_folder)
        file_path.close()
+
        self.report({'INFO'}, "FreeCad path saved successfully. Please restart Blender to see effect.")
        return {'FINISHED'}

@@ -602,6 +602,7 @@ dependencies, or install only some of them.""")
        draw_message(box, "mcubes")
        draw_message(box, "circlify")
        draw_message(box, "cython")
+        draw_message(box, "numba")

        draw_freecad_ops()