Ethan_misc/check_python_environment.m at master · proektlab/Ethan_misc · 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
function env = check_python_environment(envname, test_command, python_path)

arguments
    envname (1,:) string = "caiman"
    test_command (1,:) char = ''  % Python command to test
    python_path (1,:) char = ''  % Path to python passed explicitly
end

% If a python environment is not loaded, attempts to initialize it with the camian environment
curr_pyenv = pyenv;
if curr_pyenv.Status ~= "Loaded"
    [~, hostname] = system('hostname');
    if ~isempty(python_path)
        fprintf('Initializing python environment from %s\n', python_path);
        env = pyenv('Version', python_path);
    elseif ispc
        if envname == "caiman" && strcmpi(strtrim(hostname), 'lust')
            % use local caiman environment if we're on Lust
            python_path = 'C:\Users\ethan\AppData\Local\miniforge3\envs\mesviz\python';
        else
            synology_dir = get_synology_dir;
            python_path = fullfile(synology_dir, 'conda_envs_windows', envname, 'python');
        end

        fprintf('Initializing python environment from %s\n', python_path);
        env = pyenv('Version', python_path);
    elseif contains(hostname, 'sorcery')
        python_path = '/home/ethan/miniforge3/envs/mescore/bin/python';
        fprintf('Initializing python environment from %s\n', python_path);
        env = pyenv('Version', python_path);
    else
        % python might still be usable, but we don't know the path
        warning('Python must be initialized to load data from caiman');
        env = curr_pyenv;
    end
else
    env = curr_pyenv;
end

if strlength(env.Version) == 0
    error('Python environment "%s" needed, but none was found', envname);
end

if ~isempty(test_command)
    try pyrun(test_command)
    catch me
        if isa(me, 'matlab.exception.PyException')
            new_err = MException('EthanCode:badPythonEnv', ...
                'Could not run command "%s" in current python environment', test_command);
            new_err = addCause(new_err, me);
            throw(new_err);
        else
            rethrow(me);
        end
    end
end

end