AiPacketClassifier/env-export.sh at master · niting3c/AiPacketClassifier · 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
#!/bin/bash

# Check the number of arguments
if [ "$#" -lt 2 ]; then
    echo "Usage: $0 <operating_system> <environment_name>"
    exit 1
fi

# Assign the arguments to variables
os="$1"
environment_name="$2"

# Check the argument for the operating system
if [ "$os" == "windows" ]; then
    # Extract installed pip packages
    pip_packages=$(conda list -n "$environment_name" | grep -v "^#" | grep "pypi" | awk '{ print $1 }')

    # Export conda environment without builds, and append pip packages
    conda env export -n "$environment_name" --no-builds | grep -v "^prefix: " > environment-windows.yml
    echo "$pip_packages" >> environment-windows.yml

    echo "Environment exported to environment-windows.yml"

elif [ "$os" == "mac" ]; then
    # Extract installed pip packages
    pip_packages=$(conda list -n "$environment_name" | grep -v "^#" | grep "pypi" | awk '{ print $1 }')

    # Export conda environment without builds, and append pip packages
    conda env export -n "$environment_name" --no-builds | grep -v "^prefix: " > environment-mac.yml
    echo "$pip_packages" >> environment-mac.yml

    echo "Environment exported to environment-mac.yml"

elif [ "$os" == "linux" ]; then
    # Extract installed pip packages
    pip_packages=$(conda list -n "$environment_name" | grep -v "^#" | grep "pypi" | awk '{ print $1 }')

    # Export conda environment without builds, and append pip packages
    conda env export -n "$environment_name" --no-builds | grep -v "^prefix: " > environment-linux.yml
    echo "$pip_packages" >> environment-linux.yml

    echo "Environment exported to environment-linux.yml"

else
    echo "Invalid argument. Please specify 'windows', 'mac', or 'linux' as the first argument."
    exit 1
fi