pack_molecular.py

import sys
from zipfile import ZipFile, ZIP_DEFLATED
from os import path, walk, remove, rmdir, chdir, getcwd, mkdir, rename
import shutil
import platform
from subprocess import Popen, PIPE

is_linux = platform.system() == "Linux"
is_windows = platform.system() == "Windows"

# in python 3.8.x, sys.abiflags attribute doesnt seem to exist any more instead of returning empty string.
# so better check for existence here before accessing it.
abiflags = ''
if hasattr(sys, 'abiflags'):
    abiflags = sys.abiflags

v = str(sys.version_info.major) + str(sys.version_info.minor) + abiflags

name = 'macos'
if is_linux:
    name = 'linux'
elif is_windows:
    name = 'win'

mkdir(".//molecularplus" + name)

pyfiles = (
    "__init__.py", "creators.py", "descriptions.py", "names.py", "operators.py", "properties.py",
    "simulate.py", "ui.py", "utils.py", "addon_prefrences.py")

for file in pyfiles:
    shutil.copy(file, ".//molecularplus//" + file)

from molecularplus import bl_info

chdir(getcwd() + "//c_sources")

# TODO, blenders (or a compatible) python bin needs to be in $PATH, and if you use blender's you need to copy the python includes from SVN
# into the include folder of blenders python, too

version = '.'.join(map(str, bl_info['version']))

with Popen([sys.executable, "setup.py", "build_ext", "--inplace"], stdout=PIPE) as proc:
    proc.stdout.read()

    if is_linux:  # TODO, test
        shutil.move("core.cpython-{}-x86_64-linux-gnu.so".format(v),
                    "..//molecularplus//core.cpython-{}-x86_64-linux-gnu.so".format(v))
    elif is_windows:
        shutil.move("core.cp{}-win_amd64.pyd".format(v), "..//molecularplus//core.cp{}-win_amd64.pyd".format(v))
    else:
        shutil.move("core.cpython-{}-darwin.so".format(v), "..//molecularplus//core.cpython-{}-darwin.so".format(v))

    chdir("..")

    molfiles = (
    "__init__.py", "creators.py", "descriptions.py", "names.py", "operators.py", "properties.py", "addon_prefrences.py",
    "simulate.py", "ui.py", "utils.py", 'core.cpython-{}-darwin.so'.format(v), 'core.cp{}-win_amd64.pyd'.format(v), 'core.cpython-{}-x86_64-linux-gnu.so'.format(v))

    with ZipFile('molecular-plus_{}_'.format(version) + '{}_'.format(v) + name + '.zip', 'w') as z:
        for root, _, files in walk('molecularplus'):
            for file in files:
                if file not in molfiles:
                    continue
                z.write(path.join(root, file), compress_type=ZIP_DEFLATED)

    # cleanup

    shutil.rmtree(".//molecularplus")

    # chdir(getcwd() + "//molecularplus")
    # try:
    #     remove("*.*")
    # except:
    #     pass
    # chdir("..")
    
    chdir(getcwd() + "//c_sources")

    try:
        remove("core.html")
        remove("core.c")
        shutil.rmtree("build")
    except:
        pass

    chdir("..")

print(version)