PythonForAivia/PythonEnvForAivia/FirstTimeSetup.py at master · AiviaCommunity/PythonForAivia · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import os.path
import subprocess
import pathlib
from pathlib import Path
from shutil import copyfile
import sys
import ctypes
from skimage.io import imread, imsave
import numpy as np

"""
This Aivia python recipe will create the required virtual environment for all recipes available
on our GitHub: https://github.com/AiviaCommunity/PythonForAivia.

Unzip the content of PythonEnvForAivia.zip in a folder WITHOUT admin access restrictions.
"""

# [INPUT Name:inputImagePath Type:string DisplayName:'Any 2D Image']
# [OUTPUT Name:outputImagePath Type:string DisplayName:'To Delete']
def run(params):

    env_dir = pathlib.Path(os.path.dirname(os.path.realpath(__file__))) / 'env'

    if not os.path.exists(env_dir):
        # create a virtual environment
        env_dir.mkdir(parents=False, exist_ok=True)
        subprocess.check_call([str(Path(sys.executable).parent / 'Scripts/virtualenv.exe'), f'{env_dir}'])

        # copy essential python packages(python312.zip) to virtual environment
        # see https://github.com/pypa/virtualenv/issues/1185
        if not os.path.exists(env_dir/'Scripts/python312.zip'):
            copyfile(Path(sys.executable).parent / 'python312.zip', env_dir/'Scripts/python312.zip')

        # install requirements
        mess = 'Python packages will now be installed. An internet connection is needed.\n\n' \
               'You can follow the addition of the packages in the following subfolder:\n' + str(env_dir) + \
               '\\Lib\\site-packages'
        Mbox('Starting installing python packages', mess, 0)

        pip_path = env_dir / 'Scripts' / 'pip.exe'
        requirement_dir = pathlib.Path(os.path.dirname(os.path.realpath(__file__)))
        # subprocess.check_call(
        #     [str(pip_path), 'install', 'setuptools==70.0.0'])
        subprocess.check_call(
            [str(pip_path), 'install', '-r', str(requirement_dir/'requirements.txt')])

    # Check if input image exists
    inputImagePath_ = params['inputImagePath']
    outputImagePath_ = params['outputImagePath']
    if not os.path.exists(inputImagePath_):
        raise ValueError('Error: {inputImagePath_} does not exist')

    # Get the path of the folder that contains this python script
    parentFolder = str(Path(__file__).parent)

    # Get the path of python executable in the virtual environment
    pythonExec_ = parentFolder + '\\env\\Scripts\\python.exe'

    # Log
    message = f'Python was installed here:\n{pythonExec_}'
    print(message)

    # Load image
    input_img = imread(inputImagePath_)

    imsave(outputImagePath_, input_img)


def Mbox(title, text, style):
    return ctypes.windll.user32.MessageBoxW(0, text, title, style)


if __name__ == '__main__':
    params = {}
    params['inputImagePath'] = 'test_8b.tif'
    params['outputImagePath'] = 'testResult.tif'

    run(params)