diff options
| -rw-r--r-- | __init__.py | 116 | ||||
| -rw-r--r-- | blender_funcs.py | 1 | ||||
| -rw-r--r-- | required_modules.py | 51 |
3 files changed, 80 insertions, 88 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() diff --git a/blender_funcs.py b/blender_funcs.py index e18431e..a4a15aa 100644 --- a/blender_funcs.py +++ b/blender_funcs.py @@ -1,5 +1,4 @@ import bpy, numpy -import time # file that contains useful blender some functions that I don't # want to re-write on each individual .py file diff --git a/required_modules.py b/required_modules.py deleted file mode 100644 index 5eb1604..0000000 --- a/required_modules.py +++ /dev/null @@ -1,51 +0,0 @@ -import bpy, numpy, subprocess -from . import file_ops - -# register func -@bpy.app.handlers.persistent -def register(dummy): - # get python's binary path (literally my saviour) - 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:"] - upd_mods = ["pip", "wheel", "setuptools"] - new_mods = ["lxml"] - - # try importing pip/lxml to trigger - # the module updating on failure - try: - import pip - import lxml - except: - # hecking install pip (ensurepip) + required modules - subprocess.run([py_bin, "-B", "-m", "ensurepip"]) - # update basic modules then download new modules - subprocess.run(pip_install + upd_mods) - subprocess.run(pip_install + new_mods) - # rename numpy's installation folder (cannot delete the folder in windows) - file_ops.rename(numpy.__path__[0], numpy.__path__[0] + "_bad") - subprocess.run(pip_install + ["numpy==1.18.5"]) # install a newer version - print("First exit!") - bpy.ops.wm.quit_blender() - exit(0) - - # print module's versions installed - print("pip %s is installed!" % (pip.__version__)) - print("lxml %s is installed!" % (lxml.__version__)) - # for those who have installed blenxy before, update numpy - # hacky way to do it but it is the only way that works - if (numpy.__version__ != "1.18.5"): - # rename numpy's installation folder (cannot delete the folder in windows) - file_ops.rename(numpy.__path__[0], numpy.__path__[0] + "_bad") - subprocess.run(pip_install + ["numpy==1.18.5"]) # install a newer version - print("Second exit!") - bpy.ops.wm.quit_blender() - exit(0) - print("numpy %s is installed!" % (numpy.__version__)) - -# unregister func -def unregister(): - return |