1 files changed, 37 insertions, 0 deletions
diff --git a/src/pybind/mgr/rook/rook-client-python/generate.sh b/src/pybind/mgr/rook/rook-client-python/generate.sh
new file mode 100755
index 000000000..5aa189276
--- /dev/null
+++ b/src/pybind/mgr/rook/rook-client-python/generate.sh
@@ -0,0 +1,37 @@
+#!/bin/bash
+
+set -ex
+
+if [ -z "$1" ] ; then
+    rook_base="${GOPATH:-$HOME/go}/src/github.com/rook/rook"
+else
+    rook_base="$1"
+fi
+crd_base="$rook_base/cluster/examples/kubernetes"
+
+cd "$(dirname "$0")"
+
+if ! [ -x "$(command -v python3)" ]; then
+  echo 'Error: python3 is not installed.' >&2
+  exit 1
+fi
+
+if [ ! -d venv ]
+then
+  python3 -m venv venv
+  . venv/bin/activate
+  pip install -r requirements.txt
+else
+  . venv/bin/activate 
+fi
+
+python generate_model_classes.py "$crd_base/ceph/common.yaml" "rook_client/ceph"
+python generate_model_classes.py "$crd_base/cassandra/operator.yaml" "rook_client/cassandra"
+python generate_model_classes.py "$crd_base/edgefs/operator.yaml" "rook_client/edgefs"
+
+
+python setup.py develop
+
+tox --skip-missing-interpreters=true -- --crd_base="$crd_base"
+
+deactivate