more robust module checking/updating

1 files changed, 80 insertions, 36 deletions

diff --git a/__init__.py b/__init__.py
index d20580a..e2971fd 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2,74 +2,118 @@
 # https://s-nako.work/2020/09/blender-error-attributeerror-_restrictcontext-object-has-no-attribute-view_layer/
 # https://web.archive.org/web/20210925181415/https://blenderbrew.com/custom-application-templates-in-blender/
 
-# blenxy module
-import importlib, bpy, sys
+# elemental python modules blenxy needs
+# that the python interpreter must have built in
+import bpy, importlib, sys, subprocess, os, shutil
+import math, mathutils, warnings, struct, site
 
-# import the submmodules needed
-# all submodules have a register/unregister function
-# register function is to be executed on blenxy's load
-# name can be anything but I want to keep the same nomenclature
-# the dummy variable passed to the register funcs is needed
-# since it is passed when being assigned to the load_post list
-# (seems to be None always)
-from . import required_modules        # install needed modules for the bundled python
-from . import basic_settings          # user settings blenxy has
-from . import collada_superbmd_import # custom importer for SuperBMD collada files
-from . import collada_superbmd_export # custom exporter for SuperBMD collada files
-from . import obj_kcl_export          # custom exporter for OBJ files (collision) 
-from . import obj_neokclcreate_import # custom importer for OBJ files (collision, NeoKCLCreate)
-from . import bck_import              # custom importer for SMG BCK files
-from . import bck_export              # custom exporter for SMG BCK files
+# elemental python modules blenxy needs
+# that the python interpreter probably does not have built in/needs to be updated
+pip_modules = ["pip", "wheel", "setuptools"]
+new_modules = ["lxml", "numpy"]
 
-# global variables
-blenxy_modules = [required_modules,
-                  basic_settings,
-                  collada_superbmd_import,
-                  collada_superbmd_export,
-                  obj_kcl_export,
-                  obj_neokclcreate_import,
-                  bck_import,
-                  bck_export]
+# check which of the new modules needs to be installed
+def new_mod_check(mod_names):
+  rtn = []
+  for mod_name in mod_names:
+    try:
+      importlib.import_module(mod_name)
+    except:
+      rtn.append(mod_name)
+  return rtn
+  
+# install given modules with pip but be sure to have pip first
+def new_mod_install(mod_names):
+  # get python's interpreter binary path
+  py_bin = bpy.app.binary_path_python
+  pip_install = [py_bin, "-B", "-m", "pip", "install",
+                 "--trusted-host", "pypi.python.org",
+                 "--trusted-host", "files.pythonhosted.org",
+                 "--trusted-host", "pypi.org",
+                 "-U", "--force-reinstall", "--only-binary", ":all:"]
+  
+  # check if pip is available
+  pip = None
+  try: # import the pip module
+    pip = importlib.import_module("pip")
+  except: # install pip  
+    subprocess.run([py_bin, "-B", "-m", "ensurepip"])
+    subprocess.run(pip_install + pip_modules)
+  
+  # install the rest of the modules
+  if (mod_names != []):
+    for mod_name in mod_names:
+      if (mod_name == "numpy"): # specific to the numpy module
+        # get the numpy package path
+        numpy_path = None
+        for path in site.getsitepackages():
+          if ("2.79" in path):
+            numpy_path = path + "/numpy"
+            break
+        os.rename(numpy_path, numpy_path + "_bad")
+      # normal module installation
+      subprocess.run(pip_install + [mod_name])
+    return True
+  return False
+    
+# all blenxy custom modules
+blenxy_modules = ["basic_settings",           # user settings blenxy has
+                  "collada_superbmd_import",  # custom importer for SuperBMD collada files
+                  "collada_superbmd_export",  # custom exporter for SuperBMD collada files
+                  "obj_kcl_export",           # custom exporter for OBJ files (collision) 
+                  "obj_neokclcreate_import",  # custom importer for OBJ files (collision, NeoKCLCreate)
+                  "bck_import",               # custom importer for SMG BCK files
+                  "bck_export"]               # custom exporter for SMG BCK files
 
-# function used to refresh blenxy's used modules 
+# function used to refresh blenxy's custom modules 
 # so that they can be updated without closing blender
 @bpy.app.handlers.persistent
 def unload_blenxy_stuff(dummy):
-  
   # execute the unregister functions from each module
   # these functions won't die if the stuff to unregister isn't there
-  for mod in blenxy_modules:
+  for mod_name in blenxy_modules:
+    mod = importlib.import_module("." + mod_name, __package__)
     mod.unregister()
   # reload the modules in case they were modified (dev stuff)
-  for mod in blenxy_modules:
+  for mod_name in blenxy_modules:
+    mod = importlib.import_module("." + mod_name, __package__)
     importlib.reload(mod)
 
 # register function that blender will call when the template is loaded
 def register():
-  
   # print the welcome
   print("\nWelcome to Blenxy!")  
   print("Running on Blender: %s\n" % (bpy.app.version.__str__()))
   
+  # checking python modules
+  if (new_mod_install(new_mod_check(new_modules)) == True):
+    print("New modules installed. Exiting...")
+    exit(0)
+  for mod_name in new_modules:
+    mod = importlib.import_module(mod_name)
+    print("%s %s is installed!" % (mod_name, mod.__version__))
+  
   # add this function to load_post first
   bpy.app.handlers.load_post.append(unload_blenxy_stuff)
-  # add the register functions to load_post (old_blenxy_modules)
-  for mod in blenxy_modules:
+  # add the register functions to load_post
+  for mod_name in blenxy_modules:
+    mod = importlib.import_module("." + mod_name, __package__)
     bpy.app.handlers.load_post.append(mod.register)
   
 # unregister function that blender will call when a new template is loaded
 def unregister():
-  
   # print the goodbye
   print("\nLeaving Blenxy!\n")
   # remove the stuff added to the load_post list
   bpy.app.handlers.load_post.remove(unload_blenxy_stuff)
-  for mod in blenxy_modules:
+  for mod_name in blenxy_modules:
+    mod = importlib.import_module("." + mod_name, __package__)
     for func in bpy.app.handlers.load_post:
       if (mod.__name__ in func.__module__):
         bpy.app.handlers.load_post.remove(func)
         break
     
   # execute module's unregister functions
-  for mod in blenxy_modules:
+  for mod_name in blenxy_modules:
+    mod = importlib.import_module("." + mod_name, __package__)
     mod.unregister()