From 157dcc44bcdf3485f26eddcfa9a7b729e9f68aae Mon Sep 17 00:00:00 2001
From: Arthur Meyre <arthur.meyre@zama.ai>
Date: Thu, 16 Sep 2021 09:46:42 +0200
Subject: [PATCH] tools: add a script to easily upgrade python dependencies

---
 Makefile                          |  4 ++++
 script/make_utils/upgrade_deps.sh | 18 ++++++++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100755 script/make_utils/upgrade_deps.sh

diff --git a/Makefile b/Makefile
index d0455f3e8..54db0939b 100644
--- a/Makefile
+++ b/Makefile
@@ -191,3 +191,7 @@ jupyter:
 release_docker:
 	./docker/build_release_image.sh
 .PHONY: release_docker
+
+upgrade_py_deps:
+	./script/make_utils/upgrade_deps.sh
+.PHONY: upgrade_py_deps
diff --git a/script/make_utils/upgrade_deps.sh b/script/make_utils/upgrade_deps.sh
new file mode 100755
index 000000000..e5b40e8bc
--- /dev/null
+++ b/script/make_utils/upgrade_deps.sh
@@ -0,0 +1,18 @@
+#!/usr/bin/env bash
+
+# verbose output please
+set -v
+
+no_dev_file=$(mktemp --suffix=.txt)
+all_file=$(mktemp --suffix=.txt)
+dev_file=$(mktemp --suffix=.txt)
+
+poetry show -o -t --no-dev | grep -v -e "--" | cut -d " " -f 1 | sed 's/$/\@latest/g' > "${no_dev_file}"
+poetry show -o -t | grep -v -e "--" | cut -d " " -f 1 | sed 's/$/\@latest/g' > "${all_file}"
+join -v1 -v2 "${all_file}" "${no_dev_file}" > "${dev_file}"
+cat "${no_dev_file}" | xargs poetry add
+cat "${dev_file}" | xargs poetry add --dev
+
+rm "${no_dev_file}"
+rm "${dev_file}"
+rm "${all_file}"